common/tests/cfg_test.c

   1 /*
   2  * Copyright (c) 2016, JANET(UK)
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *
   9  * 1. Redistributions of source code must retain the above copyright
  10  *    notice, this list of conditions and the following disclaimer.
  11  *
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * 3. Neither the name of JANET(UK) nor the names of its contributors
  17  *    may be used to endorse or promote products derived from this software
  18  *    without specific prior written permission.
  19  *
  20  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  21  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  22  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  23  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  24  * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
  25  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  26  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  27  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  29  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  30  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  31  * OF THE POSSIBILITY OF SUCH DAMAGE.
  32  *
  33  */
  34
  35 #include <stdio.h>
  36 #include <talloc.h>
  37 #include <assert.h>
  38
  39 #include <trust_router/tr_name.h>
  40 #include <tr_comm.h>
  41 #include <tr_idp.h>
  42 #include <tr_config.h>
  43 #include <tr_debug.h>
  44
  45 static void tr_talloc_log(const char *msg)
  46 {
  47   tr_debug("talloc: %s", msg);
  48 }
  49
  50 static int verify_idp_cfg(TR_CFG *cfg)
  51 {
  52   TR_COMM *comm=NULL;
  53   TR_NAME *name=NULL;
  54   TR_IDP_REALM *idp_realm=NULL;
  55   TR_AAA_SERVER *aaa=NULL;
  56
  57   assert(cfg!=NULL);
  58
  59   /* test the comms attribute */
  60   assert(cfg->ctable!=NULL);
  61   name=tr_new_name("apc.example.com");
  62   comm=tr_comm_table_find_comm(cfg->ctable, name);
  63   tr_free_name(name);
  64   assert(comm!=NULL);
  65
  66   assert(comm->type==TR_COMM_APC);
  67   assert(comm->expiration_interval==TR_DEFAULT_APC_EXPIRATION_INTERVAL);
  68   assert(comm->apcs==NULL);
  69
  70   name=tr_new_name("A.idp.cfg");
  71   assert(name!=NULL);
  72   idp_realm=tr_comm_find_idp(cfg->ctable, comm, name);
  73   assert(idp_realm!=NULL);
  74   assert(idp_realm->shared_config==0);
  75   assert(idp_realm->origin==TR_REALM_LOCAL);
  76   tr_free_name(name);
  77   name=tr_new_name("apc.example.com");
  78   assert(tr_name_cmp(idp_realm->apcs->id, name)==0);
  79   tr_free_name(name);
  80
  81   assert(idp_realm->aaa_servers!=NULL);
  82   name=tr_new_name("rad1.A.idp.cfg");
  83   for (aaa=idp_realm->aaa_servers;
  84        (aaa!=NULL) && (tr_name_cmp(name, aaa->hostname)!=0);
  85        aaa=aaa->next) { }
  86   assert(aaa!=NULL);
  87   tr_free_name(name);
  88
  89   name=tr_new_name("rad2.A.idp.cfg");
  90   for (aaa=idp_realm->aaa_servers;
  91        (aaa!=NULL) && (tr_name_cmp(name, aaa->hostname)!=0);
  92        aaa=aaa->next) { }
  93   assert(aaa!=NULL);
  94   tr_free_name(name);
  95
  96   return 0;
  97 }
  98
  99 static int verify_rp_cfg(TR_CFG *cfg)
 100 {
 101   int ii=0;
 102   TR_NAME *name=NULL;
 103
 104   assert(cfg!=NULL);
 105   assert(cfg->rp_clients!=NULL);
 106   assert(cfg->rp_clients->next==NULL);
 107   assert(cfg->rp_clients->comm_next==NULL);
 108
 109   assert(cfg->rp_clients->gss_names!=NULL);
 110   for (ii=1; ii<TR_MAX_GSS_NAMES; ii++)
 111     assert(cfg->rp_clients->gss_names->names[ii]==NULL);
 112   assert(cfg->rp_clients->gss_names->names[0]!=NULL);
 113   name=tr_new_name("gss@example.com");
 114   assert(tr_name_cmp(name, cfg->rp_clients->gss_names->names[0])==0);
 115   return 0;
 116 }
 117
 118 int main(void)
 119 {
 120   TALLOC_CTX *mem_ctx=talloc_new(NULL);
 121   TR_CFG *cfg=NULL;
 122   TR_CFG_RC rc=TR_CFG_ERROR;
 123
 124   tr_log_open();
 125
 126   talloc_set_log_fn(tr_talloc_log);
 127   cfg=tr_cfg_new(mem_ctx);
 128
 129   printf("Parsing idp.cfg.\n");
 130   rc=tr_cfg_parse_one_config_file(cfg, "idp.cfg");
 131   switch(rc) {
 132   case TR_CFG_SUCCESS:
 133     tr_debug("main: TR_CFG_SUCCESS");
 134     break;
 135   case TR_CFG_ERROR:
 136     tr_debug("main: TR_CFG_ERROR");
 137     break;
 138   case TR_CFG_BAD_PARAMS:
 139     tr_debug("main: TR_CFG_BAD_PARAMS");
 140     break;
 141   case TR_CFG_NOPARSE:
 142     tr_debug("main: TR_CFG_NOPARSE");
 143     break;
 144   case TR_CFG_NOMEM:
 145     tr_debug("main: TR_CFG_NOMEM");
 146     break;
 147   }
 148
 149   printf("Verifying IDP parse results... ");
 150   if (verify_idp_cfg(cfg)!=0) {
 151     printf("Error!\n");
 152     exit(-1);
 153   }
 154   printf("success!\n");
 155
 156   printf("Verifying RP parse results... ");
 157   if (verify_rp_cfg(cfg)!=0) {
 158     printf("Error!\n");
 159     exit(-1);
 160   }
 161   printf("success!\n");
 162
 163   talloc_report_full(mem_ctx, stderr);
 164   tr_cfg_free(cfg);
 165
 166   printf("Cleared configuration for next test.\n\n");
 167
 168   cfg=tr_cfg_new(mem_ctx);
 169
 170   printf("Parsing rp.cfg.\n");
 171   rc=tr_cfg_parse_one_config_file(cfg, "rp.cfg");
 172   switch(rc) {
 173   case TR_CFG_SUCCESS:
 174     tr_debug("main: TR_CFG_SUCCESS");
 175     break;
 176   case TR_CFG_ERROR:
 177     tr_debug("main: TR_CFG_ERROR");
 178     break;
 179   case TR_CFG_BAD_PARAMS:
 180     tr_debug("main: TR_CFG_BAD_PARAMS");
 181     break;
 182   case TR_CFG_NOPARSE:
 183     tr_debug("main: TR_CFG_NOPARSE");
 184     break;
 185   case TR_CFG_NOMEM:
 186     tr_debug("main: TR_CFG_NOMEM");
 187     break;
 188   }
 189
 190 #if 0
 191   printf("Verifying RP parse results... ");
 192   if (verify_rp_cfg(cfg)!=0) {
 193     printf("Error!\n");
 194     exit(-1);
 195   }
 196   printf("success!\n");
 197 #endif
 198
 199   talloc_free(mem_ctx);
 200   return 0;
 201 }