src/my/test_my.c

   1 /*
   2      This file is part of GNUnet
   3      Copyright (C) 2016 GNUnet e.V.
   4
   5      GNUnet is free software: you can redistribute it and/or modify it
   6      under the terms of the GNU Affero General Public License as published
   7      by the Free Software Foundation, either version 3 of the License,
   8      or (at your option) any later version.
   9
  10      GNUnet is distributed in the hope that it will be useful, but
  11      WITHOUT ANY WARRANTY; without even the implied warranty of
  12      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13      Affero General Public License for more details.
  14
  15      You should have received a copy of the GNU Affero General Public License
  16      along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17
  18      SPDX-License-Identifier: AGPL3.0-or-later
  19  */
  20 /**
  21  * @file my/test_my.c
  22  * @brief Tests for convenience MySQL database
  23  * @author Christophe Genevey
  24  */
  25 #include "platform.h"
  26 #include <mysql/mysql.h>
  27 #include "gnunet_my_lib.h"
  28 #include "gnunet_mysql_lib.h"
  29 #include "gnunet_util_lib.h"
  30
  31
  32 /**
  33  * Run actual test queries.
  34  *
  35  * @param contexte the current context of mysql
  36  * @return 0 on succes
  37  */
  38 static int
  39 run_queries(struct GNUNET_MYSQL_Context *context)
  40 {
  41   struct GNUNET_CRYPTO_RsaPublicKey *pub = NULL;
  42   struct GNUNET_CRYPTO_RsaPublicKey *pub2 = NULL;
  43   struct GNUNET_CRYPTO_RsaSignature *sig = NULL;;
  44   struct GNUNET_CRYPTO_RsaSignature *sig2 = NULL;
  45   struct GNUNET_TIME_Absolute abs_time = GNUNET_TIME_absolute_get();
  46   struct GNUNET_TIME_Absolute abs_time2;
  47   struct GNUNET_TIME_Absolute forever = GNUNET_TIME_UNIT_FOREVER_ABS;
  48   struct GNUNET_TIME_Absolute forever2;
  49   const struct GNUNET_TIME_AbsoluteNBO abs_time_nbo = GNUNET_TIME_absolute_hton(abs_time);
  50   struct GNUNET_HashCode hc;
  51   struct GNUNET_HashCode hc2;
  52   const char msg[] = "hello";
  53   void *msg2 = NULL;
  54   size_t msg2_len;
  55
  56   const char msg3[] = "world";
  57   char *msg4 = "";
  58
  59   uint16_t u16;
  60   uint16_t u162;
  61   uint32_t u32;
  62   uint32_t u322;
  63   uint64_t u64;
  64   uint64_t u642;
  65
  66   int ret;
  67
  68   struct GNUNET_MYSQL_StatementHandle *statements_handle_insert = NULL;
  69   struct GNUNET_MYSQL_StatementHandle *statements_handle_select = NULL;
  70
  71   struct GNUNET_CRYPTO_RsaPrivateKey *priv = NULL;
  72   struct GNUNET_HashCode hmsg;
  73
  74   priv = GNUNET_CRYPTO_rsa_private_key_create(1024);
  75   pub = GNUNET_CRYPTO_rsa_private_key_get_public(priv);
  76   memset(&hmsg, 42, sizeof(hmsg));
  77   sig = GNUNET_CRYPTO_rsa_sign_fdh(priv,
  78                                    &hmsg);
  79   u16 = 16;
  80   u32 = 32;
  81   u64 = UINT64_MAX;
  82
  83   memset(&hc, 0, sizeof(hc));
  84   memset(&hc2, 0, sizeof(hc2));
  85
  86   statements_handle_insert
  87     = GNUNET_MYSQL_statement_prepare(context,
  88                                      "INSERT INTO test_my2 ("
  89                                      " pub"
  90                                      ",sig"
  91                                      ",abs_time"
  92                                      ",forever"
  93                                      ",abs_time_nbo"
  94                                      ",hash"
  95                                      ",vsize"
  96                                      ",str"
  97                                      ",u16"
  98                                      ",u32"
  99                                      ",u64"
 100                                      ") VALUES "
 101                                      "( ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)");
 102
 103   if (NULL == statements_handle_insert)
 104     {
 105       fprintf(stderr, "Failed to prepared statement INSERT\n");
 106       GNUNET_CRYPTO_rsa_signature_free(sig);
 107       GNUNET_CRYPTO_rsa_private_key_free(priv);
 108       GNUNET_CRYPTO_rsa_public_key_free(pub);
 109       return 1;
 110     }
 111
 112   struct GNUNET_MY_QueryParam params_insert[] = {
 113     GNUNET_MY_query_param_rsa_public_key(pub),
 114     GNUNET_MY_query_param_rsa_signature(sig),
 115     GNUNET_MY_query_param_absolute_time(&abs_time),
 116     GNUNET_MY_query_param_absolute_time(&forever),
 117     GNUNET_MY_query_param_absolute_time_nbo(&abs_time_nbo),
 118     GNUNET_MY_query_param_auto_from_type(&hc),
 119     GNUNET_MY_query_param_fixed_size(msg, strlen(msg)),
 120     GNUNET_MY_query_param_string(msg3),
 121     GNUNET_MY_query_param_uint16(&u16),
 122     GNUNET_MY_query_param_uint32(&u32),
 123     GNUNET_MY_query_param_uint64(&u64),
 124     GNUNET_MY_query_param_end
 125   };
 126
 127   if (GNUNET_OK != GNUNET_MY_exec_prepared(context,
 128                                            statements_handle_insert,
 129                                            params_insert))
 130     {
 131       fprintf(stderr, "Failed to execute prepared statement INSERT\n");
 132       GNUNET_CRYPTO_rsa_signature_free(sig);
 133       GNUNET_CRYPTO_rsa_private_key_free(priv);
 134       GNUNET_CRYPTO_rsa_public_key_free(pub);
 135       return 1;
 136     }
 137
 138   statements_handle_select
 139     = GNUNET_MYSQL_statement_prepare(context,
 140                                      "SELECT"
 141                                      " pub"
 142                                      ",sig"
 143                                      ",abs_time"
 144                                      ",forever"
 145                                      ",hash"
 146                                      ",vsize"
 147                                      ",str"
 148                                      ",u16"
 149                                      ",u32"
 150                                      ",u64"
 151                                      " FROM test_my2");
 152
 153   if (NULL == statements_handle_select)
 154     {
 155       fprintf(stderr, "Failed to prepared statement SELECT\n");
 156       GNUNET_CRYPTO_rsa_signature_free(sig);
 157       GNUNET_CRYPTO_rsa_private_key_free(priv);
 158       GNUNET_CRYPTO_rsa_public_key_free(pub);
 159       return 1;
 160     }
 161
 162   struct GNUNET_MY_QueryParam params_select[] = {
 163     GNUNET_MY_query_param_end
 164   };
 165
 166   if (GNUNET_OK != GNUNET_MY_exec_prepared(context,
 167                                            statements_handle_select,
 168                                            params_select))
 169     {
 170       fprintf(stderr, "Failed to execute prepared statement SELECT\n");
 171       GNUNET_CRYPTO_rsa_signature_free(sig);
 172       GNUNET_CRYPTO_rsa_private_key_free(priv);
 173       GNUNET_CRYPTO_rsa_public_key_free(pub);
 174       return 1;
 175     }
 176
 177   struct GNUNET_MY_ResultSpec results_select[] = {
 178     GNUNET_MY_result_spec_rsa_public_key(&pub2),
 179     GNUNET_MY_result_spec_rsa_signature(&sig2),
 180     GNUNET_MY_result_spec_absolute_time(&abs_time2),
 181     GNUNET_MY_result_spec_absolute_time(&forever2),
 182     GNUNET_MY_result_spec_auto_from_type(&hc2),
 183     GNUNET_MY_result_spec_variable_size(&msg2, &msg2_len),
 184     GNUNET_MY_result_spec_string(&msg4),
 185     GNUNET_MY_result_spec_uint16(&u162),
 186     GNUNET_MY_result_spec_uint32(&u322),
 187     GNUNET_MY_result_spec_uint64(&u642),
 188     GNUNET_MY_result_spec_end
 189   };
 190
 191   ret = GNUNET_MY_extract_result(statements_handle_select,
 192                                  results_select);
 193
 194   GNUNET_assert(GNUNET_YES == ret);
 195   GNUNET_break(abs_time.abs_value_us == abs_time2.abs_value_us);
 196   GNUNET_break(forever.abs_value_us == forever2.abs_value_us);
 197   GNUNET_break(0 ==
 198                memcmp(&hc,
 199                       &hc2,
 200                       sizeof(struct GNUNET_HashCode)));
 201
 202   GNUNET_assert(NULL != sig2);
 203   GNUNET_assert(NULL != pub2);
 204   GNUNET_break(0 ==
 205                GNUNET_CRYPTO_rsa_signature_cmp(sig,
 206                                                sig2));
 207   GNUNET_break(0 ==
 208                GNUNET_CRYPTO_rsa_public_key_cmp(pub,
 209                                                 pub2));
 210
 211   GNUNET_break(strlen(msg) == msg2_len);
 212   GNUNET_break(0 ==
 213                strncmp(msg,
 214                        msg2,
 215                        msg2_len));
 216
 217   GNUNET_break(strlen(msg3) == strlen(msg4));
 218   GNUNET_break(0 ==
 219                strcmp(msg3,
 220                       msg4));
 221
 222   GNUNET_break(16 == u162);
 223   GNUNET_break(32 == u322);
 224   GNUNET_break(UINT64_MAX == u642);
 225
 226   GNUNET_MY_cleanup_result(results_select);
 227
 228   GNUNET_CRYPTO_rsa_signature_free(sig);
 229   GNUNET_CRYPTO_rsa_private_key_free(priv);
 230   GNUNET_CRYPTO_rsa_public_key_free(pub);
 231
 232   if (GNUNET_OK != ret)
 233     return 1;
 234
 235   return 0;
 236 }
 237
 238
 239 int
 240 main(int argc, const char *const argv[])
 241 {
 242   struct GNUNET_CONFIGURATION_Handle *config;
 243   struct GNUNET_MYSQL_Context *context;
 244   int ret;
 245
 246   GNUNET_log_setup("test-my",
 247                    "WARNING",
 248                    NULL);
 249
 250   config = GNUNET_CONFIGURATION_create();
 251   if (GNUNET_OK !=
 252       GNUNET_CONFIGURATION_parse(config, "test_my.conf"))
 253     {
 254       fprintf(stderr, "Failed to parse configuaration\n");
 255       return 1;
 256     }
 257
 258   context = GNUNET_MYSQL_context_create(config,
 259                                         "datastore-mysql");
 260   if (NULL == context)
 261     {
 262       fprintf(stderr, "Failed to connect to database\n");
 263       return 77;
 264     }
 265
 266   (void)GNUNET_MYSQL_statement_run(context,
 267                                    "DROP TABLE test_my2;");
 268
 269   if (GNUNET_OK !=
 270       GNUNET_MYSQL_statement_run(context,
 271                                  "CREATE TABLE IF NOT EXISTS test_my2("
 272                                  " pub BLOB NOT NULL"
 273                                  ",sig BLOB NOT NULL"
 274                                  ",abs_time BIGINT NOT NULL"
 275                                  ",forever BIGINT NOT NULL"
 276                                  ",abs_time_nbo BIGINT NOT NULL"
 277                                  ",hash BLOB NOT NULL CHECK(LENGTH(hash)=64)"
 278                                  ",vsize BLOB NOT NULL"
 279                                  ",str BLOB NOT NULL"
 280                                  ",u16 SMALLINT NOT NULL"
 281                                  ",u32 INT NOT NULL"
 282                                  ",u64 BIGINT NOT NULL"
 283                                  ")"))
 284     {
 285       fprintf(stderr,
 286               "Failed to create table. Database likely not setup correctly.\n");
 287       GNUNET_MYSQL_statements_invalidate(context);
 288       GNUNET_MYSQL_context_destroy(context);
 289
 290       return 77;
 291     }
 292
 293   ret = run_queries(context);
 294
 295   GNUNET_MYSQL_context_destroy(context);
 296   GNUNET_free(config);
 297
 298   return ret;
 299 }