common/tr_idp.c

   1 /*
   2  * Copyright (c) 2012, 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 <talloc.h>
  36 #include <time.h>
  37
  38 #include <trust_router/tr_name.h>
  39 #include <tr_idp.h>
  40 #include <tr_config.h>
  41 #include <tr_debug.h>
  42
  43 static int tr_aaa_server_destructor(void *obj)
  44 {
  45   TR_AAA_SERVER *aaa=talloc_get_type_abort(obj, TR_AAA_SERVER);
  46   if (aaa->hostname!=NULL)
  47     tr_free_name(aaa->hostname);
  48   return 0;
  49 }
  50
  51 TR_AAA_SERVER *tr_aaa_server_new(TALLOC_CTX *mem_ctx, TR_NAME *hostname)
  52 {
  53   TR_AAA_SERVER *aaa=talloc(mem_ctx, TR_AAA_SERVER);
  54   if (aaa!=NULL) {
  55     aaa->hostname=hostname;
  56     talloc_set_destructor((void *)aaa, tr_aaa_server_destructor);
  57   }
  58   return aaa;
  59 }
  60
  61 void tr_aaa_server_free(TR_AAA_SERVER *aaa)
  62 {
  63   talloc_free(aaa);
  64 }
  65
  66 TR_AAA_SERVER *tr_idp_aaa_server_lookup(TR_IDP_REALM *idp_realms, TR_NAME *idp_realm_name, TR_NAME *comm)
  67 {
  68   TR_IDP_REALM *idp = NULL;
  69
  70   for (idp = idp_realms; idp != NULL; idp = idp->next) {
  71     if (!tr_name_cmp(idp_realm_name, idp->realm_id)) {
  72       /* TBD -- check that the community is one of the APCs for the IDP */
  73       break;
  74     }
  75   }
  76   if (idp)
  77     return idp->aaa_servers;
  78   else
  79     return NULL;
  80 }
  81
  82 TR_AAA_SERVER *tr_default_server_lookup(TR_AAA_SERVER *default_servers, TR_NAME *comm)
  83 {
  84   if (!default_servers)
  85     return NULL;
  86
  87   return(default_servers);
  88 }
  89
  90 static int tr_idp_realm_destructor(void *obj)
  91 {
  92   TR_IDP_REALM *idp=talloc_get_type_abort(obj, TR_IDP_REALM);
  93   if (idp->realm_id!=NULL)
  94     tr_free_name(idp->realm_id);
  95   return 0;
  96 }
  97
  98 /* talloc note: lists of idp realms should be assembled using
  99  * tr_idp_realm_add(). This will put all of the elements in the
 100  * list, other than the head, as children of the head context.
 101  * The head can then be placed in whatever context is desired. */
 102 TR_IDP_REALM *tr_idp_realm_new(TALLOC_CTX *mem_ctx)
 103 {
 104   TR_IDP_REALM *idp=talloc(mem_ctx, TR_IDP_REALM);
 105   if (idp!=NULL) {
 106     idp->next=NULL;
 107     idp->comm_next=NULL;
 108     idp->realm_id=NULL;
 109     idp->shared_config=0;
 110     idp->aaa_servers=NULL;
 111     idp->apcs=NULL;
 112     idp->origin=TR_REALM_LOCAL;
 113     talloc_set_destructor((void *)idp, tr_idp_realm_destructor);
 114   }
 115   return idp;
 116 }
 117
 118 void tr_idp_realm_free(TR_IDP_REALM *idp)
 119 {
 120   talloc_free(idp);
 121 }
 122
 123 TR_NAME *tr_idp_realm_get_id(TR_IDP_REALM *idp)
 124 {
 125   if (idp==NULL)
 126     return NULL;
 127
 128   return idp->realm_id;
 129 }
 130
 131 TR_NAME *tr_idp_realm_dup_id(TR_IDP_REALM *idp)
 132 {
 133   if (idp==NULL)
 134     return NULL;
 135
 136   return tr_dup_name(tr_idp_realm_get_id(idp));
 137 }
 138
 139 void tr_idp_realm_set_id(TR_IDP_REALM *idp, TR_NAME *id)
 140 {
 141   if (idp->realm_id!=NULL)
 142     tr_free_name(idp->realm_id);
 143   idp->realm_id=id;
 144 }
 145
 146 void tr_idp_realm_set_apcs(TR_IDP_REALM *idp, TR_APC *apc)
 147 {
 148   if (idp->apcs!=NULL)
 149     tr_apc_free(idp->apcs);
 150   idp->apcs=apc;
 151   talloc_steal(idp, apc);
 152 }
 153
 154 TR_APC *tr_idp_realm_get_apcs(TR_IDP_REALM *idp)
 155 {
 156   return idp->apcs;
 157 }
 158
 159 TR_IDP_REALM *tr_idp_realm_lookup(TR_IDP_REALM *idp_realms, TR_NAME *idp_name)
 160 {
 161   TR_IDP_REALM *idp = NULL;
 162
 163   if (!idp_name) {
 164     tr_debug("tr_idp_realm_lookup: Bad parameters.");
 165     return NULL;
 166   }
 167
 168   for (idp=idp_realms; NULL!=idp; idp=idp->next) {
 169     if (0==tr_name_cmp(tr_idp_realm_get_id(idp), idp_name))
 170       return idp;
 171   }
 172   return NULL;
 173 }
 174
 175
 176 static TR_IDP_REALM *tr_idp_realm_tail(TR_IDP_REALM *idp)
 177 {
 178   if (idp==NULL)
 179     return NULL;
 180
 181   while (idp->next!=NULL)
 182     idp=idp->next;
 183   return idp;
 184 }
 185
 186 /* do not call directly, use the tr_idp_realm_add() macro */
 187 TR_IDP_REALM *tr_idp_realm_add_func(TR_IDP_REALM *head, TR_IDP_REALM *new)
 188 {
 189   if (head==NULL)
 190     head=new;
 191   else {
 192     tr_idp_realm_tail(head)->next=new;
 193     while (new!=NULL) {
 194       talloc_steal(head, new); /* put it in the right context */
 195       new=new->next;
 196     }
 197   }
 198   return head;
 199 }
 200
 201 /* use the macro */
 202 TR_IDP_REALM *tr_idp_realm_remove_func(TR_IDP_REALM *head, TR_IDP_REALM *remove)
 203 {
 204   TALLOC_CTX *list_ctx=talloc_parent(head);
 205   TR_IDP_REALM *this=NULL;
 206
 207   if (head==NULL)
 208     return NULL;
 209
 210   if (head==remove) {
 211     /* if we're removing the head, put the next element (if present) into the context
 212      * the list head was in. */
 213     head=head->next;
 214     if (head!=NULL) {
 215       talloc_steal(list_ctx, head);
 216       /* now put all the other elements in the context of the list head */
 217       for (this=head->next; this!=NULL; this=this->next)
 218         talloc_steal(head, this);
 219     }
 220   } else {
 221     /* not removing the head; no need to play with contexts */
 222     for (this=head; this->next!=NULL; this=this->next) {
 223       if (this->next==remove) {
 224         this->next=remove->next;
 225         break;
 226       }
 227     }
 228   }
 229   return head;
 230 }
 231
 232 static int tr_idp_realm_apc_count(TR_IDP_REALM *idp)
 233 {
 234   int ii=0;
 235   TR_APC *apc=idp->apcs;
 236   while (apc!=NULL) {
 237     apc=apc->next;
 238     ii++;
 239   }
 240   return ii;
 241 }
 242
 243 static int tr_idp_realm_aaa_server_count(TR_IDP_REALM *idp)
 244 {
 245   int ii=0;
 246   TR_AAA_SERVER *aaa=idp->aaa_servers;
 247   while (aaa!=NULL) {
 248     aaa=aaa->next;
 249     ii++;
 250   }
 251   return ii;
 252 }
 253
 254 static char *tr_aaa_server_to_str(TALLOC_CTX *mem_ctx, TR_AAA_SERVER *aaa)
 255 {
 256   return talloc_strndup(mem_ctx, aaa->hostname->buf, aaa->hostname->len);
 257 }
 258
 259 char *tr_idp_realm_to_str(TALLOC_CTX *mem_ctx, TR_IDP_REALM *idp)
 260 {
 261   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 262   char **s_aaa=NULL, *aaa_servers=NULL;
 263   char **s_apc=NULL, *apcs=NULL;
 264   int ii=0, aaa_servers_strlen=0, apcs_strlen=0;
 265   int n_aaa_servers=tr_idp_realm_aaa_server_count(idp);
 266   int n_apcs=tr_idp_realm_apc_count(idp);
 267   TR_AAA_SERVER *aaa=NULL;
 268   TR_APC *apc=NULL;
 269   char *result=NULL;
 270
 271   /* get the AAA servers */
 272   if (n_aaa_servers<=0)
 273     aaa_servers=talloc_strdup(tmp_ctx, "");
 274   else {
 275     s_aaa=talloc_array(tmp_ctx, char *, n_aaa_servers);
 276     for (aaa=idp->aaa_servers,ii=0; aaa!=NULL; aaa=aaa->next,ii++) {
 277       s_aaa[ii]=tr_aaa_server_to_str(s_aaa, aaa);
 278       aaa_servers_strlen+=strlen(s_aaa[ii]);
 279     }
 280
 281     /* add space for comma-space separators */
 282     aaa_servers_strlen+=2*(n_aaa_servers-1);
 283
 284     aaa_servers=talloc_array(tmp_ctx, char, aaa_servers_strlen+1);
 285     aaa_servers[0]='\0';
 286     for (ii=0; ii<n_aaa_servers; ii++) {
 287       strcat(aaa_servers, s_aaa[ii]);
 288       if (ii<(n_aaa_servers-1))
 289         strcat(aaa_servers, ", ");
 290     }
 291     talloc_free(s_aaa);
 292   }
 293
 294   /* get the APCs */
 295   if (n_apcs<=0)
 296     apcs=talloc_strdup(tmp_ctx, "");
 297   else {
 298     s_apc=talloc_array(tmp_ctx, char *, n_apcs);
 299     for (apc=idp->apcs,ii=0; apc!=NULL; apc=apc->next,ii++) {
 300       s_apc[ii]=tr_apc_to_str(s_apc, apc);
 301       apcs_strlen+=strlen(s_apc[ii]);
 302     }
 303
 304     /* add space for comma-space separators */
 305     apcs_strlen+=2*(n_apcs-1);
 306
 307     apcs=talloc_array(tmp_ctx, char, apcs_strlen+1);
 308     apcs[0]='\0';
 309     for (ii=0; ii<n_apcs; ii++) {
 310       strcat(apcs, s_apc[ii]);
 311       if (ii<(n_apcs-1))
 312         strcat(apcs, ", ");
 313     }
 314     talloc_free(s_apc);
 315   }
 316
 317   result=talloc_asprintf(mem_ctx,
 318                          "IDP realm: \"%.*s\"\n"
 319                          "  shared: %s\n"
 320                          "  local: %s\n"
 321                          "  AAA servers: %s\n"
 322                          "  APCs: %s\n",
 323                          idp->realm_id->len, idp->realm_id->buf,
 324                          (idp->shared_config)?"yes":"no",
 325                          (idp->origin==TR_REALM_LOCAL)?"yes":"no",
 326                          aaa_servers,
 327                          apcs);
 328   talloc_free(tmp_ctx);
 329   return result;
 330 }
 331
 332 void tr_idp_realm_incref(TR_IDP_REALM *realm)
 333 {
 334   realm->refcount++;
 335 }
 336
 337 void tr_idp_realm_decref(TR_IDP_REALM *realm)
 338 {
 339   if (realm->refcount>0)
 340     realm->refcount--;
 341 }
 342
 343 /* remove any with zero refcount
 344  * Call via macro. */
 345 TR_IDP_REALM *tr_idp_realm_sweep_func(TR_IDP_REALM *head)
 346 {
 347   TR_IDP_REALM *idp=NULL;
 348   TR_IDP_REALM *old_next=NULL;
 349
 350   if (head==NULL)
 351     return NULL;
 352
 353   while ((head!=NULL) && (head->refcount==0)) {
 354     idp=head; /* keep a pointer so we can remove it */
 355     tr_idp_realm_remove(head, idp); /* use this to get talloc contexts right */
 356     tr_idp_realm_free(idp);
 357   }
 358
 359   if (head==NULL)
 360     return NULL;
 361
 362   /* will not remove the head here, that has already been done */
 363   for (idp=head; idp->next!=NULL; idp=idp->next) {
 364     if (idp->next->refcount==0) {
 365       old_next=idp->next;
 366       tr_idp_realm_remove(head, idp->next); /* changes idp->next */
 367       tr_idp_realm_free(old_next);
 368     }
 369   }
 370
 371   return head;
 372 }
 373