common/tr_comm.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 <jansson.h>
  36 #include <talloc.h>
  37 #include <sys/time.h>
  38
  39 #include <tr_rp.h>
  40 #include <tr_idp.h>
  41 #include <trust_router/tr_name.h>
  42 #include <tr_comm.h>
  43 #include <tr_debug.h>
  44
  45
  46 /* static prototypes */
  47 static TR_NAME *tr_comm_memb_get_realm_id(TR_COMM_MEMB *memb);
  48
  49
  50 static int tr_comm_destructor(void *obj)
  51 {
  52   TR_COMM *comm=talloc_get_type_abort(obj, TR_COMM);
  53   if (comm->id!=NULL)
  54     tr_free_name(comm->id);
  55   if (comm->owner_realm!=NULL)
  56     tr_free_name(comm->owner_realm);
  57   if (comm->owner_contact!=NULL)
  58     tr_free_name(comm->owner_contact);
  59   return 0;
  60 }
  61
  62 TR_COMM *tr_comm_new(TALLOC_CTX *mem_ctx)
  63 {
  64   TR_COMM *comm=talloc(mem_ctx, TR_COMM);
  65   if (comm!=NULL) {
  66     comm->next=NULL;
  67     comm->id=NULL;
  68     comm->type=TR_COMM_UNKNOWN;
  69     comm->apcs=NULL;
  70     comm->owner_realm=NULL;
  71     comm->owner_contact=NULL;
  72     comm->expiration_interval=0;
  73     comm->refcount=0;
  74     talloc_set_destructor((void *)comm, tr_comm_destructor);
  75   }
  76   return comm;
  77 }
  78
  79 void tr_comm_free(TR_COMM *comm)
  80 {
  81   talloc_free(comm);
  82 }
  83
  84 void tr_comm_set_id(TR_COMM *comm, TR_NAME *id)
  85 {
  86   if (comm->id != NULL)
  87     tr_free_name(comm->id);
  88   comm->id=id;
  89 }
  90
  91 void tr_comm_incref(TR_COMM *comm)
  92 {
  93   comm->refcount++;
  94 }
  95
  96 void tr_comm_decref(TR_COMM *comm)
  97 {
  98   if (comm->refcount>0)
  99     comm->refcount--;
 100 }
 101
 102 void tr_comm_set_apcs(TR_COMM *comm, TR_APC *apc)
 103 {
 104   if (comm->apcs!=NULL)
 105     tr_apc_free(comm->apcs);
 106   comm->apcs=apc;
 107   talloc_steal(comm, apc);
 108 }
 109
 110 TR_APC *tr_comm_get_apcs(TR_COMM *comm)
 111 {
 112   return comm->apcs;
 113 }
 114
 115 TR_NAME *tr_comm_get_id(TR_COMM *comm)
 116 {
 117   return comm->id;
 118 }
 119
 120 TR_NAME *tr_comm_dup_id(TR_COMM *comm)
 121 {
 122   return tr_dup_name(comm->id);
 123 }
 124
 125 void tr_comm_set_type(TR_COMM *comm, TR_COMM_TYPE type)
 126 {
 127   comm->type=type;
 128 }
 129
 130 TR_COMM_TYPE tr_comm_get_type(TR_COMM *comm)
 131 {
 132   return comm->type;
 133 }
 134
 135 void tr_comm_set_owner_realm(TR_COMM *comm, TR_NAME *realm)
 136 {
 137   if (comm->owner_realm!=NULL)
 138     tr_free_name(comm->owner_realm);
 139   comm->owner_realm=realm;
 140 }
 141
 142 TR_NAME *tr_comm_get_owner_realm(TR_COMM *comm)
 143 {
 144   return comm->owner_realm;
 145 }
 146
 147 TR_NAME *tr_comm_dup_owner_realm(TR_COMM *comm)
 148 {
 149   return tr_dup_name(comm->owner_realm);
 150 }
 151
 152 void tr_comm_set_owner_contact(TR_COMM *comm, TR_NAME *contact)
 153 {
 154   if (comm->owner_contact != NULL)
 155     tr_free_name(comm->owner_contact);
 156   comm->owner_contact=contact;
 157 }
 158
 159 TR_NAME *tr_comm_get_owner_contact(TR_COMM *comm)
 160 {
 161   return comm->owner_contact;
 162 }
 163
 164 TR_NAME *tr_comm_dup_owner_contact(TR_COMM *comm)
 165 {
 166   return tr_dup_name(comm->owner_contact);
 167 }
 168
 169 unsigned int tr_comm_get_refcount(TR_COMM *comm)
 170 {
 171   return comm->refcount;
 172 }
 173
 174 /* add to the table if it's a new membership or has a shorter
 175  * provenance list than our existing membership */
 176 static void tr_comm_add_if_shorter(TR_COMM_TABLE *ctab, TR_COMM_MEMB *existing, TR_COMM_MEMB *newmemb)
 177 {
 178   if (existing==NULL) {
 179     /* not in the table */
 180     tr_comm_table_add_memb(ctab, newmemb);
 181   } else {
 182     /* Had an entry. Replace if we have shorter provenance. */
 183     if (tr_comm_memb_provenance_len(newmemb) < tr_comm_memb_provenance_len(existing)) {
 184       tr_comm_table_remove_memb(ctab, existing);
 185       tr_comm_memb_free(existing);
 186       tr_comm_table_add_memb(ctab, newmemb);
 187     }
 188   }
 189 }
 190
 191 /* does not take responsibility for freeing IDP realm */
 192 void tr_comm_add_idp_realm(TR_COMM_TABLE *ctab,
 193                            TR_COMM *comm,
 194                            TR_IDP_REALM *realm,
 195                            json_t *provenance,
 196                            struct timespec *expiry)
 197 {
 198   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 199   TR_COMM_MEMB *newmemb=tr_comm_memb_new(tmp_ctx);
 200   TR_COMM_MEMB *existing=NULL;
 201
 202   if (newmemb==NULL) {
 203     tr_err("tr_comm_add_idp_realm: unable to allocate new membership record.");
 204     talloc_free(tmp_ctx);
 205     return;
 206   }
 207
 208   tr_comm_memb_set_idp_realm(newmemb, realm);
 209   tr_comm_memb_set_comm(newmemb, comm);
 210   tr_comm_memb_set_provenance(newmemb, provenance);
 211   tr_comm_memb_set_expiry(newmemb, expiry);
 212
 213   existing=tr_comm_table_find_idp_memb_origin(ctab,
 214                                               tr_idp_realm_get_id(realm),
 215                                               tr_comm_get_id(comm),
 216                                               tr_comm_memb_get_origin(newmemb));
 217   tr_comm_add_if_shorter(ctab, existing, newmemb); /* takes newmemb out of tmp_ctx if needed */
 218   talloc_free(tmp_ctx);
 219 }
 220
 221 /* does not take responsibility for freeing RP realm */
 222 void tr_comm_add_rp_realm(TR_COMM_TABLE *ctab,
 223                           TR_COMM *comm,
 224                           TR_RP_REALM *realm,
 225                           json_t *provenance,
 226                           struct timespec *expiry)
 227 {
 228   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 229   TR_COMM_MEMB *newmemb=tr_comm_memb_new(tmp_ctx);
 230   TR_COMM_MEMB *existing=NULL;
 231
 232   if (newmemb==NULL) {
 233     tr_err("tr_comm_add_idp_realm: unable to allocate new membership record.");
 234     talloc_free(tmp_ctx);
 235     return;
 236   }
 237
 238   tr_comm_memb_set_rp_realm(newmemb, realm);
 239   tr_comm_memb_set_comm(newmemb, comm);
 240   tr_comm_memb_set_provenance(newmemb, provenance);
 241   tr_comm_memb_set_expiry(newmemb, expiry);
 242
 243   existing=tr_comm_table_find_rp_memb_origin(ctab,
 244                                              tr_rp_realm_get_id(realm),
 245                                              tr_comm_get_id(comm),
 246                                              tr_comm_memb_get_origin(newmemb));
 247   tr_comm_add_if_shorter(ctab, existing, newmemb); /* takes newmemb out of tmp_ctx if needed */
 248   talloc_free(tmp_ctx);
 249 }
 250
 251 static TR_COMM *tr_comm_tail(TR_COMM *comm)
 252 {
 253   if (comm==NULL)
 254     return comm;
 255
 256   while (comm->next!=NULL)
 257     comm=comm->next;
 258   return comm;
 259 }
 260
 261 /* All list members are in the talloc context of the head.
 262  * This will require careful thought if entries are ever removed
 263  * Call like comms=tr_comm_add_func(comms, new_comm);
 264  * or just use the tr_comm_add(comms, new) macro. */
 265 #define tr_comm_add(comms, new) ((comms)=tr_comm_add_func((comms), (new)))
 266 static TR_COMM *tr_comm_add_func(TR_COMM *comms, TR_COMM *new)
 267 {
 268   if (comms==NULL)
 269     comms=new;
 270   else {
 271     tr_comm_tail(comms)->next=new;
 272     while(new!=NULL) {
 273       talloc_steal(comms, new);
 274       new=new->next;
 275     }
 276   }
 277   return comms;
 278 }
 279
 280 /* Guarantees comm is not in the list, not an error if it was't there.
 281  * Does not free the removed element, nor change its talloc context. */
 282 #define tr_comm_remove(comms, c) ((comms)=tr_comm_remove_func((comms), (c)))
 283 static TR_COMM *tr_comm_remove_func(TR_COMM *comms, TR_COMM *remove)
 284 {
 285   TALLOC_CTX *list_ctx=talloc_parent(comms); /* in case we need to remove the head */
 286   TR_COMM *this=NULL;
 287
 288   if (comms==NULL)
 289     return NULL;
 290
 291   if (comms==remove) {
 292     /* if we're removing the head, put the next element (if present) into the context
 293      * the list head was in. */
 294     comms=comms->next;
 295     if (comms!=NULL) {
 296       talloc_steal(list_ctx, comms->next);
 297       /* now put all the other elements in the context of the list head */
 298       for (this=comms->next; this!=NULL; this=this->next)
 299         talloc_steal(comms, this);
 300     }
 301   } else {
 302     /* not removing the head; no need to play with contexts */
 303     for (this=comms; this->next!=NULL; this=this->next) {
 304       if (this->next==remove)
 305         this->next=remove->next;
 306     }
 307   }
 308   return comms;
 309 }
 310
 311 TR_IDP_REALM *tr_comm_find_idp(TR_COMM_TABLE *ctab, TR_COMM *comm, TR_NAME *idp_realm)
 312 {
 313   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 314   TR_COMM_ITER *iter=NULL;
 315   TR_IDP_REALM *this_idp=NULL;
 316
 317   if ((NULL==ctab) || (NULL==comm) || (NULL==idp_realm)) {
 318     talloc_free(tmp_ctx);
 319     return NULL;
 320   }
 321
 322   iter=tr_comm_iter_new(tmp_ctx);
 323   for (this_idp=tr_idp_realm_iter_first(iter, ctab, tr_comm_get_id(comm));
 324        this_idp!=NULL;
 325        this_idp=tr_idp_realm_iter_next(iter)) {
 326     if (0==tr_name_cmp(idp_realm, tr_idp_realm_get_id(this_idp))) {
 327       tr_debug("tr_comm_find_idp: Found IdP %s in community %s.", idp_realm->buf, tr_comm_get_id(comm)->buf);
 328       talloc_free(tmp_ctx);
 329       return this_idp;
 330     }
 331   }
 332   tr_debug("tr_comm_find_idp: Unable to find IdP %s in community %s.", idp_realm->buf, tr_comm_get_id(comm)->buf);
 333   talloc_free(tmp_ctx);
 334   return NULL;
 335 }
 336
 337 TR_RP_REALM *tr_comm_find_rp (TR_COMM_TABLE *ctab, TR_COMM *comm, TR_NAME *rp_realm)
 338 {
 339   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 340   TR_COMM_ITER *iter=NULL;
 341   TR_RP_REALM *this_rp=NULL;
 342
 343   if ((NULL==ctab) || (NULL==comm) || (NULL==rp_realm)) {
 344     talloc_free(tmp_ctx);
 345     return NULL;
 346   }
 347
 348   iter=tr_comm_iter_new(tmp_ctx);
 349   for (this_rp=tr_rp_realm_iter_first(iter, ctab, tr_comm_get_id(comm));
 350        this_rp!=NULL;
 351        this_rp=tr_rp_realm_iter_next(iter)) {
 352     if (0==tr_name_cmp(rp_realm, tr_rp_realm_get_id(this_rp))) {
 353       tr_debug("tr_comm_find_rp: Found RP %s in community %s.", rp_realm->buf, tr_comm_get_id(comm)->buf);
 354       talloc_free(tmp_ctx);
 355       return this_rp;
 356     }
 357   }
 358   tr_debug("tr_comm_find_rp: Unable to find RP %s in community %s.", rp_realm->buf, tr_comm_get_id(comm)->buf);
 359   talloc_free(tmp_ctx);
 360   return NULL;
 361 }
 362
 363 static TR_COMM *tr_comm_lookup(TR_COMM *comms, TR_NAME *comm_name)
 364 {
 365   TR_COMM *cfg_comm = NULL;
 366
 367   for (cfg_comm = comms; NULL != cfg_comm; cfg_comm = cfg_comm->next) {
 368     if (0==tr_name_cmp(cfg_comm->id, comm_name))
 369       return cfg_comm;
 370   }
 371   return NULL;
 372 }
 373
 374 TR_COMM_ITER *tr_comm_iter_new(TALLOC_CTX *mem_ctx)
 375 {
 376   TR_COMM_ITER *iter=talloc(mem_ctx, TR_COMM_ITER);
 377   if (iter!=NULL) {
 378     iter->cur_comm=NULL;
 379     iter->cur_memb=NULL;
 380     iter->match=NULL;
 381     iter->realm=NULL;
 382   }
 383   return iter;
 384 }
 385
 386 void tr_comm_iter_free(TR_COMM_ITER *iter)
 387 {
 388   talloc_free(iter);
 389 }
 390
 391
 392 TR_COMM *tr_comm_iter_first(TR_COMM_ITER *iter, TR_COMM_TABLE *ctab, TR_NAME *realm)
 393 {
 394   iter->match=realm;
 395
 396   /* find memberships for this realm */
 397   for (iter->cur_memb=ctab->memberships;
 398        iter->cur_memb!=NULL;
 399        iter->cur_memb=iter->cur_memb->next) {
 400     if (0==tr_name_cmp(iter->match, tr_comm_memb_get_realm_id(iter->cur_memb)))
 401       return tr_comm_memb_get_comm(iter->cur_memb);
 402   }
 403   return NULL;
 404 }
 405
 406 TR_COMM *tr_comm_iter_next(TR_COMM_ITER *iter)
 407 {
 408   for (iter->cur_memb=iter->cur_memb->next;
 409        iter->cur_memb!=NULL;
 410        iter->cur_memb=iter->cur_memb->next) {
 411     if (0==tr_name_cmp(iter->match, tr_comm_memb_get_realm_id(iter->cur_memb)))
 412       return tr_comm_memb_get_comm(iter->cur_memb);
 413   }
 414   return NULL;
 415 }
 416
 417 /* iterate only over RPs */
 418 TR_COMM *tr_comm_iter_first_rp(TR_COMM_ITER *iter, TR_COMM_TABLE *ctab, TR_NAME *realm)
 419 {
 420   iter->match=realm;
 421
 422   /* find memberships for this realm */
 423   for (iter->cur_memb=ctab->memberships;
 424        iter->cur_memb!=NULL;
 425        iter->cur_memb=iter->cur_memb->next) {
 426     if ((tr_comm_memb_get_rp_realm(iter->cur_memb)!=NULL) &&
 427         (0==tr_name_cmp(iter->match, tr_comm_memb_get_realm_id(iter->cur_memb))))
 428       return tr_comm_memb_get_comm(iter->cur_memb);
 429   }
 430   return NULL;
 431 }
 432
 433 TR_COMM *tr_comm_iter_next_rp(TR_COMM_ITER *iter)
 434 {
 435   for (iter->cur_memb=iter->cur_memb->next;
 436        iter->cur_memb!=NULL;
 437        iter->cur_memb=iter->cur_memb->next) {
 438     if ((tr_comm_memb_get_rp_realm(iter->cur_memb)!=NULL) &&
 439         (0==tr_name_cmp(iter->match, tr_comm_memb_get_realm_id(iter->cur_memb))))
 440       return tr_comm_memb_get_comm(iter->cur_memb);
 441   }
 442   return NULL;
 443 }
 444
 445 /* iterate only over IDPs */
 446 TR_COMM *tr_comm_iter_first_idp(TR_COMM_ITER *iter, TR_COMM_TABLE *ctab, TR_NAME *realm)
 447 {
 448   iter->match=realm;
 449
 450   /* find memberships for this realm */
 451   for (iter->cur_memb=ctab->memberships;
 452        iter->cur_memb!=NULL;
 453        iter->cur_memb=iter->cur_memb->next) {
 454     if ((tr_comm_memb_get_idp_realm(iter->cur_memb)!=NULL) &&
 455         (0==tr_name_cmp(iter->match, tr_comm_memb_get_realm_id(iter->cur_memb))))
 456       return tr_comm_memb_get_comm(iter->cur_memb);
 457   }
 458   return NULL;
 459 }
 460
 461 TR_COMM *tr_comm_iter_next_idp(TR_COMM_ITER *iter)
 462 {
 463   for (iter->cur_memb=iter->cur_memb->next;
 464        iter->cur_memb!=NULL;
 465        iter->cur_memb=iter->cur_memb->next) {
 466     if ((tr_comm_memb_get_idp_realm(iter->cur_memb)!=NULL) &&
 467         (0==tr_name_cmp(iter->match, tr_comm_memb_get_realm_id(iter->cur_memb))))
 468       return tr_comm_memb_get_comm(iter->cur_memb);
 469   }
 470   return NULL;
 471 }
 472
 473 static TR_REALM *tr_realm_new(TALLOC_CTX *mem_ctx)
 474 {
 475   TR_REALM *realm=talloc(mem_ctx, TR_REALM);
 476   if (realm!=NULL) {
 477     realm->role=TR_ROLE_UNKNOWN;
 478     realm->rp=NULL;
 479     realm->idp=NULL;
 480   }
 481   return realm;
 482 }
 483
 484 static void tr_realm_free(TR_REALM *realm)
 485 {
 486   talloc_free(realm);
 487 }
 488
 489 static void tr_realm_set_rp(TR_REALM *realm, TR_RP_REALM *rp)
 490 {
 491   if (realm->idp!=NULL)
 492     realm->idp=NULL;
 493   realm->role=TR_ROLE_RP;
 494   realm->rp=rp;
 495 }
 496
 497 static void tr_realm_set_idp(TR_REALM *realm, TR_IDP_REALM *idp)
 498 {
 499   if (realm->rp!=NULL)
 500     realm->rp=NULL;
 501   realm->role=TR_ROLE_IDP;
 502   realm->idp=idp;
 503 }
 504
 505 TR_NAME *tr_realm_get_id(TR_REALM *realm)
 506 {
 507   switch (realm->role) {
 508   case TR_ROLE_RP:
 509     return tr_rp_realm_get_id(realm->rp);
 510   case TR_ROLE_IDP:
 511     return tr_idp_realm_get_id(realm->idp);
 512   default:
 513     break;
 514   }
 515   return NULL;
 516 }
 517
 518 TR_NAME *tr_realm_dup_id(TR_REALM *realm)
 519 {
 520   return tr_dup_name(tr_realm_get_id(realm));
 521 }
 522
 523 /* Iterate over either sort of realm. Do not free the TR_REALM returned. It becomes
 524  * undefined/invalid after the next operation affecting the iterator. */
 525 TR_REALM *tr_realm_iter_first(TR_COMM_ITER *iter, TR_COMM_TABLE *ctab, TR_NAME *comm)
 526 {
 527   iter->match=comm;
 528   if (iter->realm==NULL)
 529     iter->realm=tr_realm_new(iter);
 530   if (iter->realm==NULL)
 531     return NULL;
 532
 533   /* find memberships for this comm */
 534   for (iter->cur_memb=ctab->memberships;
 535        iter->cur_memb!=NULL;
 536        iter->cur_memb=iter->cur_memb->next) {
 537     if (0==tr_name_cmp(iter->match,
 538                        tr_comm_get_id(tr_comm_memb_get_comm(iter->cur_memb)))) {
 539       /* found a match, determine whether it's an rp realm or an idp realm */
 540       if (tr_comm_memb_get_rp_realm(iter->cur_memb)!=NULL)
 541         tr_realm_set_rp(iter->realm, tr_comm_memb_get_rp_realm(iter->cur_memb));
 542       else if (tr_comm_memb_get_idp_realm(iter->cur_memb)!=NULL)
 543         tr_realm_set_idp(iter->realm, tr_comm_memb_get_idp_realm(iter->cur_memb));
 544       else {
 545         if (iter->realm!=NULL)
 546           tr_realm_free(iter->realm);
 547         iter->realm=NULL;
 548       }
 549       return iter->realm;
 550     }
 551   }
 552   if (iter->realm!=NULL)
 553     tr_realm_free(iter->realm);
 554   iter->realm=NULL;
 555   return NULL;
 556 }
 557
 558 TR_REALM *tr_realm_iter_next(TR_COMM_ITER *iter)
 559 {
 560   if (iter->realm==NULL)
 561     return NULL;
 562
 563   /* find memberships for this comm */
 564   for (iter->cur_memb=iter->cur_memb->next;
 565        iter->cur_memb!=NULL;
 566        iter->cur_memb=iter->cur_memb->next) {
 567     if (0==tr_name_cmp(iter->match,
 568                        tr_comm_get_id(tr_comm_memb_get_comm(iter->cur_memb)))) {
 569       /* found a match, determine whether it's an rp realm or an idp realm */
 570       if (tr_comm_memb_get_rp_realm(iter->cur_memb)!=NULL)
 571         tr_realm_set_rp(iter->realm, tr_comm_memb_get_rp_realm(iter->cur_memb));
 572       else if (tr_comm_memb_get_idp_realm(iter->cur_memb)!=NULL)
 573         tr_realm_set_idp(iter->realm, tr_comm_memb_get_idp_realm(iter->cur_memb));
 574       else {
 575         if (iter->realm!=NULL)
 576           tr_realm_free(iter->realm);
 577         iter->realm=NULL;
 578       }
 579       return iter->realm;
 580     }
 581   }
 582   if (iter->realm!=NULL)
 583     tr_realm_free(iter->realm);
 584   iter->realm=NULL;
 585   return NULL;
 586 }
 587
 588 TR_RP_REALM *tr_rp_realm_iter_first(TR_COMM_ITER *iter, TR_COMM_TABLE *ctab, TR_NAME *comm)
 589 {
 590   iter->match=comm;
 591
 592   /* find memberships for this comm */
 593   for (iter->cur_memb=ctab->memberships;
 594        iter->cur_memb!=NULL;
 595        iter->cur_memb=iter->cur_memb->next) {
 596     if ((tr_comm_memb_get_rp_realm(iter->cur_memb)!=NULL) &&
 597         (0==tr_name_cmp(iter->match, tr_comm_get_id(tr_comm_memb_get_comm(iter->cur_memb)))))
 598       return tr_comm_memb_get_rp_realm(iter->cur_memb);
 599   }
 600   return NULL;
 601 }
 602
 603 TR_RP_REALM *tr_rp_realm_iter_next(TR_COMM_ITER *iter)
 604 {
 605   for (iter->cur_memb=iter->cur_memb->next;
 606        iter->cur_memb!=NULL;
 607        iter->cur_memb=iter->cur_memb->next) {
 608     if ((tr_comm_memb_get_rp_realm(iter->cur_memb)!=NULL) &&
 609         (0==tr_name_cmp(iter->match, tr_comm_get_id(tr_comm_memb_get_comm(iter->cur_memb)))))
 610       return tr_comm_memb_get_rp_realm(iter->cur_memb);
 611   }
 612   return NULL;
 613 }
 614
 615 TR_IDP_REALM *tr_idp_realm_iter_first(TR_COMM_ITER *iter, TR_COMM_TABLE *ctab, TR_NAME *comm)
 616 {
 617   iter->match=comm;
 618
 619   /* find memberships for this comm */
 620   for (iter->cur_memb=ctab->memberships;
 621        iter->cur_memb!=NULL;
 622        iter->cur_memb=iter->cur_memb->next) {
 623     if ((tr_comm_memb_get_idp_realm(iter->cur_memb)!=NULL) &&
 624         (0==tr_name_cmp(iter->match, tr_comm_get_id(tr_comm_memb_get_comm(iter->cur_memb)))))
 625       return tr_comm_memb_get_idp_realm(iter->cur_memb);
 626   }
 627   return NULL;
 628 }
 629
 630 TR_IDP_REALM *tr_idp_realm_iter_next(TR_COMM_ITER *iter)
 631 {
 632   for (iter->cur_memb=iter->cur_memb->next;
 633        iter->cur_memb!=NULL;
 634        iter->cur_memb=iter->cur_memb->next) {
 635     if ((tr_comm_memb_get_idp_realm(iter->cur_memb)!=NULL) &&
 636         (0==tr_name_cmp(iter->match, tr_comm_get_id(tr_comm_memb_get_comm(iter->cur_memb)))))
 637       return tr_comm_memb_get_idp_realm(iter->cur_memb);
 638   }
 639   return NULL;
 640 }
 641
 642 /* iterators for all communities in a table */
 643 TR_COMM *tr_comm_table_iter_first(TR_COMM_ITER *iter, TR_COMM_TABLE *ctab)
 644 {
 645   iter->cur_comm=ctab->comms;
 646   return iter->cur_comm;
 647 }
 648
 649 TR_COMM *tr_comm_table_iter_next(TR_COMM_ITER *iter)
 650 {
 651   return iter->cur_comm=iter->cur_comm->next;
 652 }
 653
 654 const char *tr_comm_type_to_str(TR_COMM_TYPE type)
 655 {
 656   const char *s=NULL;
 657   switch(type) {
 658   case TR_COMM_UNKNOWN:
 659     s="unknown";
 660     break;
 661   case TR_COMM_APC:
 662     s="apc";
 663     break;
 664   case TR_COMM_COI:
 665     s="coi";
 666     break;
 667   default:
 668     s="invalid";
 669   }
 670   return s;
 671 }
 672
 673 TR_COMM_MEMB *tr_comm_memb_iter_first(TR_COMM_ITER *iter, TR_COMM_MEMB *memb)
 674 {
 675   iter->cur_memb=memb;
 676   return iter->cur_memb;
 677 }
 678
 679 TR_COMM_MEMB *tr_comm_memb_iter_next(TR_COMM_ITER *iter)
 680 {
 681   if (iter->cur_memb!=NULL)
 682     iter->cur_memb=iter->cur_memb->origin_next;
 683   return iter->cur_memb;
 684 }
 685
 686 TR_COMM_TYPE tr_comm_type_from_str(const char *s)
 687 {
 688   if (strcmp(s, "apc")==0)
 689     return TR_COMM_APC;
 690   if (strcmp(s,"coi")==0)
 691     return TR_COMM_COI;
 692   return TR_COMM_UNKNOWN;
 693 }
 694
 695
 696 static int tr_comm_memb_destructor(void *obj)
 697 {
 698   TR_COMM_MEMB *memb=talloc_get_type_abort(obj, TR_COMM_MEMB);
 699   if (memb->origin!=NULL)
 700     tr_free_name(memb->origin);
 701
 702   if (memb->rp!=NULL)
 703     tr_rp_realm_decref(memb->rp);
 704   if (memb->idp!=NULL)
 705     tr_idp_realm_decref(memb->idp);
 706   if (memb->comm!=NULL)
 707     tr_comm_decref(memb->comm);
 708   if (memb->provenance!=NULL)
 709     json_decref(memb->provenance);
 710   return 0;
 711 }
 712
 713 TR_COMM_MEMB *tr_comm_memb_new(TALLOC_CTX *mem_ctx)
 714 {
 715   TR_COMM_MEMB *memb=talloc(mem_ctx, TR_COMM_MEMB);
 716   if (memb!=NULL) {
 717     memb->next=NULL;
 718     memb->origin_next=NULL;
 719     memb->idp=NULL;
 720     memb->rp=NULL;
 721     memb->comm=NULL;
 722     memb->origin=NULL;
 723     memb->provenance=NULL;
 724     memb->interval=0;
 725     memb->expiry=talloc(memb, struct timespec);
 726     if (memb->expiry==NULL) {
 727       talloc_free(memb);
 728       return NULL;
 729     }
 730     *(memb->expiry)=(struct timespec){0,0};
 731     talloc_set_destructor(memb, tr_comm_memb_destructor);
 732   }
 733   return memb;
 734 }
 735
 736 void tr_comm_memb_free(TR_COMM_MEMB *memb)
 737 {
 738   talloc_free(memb);
 739 }
 740
 741 TR_REALM_ROLE tr_comm_memb_get_role(TR_COMM_MEMB *memb)
 742 {
 743   if (memb->rp!=NULL)
 744     return TR_ROLE_RP;
 745   if (memb->idp!=NULL)
 746     return TR_ROLE_IDP;
 747   return TR_ROLE_UNKNOWN;
 748 }
 749
 750 void tr_comm_memb_set_rp_realm(TR_COMM_MEMB *memb, TR_RP_REALM *realm)
 751 {
 752   if (memb->idp!=NULL) {
 753     tr_idp_realm_decref(memb->idp);
 754     memb->idp=NULL;
 755   }
 756   if (memb->rp!=NULL)
 757     tr_rp_realm_decref(memb->rp);
 758
 759
 760   memb->rp=realm;
 761   tr_rp_realm_incref(realm);
 762 }
 763
 764 TR_RP_REALM *tr_comm_memb_get_rp_realm(TR_COMM_MEMB *memb)
 765 {
 766   return memb->rp;
 767 }
 768
 769 void tr_comm_memb_set_idp_realm(TR_COMM_MEMB *memb, TR_IDP_REALM *realm)
 770 {
 771   if (memb->rp!=NULL) {
 772     tr_rp_realm_decref(memb->rp);
 773     memb->rp=NULL;
 774   }
 775   if (memb->idp!=NULL)
 776     tr_idp_realm_decref(memb->idp);
 777
 778   memb->idp=realm;
 779   tr_idp_realm_incref(realm);
 780 }
 781
 782 TR_IDP_REALM *tr_comm_memb_get_idp_realm(TR_COMM_MEMB *memb)
 783 {
 784   return memb->idp;
 785 }
 786
 787 void tr_comm_memb_set_comm(TR_COMM_MEMB *memb, TR_COMM *comm)
 788 {
 789   if (memb->comm!=NULL)
 790     tr_comm_decref(memb->comm);
 791   memb->comm=comm;
 792   tr_comm_incref(comm);
 793 }
 794
 795 TR_COMM *tr_comm_memb_get_comm(TR_COMM_MEMB *memb)
 796 {
 797   return memb->comm;
 798 }
 799
 800 static void tr_comm_memb_set_origin(TR_COMM_MEMB *memb, TR_NAME *origin)
 801 {
 802   if (memb->origin!=NULL)
 803     tr_free_name(memb->origin);
 804   memb->origin=origin;
 805 }
 806
 807 TR_NAME *tr_comm_memb_get_origin(TR_COMM_MEMB *memb)
 808 {
 809   return memb->origin;
 810 }
 811
 812 TR_NAME *tr_comm_memb_dup_origin(TR_COMM_MEMB *memb)
 813 {
 814   if (memb->origin!=NULL)
 815     return tr_dup_name(memb->origin);
 816   return NULL;
 817 }
 818
 819 json_t *tr_comm_memb_get_provenance(TR_COMM_MEMB *memb)
 820 {
 821   if (memb!=NULL)
 822     return memb->provenance;
 823   return NULL;
 824 }
 825
 826 void tr_comm_memb_set_provenance(TR_COMM_MEMB *memb, json_t *prov)
 827 {
 828   if (memb->provenance)
 829     json_decref(memb->provenance);
 830
 831   memb->provenance=prov;
 832   if (prov!=NULL) {
 833     json_incref(prov);
 834
 835     /* next line sets origin to NULL if provenance is empty because jansson
 836      * routines return NULL on error */
 837     memb->origin=tr_new_name(json_string_value(json_array_get(prov, 0)));
 838   } else {
 839     tr_comm_memb_set_origin(memb, NULL);
 840   }
 841 }
 842
 843 void tr_comm_memb_add_to_provenance(TR_COMM_MEMB *memb, TR_NAME *hop)
 844 {
 845   if (memb->provenance==NULL) {
 846     memb->provenance=json_array();
 847     if (memb->provenance==NULL) {
 848       tr_err("tr_comm_memb_add_to_provenance: unable to allocate provenance list.");
 849       return;
 850     }
 851     /* this is the first entry in the provenance, so it is the origin */
 852     tr_comm_memb_set_origin(memb,tr_dup_name(hop));
 853     if (memb->origin==NULL) {
 854       tr_err("tr_comm_memb_add_to_provenance: unable to allocate origin.");
 855       json_decref(memb->provenance);
 856       memb->provenance=NULL;
 857       return;
 858     }
 859   }
 860   if (0!=json_array_append_new(memb->provenance, tr_name_to_json_string(hop)))
 861     tr_err("tr_comm_memb_add_to_provenance: unable to extend provenance list.");
 862 }
 863
 864 size_t tr_comm_memb_provenance_len(TR_COMM_MEMB *memb)
 865 {
 866   if (memb->provenance==NULL)
 867     return 0;
 868   return json_array_size(memb->provenance);
 869 }
 870
 871 void tr_comm_memb_set_interval(TR_COMM_MEMB *memb, unsigned int interval)
 872 {
 873   memb->interval=interval;
 874 }
 875
 876 unsigned int tr_comm_memb_get_interval(TR_COMM_MEMB *memb)
 877 {
 878   return memb->interval;
 879 }
 880
 881 void tr_comm_memb_set_expiry(TR_COMM_MEMB *memb, struct timespec *time)
 882 {
 883   memb->expiry->tv_sec=time->tv_sec;
 884   memb->expiry->tv_nsec=time->tv_nsec;
 885 }
 886
 887 struct timespec *tr_comm_memb_get_expiry(TR_COMM_MEMB *memb)
 888 {
 889   return memb->expiry;
 890 }
 891
 892 int tr_comm_memb_is_expired(TR_COMM_MEMB *memb, struct timespec *curtime)
 893 {
 894   return ((curtime->tv_sec > memb->expiry->tv_sec)
 895          || ((curtime->tv_sec == memb->expiry->tv_sec)
 896             &&(curtime->tv_nsec >= memb->expiry->tv_nsec)));
 897 }
 898
 899 TR_COMM_TABLE *tr_comm_table_new(TALLOC_CTX *mem_ctx)
 900 {
 901   TR_COMM_TABLE *ctab=talloc(mem_ctx, TR_COMM_TABLE);
 902   if (ctab!=NULL) {
 903     ctab->comms=NULL;
 904     ctab->memberships=NULL;
 905     ctab->idp_realms=NULL;
 906     ctab->rp_realms=NULL;
 907   }
 908   return ctab;
 909 }
 910
 911 void tr_comm_table_free(TR_COMM_TABLE *ctab)
 912 {
 913   talloc_free(ctab);
 914 }
 915
 916 static TR_REALM_ROLE tr_comm_memb_role(TR_COMM_MEMB *memb)
 917 {
 918   if (memb->rp!=NULL)
 919     return TR_ROLE_RP;
 920   if (memb->idp!=NULL)
 921     return TR_ROLE_IDP;
 922
 923   return TR_ROLE_UNKNOWN;
 924 }
 925
 926 void tr_comm_table_add_memb(TR_COMM_TABLE *ctab, TR_COMM_MEMB *new)
 927 {
 928   TR_COMM_MEMB *cur=NULL;
 929
 930   /* TODO: validate the member (must have valid comm and realm) */
 931
 932   /* handle the empty list case */
 933   if (ctab->memberships==NULL) {
 934     ctab->memberships=new;
 935     talloc_steal(ctab, new);
 936     return;
 937   }
 938
 939   /* The list was not empty. See if we already have a membership for this realm/comm/role */
 940   switch (tr_comm_memb_role(new)) {
 941   case TR_ROLE_RP:
 942     cur=tr_comm_table_find_rp_memb(ctab,
 943                                    tr_rp_realm_get_id(tr_comm_memb_get_rp_realm(new)),
 944                                    tr_comm_get_id(tr_comm_memb_get_comm(new)));
 945     break;
 946   case TR_ROLE_IDP:
 947     cur=tr_comm_table_find_idp_memb(ctab,
 948                                     tr_idp_realm_get_id(tr_comm_memb_get_idp_realm(new)),
 949                                     tr_comm_get_id(tr_comm_memb_get_comm(new)));
 950     break;
 951   case TR_ROLE_UNKNOWN:
 952   default:
 953     tr_err("tr_comm_table_add_memb: realm with unknown role added.");
 954     cur=NULL;
 955   }
 956
 957   if (cur==NULL) {
 958     /* no entry for this realm/comm/role, tack it on the end */
 959     for (cur=ctab->memberships; cur->next!=NULL; cur=cur->next) { }
 960     cur->next=new;
 961   } else {
 962     /* Found an entry. Add to the end of its same-origin list. */
 963     while (cur->origin_next!=NULL) {
 964       cur=cur->origin_next;
 965     }
 966     cur->origin_next=new;
 967   }
 968
 969   talloc_steal(ctab, new);
 970 }
 971
 972 /* Remove memb from ctab. Do not free anything. Do nothing if memb not in ctab. */
 973 void tr_comm_table_remove_memb(TR_COMM_TABLE *ctab, TR_COMM_MEMB *memb)
 974 {
 975   TR_COMM_MEMB *cur=NULL; /* for walking the main list */
 976   TR_COMM_MEMB *orig_cur=NULL; /* for walking the origin list */
 977
 978   if ((memb==NULL) || (ctab->memberships==NULL))
 979     return;
 980
 981   /* see if it's the first member */
 982   if (ctab->memberships==memb) {
 983     if (memb->origin_next!=NULL) {
 984       memb->origin_next->next=ctab->memberships->next;
 985       ctab->memberships=memb->origin_next;
 986     } else
 987       ctab->memberships=memb->next;
 988
 989     return;
 990   }
 991
 992   /* see if it's in first member's origin list */
 993   for (orig_cur=ctab->memberships->origin_next;
 994        orig_cur!=NULL;
 995        orig_cur=ctab->memberships->origin_next) {
 996     if (orig_cur==memb) {
 997       orig_cur->origin_next=memb->origin_next;
 998       return;
 999     }
1000   }
1001
1002   /* now we have to walk the rest of the tree */
1003   for (cur=ctab->memberships; cur->next!=NULL; cur=cur->next) {
1004     if (cur->next==memb) {
1005       /* it matched an entry on the main list */
1006       if (memb->origin_next!=NULL) {
1007         /* replace the entry in the main list with the next element on the origin list */
1008         memb->origin_next->next=memb->next;
1009         cur->next=memb->origin_next;
1010       } else
1011         cur->next=memb->next; /* no origin list, just drop memb */
1012       return;
1013     } else {
1014       /* it was not on the main list, walk the origin list */
1015       for (orig_cur=cur; orig_cur->next!=NULL; orig_cur=orig_cur->next) {
1016         if (orig_cur->next==memb) {
1017           orig_cur->next=memb->next;
1018           return; /* just drop the element from the origin list */
1019         }
1020       }
1021     }
1022   }
1023 }
1024
1025 static TR_NAME *tr_comm_memb_get_realm_id(TR_COMM_MEMB *memb)
1026 {
1027   if (memb->rp!=NULL)
1028     return tr_rp_realm_get_id(memb->rp);
1029   else
1030     return tr_idp_realm_get_id(memb->idp);
1031 }
1032
1033 /* find a membership from any origin */
1034 TR_COMM_MEMB *tr_comm_table_find_memb(TR_COMM_TABLE *ctab, TR_NAME *realm, TR_NAME *comm)
1035 {
1036   TR_COMM_MEMB *cur=NULL;
1037   TR_NAME *cur_realm_name=NULL;
1038
1039   for (cur=ctab->memberships; cur!=NULL; cur=cur->next) {
1040     cur_realm_name=tr_comm_memb_get_realm_id(cur);
1041     if (cur_realm_name==NULL) {
1042       tr_warning("tr_comm_table_find: encountered realm with no name.");
1043       continue;
1044     }
1045     if ((0==tr_name_cmp(realm, cur_realm_name)) &&
1046         (0==tr_name_cmp(comm, tr_comm_get_id(tr_comm_memb_get_comm(cur))))) {
1047       return cur;
1048     }
1049   }
1050   return NULL;
1051 }
1052
1053 /* find a membership from a particular origin */
1054 TR_COMM_MEMB *tr_comm_table_find_memb_origin(TR_COMM_TABLE *ctab, TR_NAME *realm, TR_NAME *comm, TR_NAME *origin)
1055 {
1056   TR_NAME *cur_orig=NULL;
1057   TR_COMM_MEMB *cur=tr_comm_table_find_memb(ctab, realm, comm);
1058   if (cur==NULL)
1059     return NULL; /* no match */
1060
1061   /* had a match for comm/realm; find origin match */
1062   while (cur!=NULL) {
1063     if (((origin==NULL) && (cur_orig==NULL)) ||
1064         ((origin!=NULL) && (cur_orig!=NULL) && (0==tr_name_cmp(origin, cur_orig))))
1065       return cur; /* found a match */
1066     cur=cur->origin_next;
1067   }
1068   return NULL; /* no match */
1069 }
1070
1071
1072 /* find an idp membership regardless of its origin */
1073 TR_COMM_MEMB *tr_comm_table_find_idp_memb(TR_COMM_TABLE *ctab, TR_NAME *realm, TR_NAME *comm)
1074 {
1075   TR_COMM_MEMB *cur=NULL;
1076   TR_IDP_REALM *idp_realm=NULL;
1077
1078   for (cur=ctab->memberships; cur!=NULL; cur=cur->next) {
1079     idp_realm=tr_comm_memb_get_idp_realm(cur);
1080     if (idp_realm==NULL)
1081       continue; /* was not an idp */
1082
1083     if ((0==tr_name_cmp(realm, idp_realm->realm_id)) &&
1084         (0==tr_name_cmp(comm, tr_comm_get_id(tr_comm_memb_get_comm(cur))))) {
1085       return cur;
1086     }
1087   }
1088   return NULL;
1089 }
1090
1091 /* find an idp membership from a particular origin */
1092 TR_COMM_MEMB *tr_comm_table_find_idp_memb_origin(TR_COMM_TABLE *ctab, TR_NAME *realm, TR_NAME *comm, TR_NAME *origin)
1093 {
1094   TR_NAME *cur_orig=NULL;
1095   TR_COMM_MEMB *cur=tr_comm_table_find_idp_memb(ctab, realm, comm);
1096   if (cur==NULL)
1097     return NULL; /* no match */
1098
1099   /* had a match for comm/realm; find origin match */
1100   while (cur!=NULL) {
1101     cur_orig=tr_comm_memb_get_origin(cur);
1102     if (((origin==NULL) && (cur_orig==NULL)) ||
1103         ((origin!=NULL) && (cur_orig!=NULL) && (0==tr_name_cmp(origin, cur_orig))))
1104       return cur; /* found a match */
1105     cur=cur->origin_next;
1106   }
1107   return NULL; /* no match */
1108 }
1109
1110 /* find an rp membership from any origin */
1111 TR_COMM_MEMB *tr_comm_table_find_rp_memb(TR_COMM_TABLE *ctab, TR_NAME *realm, TR_NAME *comm)
1112 {
1113   TR_COMM_MEMB *cur=NULL;
1114   TR_RP_REALM *rp_realm=NULL;
1115
1116   for (cur=ctab->memberships; cur!=NULL; cur=cur->next) {
1117     rp_realm=tr_comm_memb_get_rp_realm(cur);
1118     if (rp_realm==NULL)
1119       continue; /* was not an rp */
1120
1121     if ((0==tr_name_cmp(realm, tr_rp_realm_get_id(rp_realm))) &&
1122         (0==tr_name_cmp(comm, tr_comm_get_id(tr_comm_memb_get_comm(cur))))) {
1123       return cur;
1124     }
1125   }
1126   return NULL;
1127 }
1128
1129 /* find an rp membership from a particular origin */
1130 TR_COMM_MEMB *tr_comm_table_find_rp_memb_origin(TR_COMM_TABLE *ctab, TR_NAME *realm, TR_NAME *comm, TR_NAME *origin)
1131 {
1132   TR_NAME *cur_orig=NULL;
1133   TR_COMM_MEMB *cur=tr_comm_table_find_rp_memb(ctab, realm, comm);
1134   if (cur==NULL)
1135     return NULL; /* no match */
1136
1137   /* had a match for comm/realm; find origin match */
1138   while (cur!=NULL) {
1139     cur_orig=tr_comm_memb_get_origin(cur);
1140     if (((origin==NULL) && (cur_orig==NULL)) ||
1141         ((origin!=NULL) && (cur_orig!=NULL) && (0==tr_name_cmp(origin, cur_orig))))
1142       return cur; /* found a match */
1143     cur=cur->origin_next;
1144   }
1145   return NULL; /* no match */
1146 }
1147
1148 TR_COMM *tr_comm_table_find_comm(TR_COMM_TABLE *ctab, TR_NAME *comm_id)
1149 {
1150   return tr_comm_lookup(ctab->comms, comm_id);
1151 }
1152
1153 void tr_comm_table_add_comm(TR_COMM_TABLE *ctab, TR_COMM *new)
1154 {
1155   tr_comm_add(ctab->comms, new);
1156 }
1157
1158 void tr_comm_table_remove_comm(TR_COMM_TABLE *ctab, TR_COMM *comm)
1159 {
1160   tr_comm_remove(ctab->comms, comm);
1161 }
1162
1163 /* how many communities in the table? */
1164 size_t tr_comm_table_size(TR_COMM_TABLE *ctab)
1165 {
1166   size_t count=0;
1167   TR_COMM *this=ctab->comms;
1168   while(this!=NULL) {
1169     this=this->next;
1170     count++;
1171   }
1172   return count;
1173 }
1174
1175
1176 const char *tr_realm_role_to_str(TR_REALM_ROLE role)
1177 {
1178   switch(role) {
1179   case TR_ROLE_IDP:
1180     return "idp";
1181   case TR_ROLE_RP:
1182     return "rp";
1183   default:
1184     return NULL;
1185   }
1186 }
1187
1188 TR_REALM_ROLE tr_realm_role_from_str(const char *s)
1189 {
1190   if (strcmp(s, "idp")==0)
1191     return TR_ROLE_IDP;
1192   if (strcmp(s, "rp")==0)
1193     return TR_ROLE_RP;
1194   return TR_ROLE_UNKNOWN;
1195 }
1196