Fix for #4553
[oweals/gnunet.git] / src / my / test_my.c
index 1ab6ef1b77f55bb407d83d14bb70e910c4416a3b..d213bf547c56914c7f358d97d9e12e5ea58b53fa 100644 (file)
 #include "platform.h"
 #include <mysql/mysql.h>
 #include "gnunet_my_lib.h"
-
-/**
-  * Setup prepared statements.
-  *
-  * @param mysql connection handle to initialize
-  * @return 
-          #GNUNET_OK on success
-          #GNUNET_SYSERR on failure
-  */
-static int
-mysql_prepare (MYSQL * mysql)
-{
-     MYSQL_RES *result;
-     MYSQL_STMT *stmt;
-
-     stmt = mysql_stmt_init (mysql);
-#define PREPARE (name, sql, ...)
-     do
-     {
-          result = mysql_stmt_prepare(stmt, name, sql, __VA_ARGS__);
-          if (result)
-          {
-               GNUNET_break (0);
-               mysql_stmt_free_result (stmt);
-               result = NULL;
-               return GNUNET_SYSERR; 
-          }
-          mysql_stmt_free_result (stmt);
-          result = NULL;
-     } while (0);
-
-     PREPARE ("test_insert",
-               "INSERT INTO test_my ("
-               "pub"
-               ",sig"
-               ",abs_time"
-               ",forever"
-               ",hash"
-               ",vsize"
-               ",u16"
-               ",u32"
-               ",u64"
-               ") VALUES "
-               "($1, $2, $3, $4, $5, $6,
-                    $7, $8, $9);",
-               9, NULL);
-
-     PREPARE ("test_select",
-               "SELECT"
-               "pub"
-               ",sig"
-               ",abs_time"
-               ",forever"
-               ",hash"
-               ",vsize"
-               ",u16"
-               ",u32"
-               ",u64"
-               " FROM test_my"
-               " ORDER BY abs_time DESC"
-               " LIMIT 1;",
-               0, NULL);
-     return GNUNET_OK;
-#undef PREPARE
-}
+#include "gnunet_mysql_lib.h"
+#include "gnunet_util_lib.h"
 
 /**
   * Run actual test queries.
   *
-  * @param mysql coonection handle to initialize
+  * @param contexte the current context of mysql
   * @return 0 on succes
   */
 static int
-run_queries (MYSQL * mysql)
+run_queries (struct GNUNET_MYSQL_Context *context)
 {
      struct GNUNET_CRYPTO_RsaPublicKey *pub;
-     struct GNUNET_CRYPTO_RsaPublicKey *pub2 = NULL;
+//     struct GNUNET_CRYPTO_RsaPublicKey *pub2 = NULL;
      struct GNUNET_CRYPTO_RsaSignature *sig;
-     struct GNUNET_CRYPTO_RsaSignature *sig2 = NULL;
-     struct GNUNET_TIME_Absolute abs_time = GNUNET_TIME_absolute_get () ;
-     struct GNUNET_TIME_Absolute abs_time2;
+//     struct GNUNET_CRYPTO_RsaSignature *sig2 = NULL;
+     struct GNUNET_TIME_Absolute abs_time = GNUNET_TIME_absolute_get ();
+//     struct GNUNET_TIME_Absolute abs_time2;
      struct GNUNET_TIME_Absolute forever = GNUNET_TIME_UNIT_FOREVER_ABS;
-     struct GNUNET_TIME_Absolute forever2;
+//     struct GNUNET_TIME_Absolute forever2;
      struct GNUNET_HashCode hc;
-     struct GNUNET_HashCode hc2;
-     MYSQL_RES * result;
-     int ret;
-     struct GNUNET_CRYPTO_RsaPrivateKey *priv;
+//     struct GNUNET_HashCode hc2;
      const char msg[] = "hello";
-     void *msg2;
-     struct GNUNET_HashCode hmsg;
-     size_t msg2_len;
+//     void *msg2;
+     size_t msg_len;
+//     size_t msg2_len;
+
      uint16_t u16;
-     uint16_t u162;
+//     uint16_t u162;
      uint32_t u32;
-     uint32_t u322;
+//     uint32_t u322;
      uint64_t u64;
-     uint64_t u642;
+//     uint64_t u642;
+
+     msg_len = sizeof(msg);
+
+//     int ret;
+
+     struct GNUNET_MYSQL_StatementHandle *statements_handle_insert;
+
+//     struct GNUNET_MYSQL_StatementHandle *statements_handle_select;
+
+     struct GNUNET_CRYPTO_RsaPrivateKey *priv;
+     struct GNUNET_HashCode hmsg;
 
      priv = GNUNET_CRYPTO_rsa_private_key_create (1024);
-     pub = GNUNET_CRYPTO_rsa_private_key_get_public (priv);
-     memset (&hmsg, 42, sizeof (hmsg));
+     pub =  GNUNET_CRYPTO_rsa_private_key_get_public (priv);
+     memset (&hmsg, 42, sizeof(hmsg));
      sig = GNUNET_CRYPTO_rsa_sign_fdh (priv,
-                                       &hmsg);
+                                        &hmsg);
      u16 = 16;
      u32 = 32;
      u64 = 64;
 
+     statements_handle_insert = GNUNET_MYSQL_statement_prepare (context,
+                                        "INSERT INTO test_my ("
+                                        " pub"
+                                        ",sig"
+                                        ",abs_time"
+                                        ",forever"
+                                        ",hash"
+                                        ",vsize"
+                                        ",u16"
+                                        ",u32"
+                                        ",u64"
+                                        ") VALUES "
+                                        "( ?, ?, ?, ?, ?, ?, ?, ?, ?)");
+
+     if (NULL == statements_handle_insert)
      {
-          struct GNUNET_MY_QueryParam params_insert[] = {
+          fprintf (stderr, "Failed to prepared statement INSERT\n");
+          return 1;
+     }
+
+     struct GNUNET_MY_QueryParam params_insert[] = {
           GNUNET_MY_query_param_rsa_public_key (pub),
           GNUNET_MY_query_param_rsa_signature (sig),
           GNUNET_MY_query_param_absolute_time (&abs_time),
@@ -145,98 +110,75 @@ run_queries (MYSQL * mysql)
           GNUNET_MY_query_param_uint32 (&u32),
           GNUNET_MY_query_param_uint64 (&u64),
           GNUNET_MY_query_param_end
-          };
-
-          struct GNUNET_MY_QueryParam params_select[] = {
-               GNUNET_MY_query_param_end
-          };
-
-          struct GNUNET_MY_ResultSpec results_select[] = {
-               GNUNET_MY_result_spec_rsa_public_key ("pub", &pub2),
-               GNUNET_MY_result_spec_rsa_signature ("sig", &sig2),
-               GNUNET_MY_result_spec_absolute_time ("abs_time", &abs_time2),
-               GNUNET_MY_result_spec_absolute_time ("forever", &forever2),
-               GNUNET_MY_result_spec_auto_from_type ("hash", &hc2),
-               GNUNET_MY_result_spec_variable_size ("vsize", &msg2, &msg2_len),
-               GNUNET_MY_result_spec_uint16 ("u16", &u162),
-               GNUNET_MY_result_spec_uint32 ("u32", &u322),
-               GNUNET_MY_result_spec_uint64 ("u64", &u642),
-               GNUNET_MY_result_epc_end
-          };
-
-          if(GNUNET_MY_exec_prepared (  mysql,
-                                        "test_insert",
-                                        params_insert))
-          {
-               GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
-                         "Database failure: \n");
-               
-               //free result
-
-               GNUNET_CRYPTO_rsa_signature_free (sig);
-               GNUNET_CRYPTO_rsa_private_key_free (priv);
-               GNUNET_CYPTO_rsa_public_key_free (pub);
-               return 1;
-          }
-
-          //free result
-
-          result = GNUNET_MY_exec_prepared (mysql, "test_select", params_select);
-/*          if(1 != mysql_fetch_length (result))
-          {
-               GNUNET_break (0);
-               GNUNET_CRYPTO_rsa_signature_free (sig);
-               GNUNET_CRYPTO_rsa_private_key_free (priv);
-               GNUNET_CRYPTO_rsa_public_key_free (pub);
-
-               return 1;
-          }
-*/
-          if (GNUNET_MY_exec_prepared (mysql
-                                        , "test_select"
-                                        , params_select))
-          {
-               GNUNET_break (0);
-               GNUNET_CRYPTO_rsa_signature_free (sig);
-               GNUNET_CRYPTO_rsa_private_key_free (priv);
-               GNUNET_CRYPTO_rsa_public_key_free (pub);
+     };
 
-               return 1;
-          }
+     if (GNUNET_OK != GNUNET_MY_exec_prepared(context,
+                                             statements_handle_insert,
+                                             params_insert))
+     {
+          fprintf (stderr, "Failed to execute prepared statement INSERT\n");
+          return 1;
+     }
 
 
-          ret = GNUNET_MY_extract_result (result,
-                                        results_select,
-                                        0);
-          GNUNET_break (GNUNET_YES == ret);
-          GNUNET_break (abs_time.abs_value_us == abs_time2.abs_value_us);
-          GNUNET_break (forever.abs_value_us == forever2.abs_value_us);
-          GNUNET_break (0 == 
-                         memcmp (&hc,
-                                 &hc2,
-                                 sizeof (struct GNUNET_HashCode)));
-          GNUNET_break (0 ==
-                         GNUNET_CRYPTO_rsa_signature_cmp (sig,
-                                                          sig2));
-          GNUNET_break (0 ==
-                         GNUNET_CRYPTO_rsa_public_key_cmp (pub,
-                                                            pub2));
-          GNUNET_break (strlen (msg) == msg2_len);
-          GNUNET_break (0 == 
-                         strncmp (msg,
-                                   msg2,
-                                   msg2_len));
-          GNUNET_break (16 == u162);
-          GNUNET_break (32 == u322);
-          GNUNET_break (64 == u642);
+
+/*   statements_handle_select = GNUNET_MYSQL_statement_prepare (context,
+                                                                 "SELECT"
+                                                                 " pub"
+                                                                 ",sig"
+                                                                 ",abs_time"
+                                                                 ",forever"
+                                                                 ",hash"
+                                                                 ",vsize"
+                                                                 ",u16"
+                                                                 ",u32"
+                                                                 ",u64"
+                                                                 " FROM test_my"
+                                                                 " ORDER BY abs_time DESC "
+                                                                 " LIMIT 1;");
+
+     if (NULL == statements_handle_select)
+     {
+          fprintf(stderr, "Failed to prepared statement SELECT\n");
+          return 1;
      }
-     
-     GNUNET_CRYPTO_rsa_signature_free (sig);
-     GNUNET_CRYPTO_rsa_private_key_free (priv);
-     GNUNET_CRYPTO_rsa_public_key_free (pub);
 
+     struct GNUNET_MY_QueryParam params_select[] = {
+          GNUNET_MY_query_param_end
+     };
+
+     if (GNUNET_OK != GNUNET_MY_exec_prepared (context,
+                                             statements_handle_select,
+                                             params_select))
+     {
+          fprintf (stderr, "Failed to execute prepared statement SELECT\n");
+          return 1;
+     }
+
+
+     struct GNUNET_MY_ResultSpec results_select[] = {
+          GNUNET_MY_result_spec_rsa_public_key (&pub2),
+          GNUNET_MY_result_spec_rsa_signature (&sig2),
+          GNUNET_MY_result_spec_absolute_time (&abs_time2),
+          GNUNET_MY_result_spec_absolute_time (&forever2),
+          GNUNET_MY_result_spec_auto_from_type (&hc2),
+          GNUNET_MY_result_spec_variable_size (&msg2, &msg2_len),
+          GNUNET_MY_result_spec_uint16 (&u162),
+          GNUNET_MY_result_spec_uint32 (&u322),
+          GNUNET_MY_result_spec_uint64 (&u642),
+          GNUNET_MY_result_spec_end
+     };
+
+     ret = GNUNET_MY_extract_result (statements_handle_select,
+                                        NULL,
+                                        results_select,
+                                        0);
      if (GNUNET_OK != ret)
+     {
+          fprintf(stderr, "Failed to extract result\n");
           return 1;
+     }
+*/
      return 0;
 }
 
@@ -244,84 +186,67 @@ run_queries (MYSQL * mysql)
 int 
 main (int argc, const char * const argv[])
 {
-     
-     MYSQL *mysql;
-//     MYSQL_RES *result;
+     struct GNUNET_CONFIGURATION_Handle *config;
+     struct GNUNET_MYSQL_Context *context;
 
      int ret;
 
-     char *hote = "";
-     char *pseudo = "";
-     char *mdp = "";
-     char *database = "";
-
-     mysql_init (mysql);
-
-     mysql_options (mysql, 
-                    MYSQL_READ_DEFAULT_GROUP,
-                    NULL);
-
      GNUNET_log_setup (  "test-my",
                          "WARNING",
                          NULL);
 
-     if (! mysql_real_connect (mysql
-                              ,hote
-                              ,pseudo
-                              ,mdp,database
-                              ,0
-                              ,NULL
-                              ,0
-                              )) 
+     config = GNUNET_CONFIGURATION_create ();
+     if (NULL == config)
      {
-          fprintf(  stderr,
-                    "Cannot run test, database connection failed : %s\n",
-                    mysql_error (mysql));
-          GNUNET_break (0);
-
-          return 0;
+          fprintf (stderr, "Failed to create a configuration\n");
+          return 1;
      }
 
-     if (mysql_query (mysql,  "CREATE TABLE test_my("
-                              "pub INT"
-                              ", sig INT"
-                              ", abs_time BIGINT"
-                              ", forever BIGINT"
-                              ", hash INT"
-                              ", vsize VARCHAR"
-                              ", u16 SMALLINT"
-                              ", u32 INT"
-                              ", u64 BIGINT"
-                              ")"))
+     if (GNUNET_OK != GNUNET_CONFIGURATION_parse (config, "test_my.conf"))
      {
-          fprintf (stderr, 
-                    "Failed to create table : %s\n",
-                    mysql_error (mysql));
-
-          mysql_close (mysql);
+          fprintf (stderr, "Failed to parse configuaration\n");
           return 1;
      }
 
-     if (GNUNET_OK != 
-          mysql_prepare (mysql))
+     context = GNUNET_MYSQL_context_create (config, 
+                                             "datastore-mysql");
+     if (NULL == context)
      {
-          GNUNET_break (0) ;
-          mysql_close (mysql);
-          return 1;
+          fprintf(stderr, "Failed to connect to database\n");
+          return 77;
      }
 
-     ret = run_queries (mysql);
-
-     if (mysql_query (mysql, 
-                    "DROP TABLE test_my;"))
+     if (GNUNET_OK != GNUNET_MYSQL_statement_run (context,
+                                                  "CREATE TABLE test_my("
+                                                  "pub INT NOT NULL"
+                                                  ", sig INT NOT NULL"
+                                                  ", abs_time BIGINT NOT NULL"
+                                                  ", forever BIGINT NOT NULL"
+                                                  ", hash VARCHAR(32) NOT NULL CHECK(LENGTH(hash)=64)"
+                                                  ", vsize VARCHAR(32) NOT NULL"
+                                                  ", u16 SMALLINT NOT NULL"
+                                                  ", u32 INT NOT NULL"
+                                                  ", u64 BIGINT NOT NULL"
+                                                  ")"))
      {
-          fprintf (stderr, "Failed to drop table : %s\n",
-                         mysql_error (mysql));
-          mysql_close (mysql);
+          fprintf (stderr, 
+                    "Failed to create table \n"); 
+          GNUNET_MYSQL_statements_invalidate (context);    
+          GNUNET_MYSQL_context_destroy (context);
+          
           return 1;
      }
 
-     mysql_close (mysql);
+     ret = run_queries (context);
+
+/*     if(GNUNET_OK != GNUNET_MYSQL_statement_run (context,
+                                                  "DROP TABLE test_my"))
+     {
+          fprintf (stderr, "Failed to drop table test_my\n");
+          GNUNET_MYSQL_statements_invalidate (context);
+     }
+*/
+     GNUNET_MYSQL_context_destroy (context);
 
      return ret;
 }