tests/virauthconfigtest.c

   1 /*
   2  * Copyright (C) 2012, 2014 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 #include <signal.h>
  22
  23 #include "testutils.h"
  24 #include "virerror.h"
  25 #include "viralloc.h"
  26 #include "virlog.h"
  27
  28 #include "virauthconfig.h"
  29
  30 #define VIR_FROM_THIS VIR_FROM_RPC
  31
  32 VIR_LOG_INIT("tests.authconfigtest");
  33
  34 struct ConfigLookupData {
  35     virAuthConfigPtr config;
  36     const char *hostname;
  37     const char *service;
  38     const char *credname;
  39     const char *expect;
  40 };
  41
  42 static int testAuthLookup(const void *args)
  43 {
  44     int ret = -1;
  45     const struct ConfigLookupData *data = args;
  46     const char *actual = NULL;
  47     int rv;
  48
  49     rv = virAuthConfigLookup(data->config,
  50                              data->service,
  51                              data->hostname,
  52                              data->credname,
  53                              &actual);
  54
  55     if (rv < 0)
  56         goto cleanup;
  57
  58     if (data->expect) {
  59         if (!actual ||
  60             STRNEQ(actual, data->expect)) {
  61             VIR_WARN("Expected value '%s' for '%s' '%s' '%s', but got '%s'",
  62                      data->expect, data->hostname,
  63                      data->service, data->credname,
  64                      NULLSTR(actual));
  65             goto cleanup;
  66         }
  67     } else {
  68         if (actual) {
  69             VIR_WARN("Did not expect a value for '%s' '%s' '%s', but got '%s'",
  70                      data->hostname,
  71                      data->service, data->credname,
  72                      actual);
  73             goto cleanup;
  74         }
  75     }
  76
  77     ret = 0;
  78  cleanup:
  79     return ret;
  80 }
  81
  82
  83 static int
  84 mymain(void)
  85 {
  86     int ret = 0;
  87
  88     virAuthConfigPtr config;
  89
  90     signal(SIGPIPE, SIG_IGN);
  91
  92 #define TEST_LOOKUP(config, hostname, service, credname, expect) \
  93     do  { \
  94         const struct ConfigLookupData data = { \
  95             config, hostname, service, credname, expect \
  96         }; \
  97         if (virTestRun("Test Lookup " hostname "-" service "-" credname, \
  98                         testAuthLookup, &data) < 0) \
  99             ret = -1; \
 100     } while (0)
 101
 102     const char *confdata =
 103         "[credentials-test]\n"
 104         "username=fred\n"
 105         "password=123456\n"
 106         "\n"
 107         "[credentials-prod]\n"
 108         "username=bar\n"
 109         "password=letmein\n"
 110         "\n"
 111         "[auth-libvirt-test1.example.com]\n"
 112         "credentials=test\n"
 113         "\n"
 114         "[auth-libvirt-test2.example.com]\n"
 115         "credentials=test\n"
 116         "\n"
 117         "[auth-libvirt-demo3.example.com]\n"
 118         "credentials=test\n"
 119         "\n"
 120         "[auth-libvirt-prod1.example.com]\n"
 121         "credentials=prod\n";
 122
 123     if (!(config = virAuthConfigNewData("auth.conf", confdata, strlen(confdata))))
 124         return EXIT_FAILURE;
 125
 126     TEST_LOOKUP(config, "test1.example.com", "libvirt", "username", "fred");
 127     TEST_LOOKUP(config, "test1.example.com", "vnc", "username", NULL);
 128     TEST_LOOKUP(config, "test1.example.com", "libvirt", "realm", NULL);
 129     TEST_LOOKUP(config, "test66.example.com", "libvirt", "username", NULL);
 130     TEST_LOOKUP(config, "prod1.example.com", "libvirt", "username", "bar");
 131     TEST_LOOKUP(config, "prod1.example.com", "libvirt", "password", "letmein");
 132
 133     virAuthConfigFree(config);
 134
 135     return ret == 0 ? EXIT_SUCCESS : EXIT_FAILURE;
 136 }
 137
 138 VIR_TEST_MAIN(mymain)