tests/virrandommock.c

   1 /*
   2  * Copyright (C) 2016 Red Hat, Inc.
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library.  If not, see
  16  * <http://www.gnu.org/licenses/>.
  17  */
  18
  19 #include <config.h>
  20
  21 #ifndef WIN32
  22
  23 # include <gnutls/gnutls.h>
  24
  25 # include "internal.h"
  26 # include "virstring.h"
  27 # include "virrandom.h"
  28 # include "virmock.h"
  29
  30 # define VIR_FROM_THIS VIR_FROM_NONE
  31
  32 int
  33 virRandomBytes(unsigned char *buf,
  34                size_t buflen)
  35 {
  36     size_t i;
  37
  38     for (i = 0; i < buflen; i++)
  39         buf[i] = i;
  40
  41     return 0;
  42 }
  43
  44 uint64_t virRandomBits(int nbits)
  45 {
  46     /* Chosen by a fair roll of a 2^64 sided dice */
  47     uint64_t ret = 0x0706050403020100;
  48     if (nbits < 64)
  49         ret &= ((1ULL << nbits) - 1);
  50     return ret;
  51 }
  52
  53 int virRandomGenerateWWN(char **wwn,
  54                          const char *virt_type ATTRIBUTE_UNUSED)
  55 {
  56     return virAsprintf(wwn, "5100000%09llx",
  57                        (unsigned long long)virRandomBits(36));
  58 }
  59
  60
  61 static int (*real_gnutls_dh_params_generate2)(gnutls_dh_params_t dparams,
  62                                               unsigned int bits);
  63
  64 static gnutls_dh_params_t params_cache;
  65 static unsigned int cachebits;
  66
  67 int
  68 gnutls_dh_params_generate2(gnutls_dh_params_t dparams,
  69                            unsigned int bits)
  70 {
  71     int rc = 0;
  72
  73     VIR_MOCK_REAL_INIT(gnutls_dh_params_generate2);
  74
  75     if (!params_cache) {
  76         if (gnutls_dh_params_init(&params_cache) < 0) {
  77             fprintf(stderr, "Error initializing params cache");
  78             abort();
  79         }
  80         rc = real_gnutls_dh_params_generate2(params_cache, bits);
  81
  82         if (rc < 0)
  83             return rc;
  84         cachebits = bits;
  85     }
  86
  87     if (cachebits != bits) {
  88         fprintf(stderr, "Requested bits do not match the cached value");
  89         abort();
  90     }
  91
  92     return gnutls_dh_params_cpy(dparams, params_cache);
  93 }
  94 #else /* WIN32 */
  95 /* Can't mock on WIN32 */
  96 #endif