trp/trps.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 <fcntl.h>
  36 #include <talloc.h>
  37 #include <errno.h>
  38 #include <unistd.h>
  39 #include <sys/time.h>
  40 #include <glib.h>
  41 #include <string.h>
  42
  43 #include <gsscon.h>
  44 #include <tr_comm.h>
  45 #include <tr_apc.h>
  46 #include <tr_rp.h>
  47 #include <trust_router/tr_name.h>
  48 #include <trp_internal.h>
  49 #include <tr_gss.h>
  50 #include <trp_ptable.h>
  51 #include <trp_rtable.h>
  52 #include <tr_debug.h>
  53 #include <tr_util.h>
  54
  55 static int trps_destructor(void *object)
  56 {
  57   TRPS_INSTANCE *trps=talloc_get_type_abort(object, TRPS_INSTANCE);
  58   if (trps->rtable!=NULL)
  59     trp_rtable_free(trps->rtable);
  60   return 0;
  61 }
  62
  63 TRPS_INSTANCE *trps_new (TALLOC_CTX *mem_ctx)
  64 {
  65   TRPS_INSTANCE *trps=talloc(mem_ctx, TRPS_INSTANCE);
  66   if (trps!=NULL)  {
  67     trps->hostname=NULL;
  68     trps->port=0;
  69     trps->cookie=NULL;
  70     trps->conn=NULL;
  71     trps->trpc=NULL;
  72     trps->update_interval=(struct timeval){0,0};
  73     trps->sweep_interval=(struct timeval){0,0};
  74     trps->ptable=NULL;
  75
  76     trps->mq=tr_mq_new(trps);
  77     if (trps->mq==NULL) {
  78       /* failed to allocate mq */
  79       talloc_free(trps);
  80       return NULL;
  81     }
  82
  83     trps->rtable=NULL;
  84     if (trps_init_rtable(trps) != TRP_SUCCESS) {
  85       /* failed to allocate rtable */
  86       talloc_free(trps);
  87       return NULL;
  88     }
  89
  90     talloc_set_destructor((void *)trps, trps_destructor);
  91   }
  92   return trps;
  93 }
  94
  95 /* create a new route table, first discarding an old one if necessary */
  96 TRP_RC trps_init_rtable(TRPS_INSTANCE *trps)
  97 {
  98   if (trps->rtable != NULL) {
  99     trp_rtable_free(trps->rtable);
 100     trps->rtable=NULL;
 101   }
 102
 103   trps->rtable=trp_rtable_new();
 104   if (trps->rtable==NULL) {
 105     return TRP_NOMEM;
 106   }
 107   return TRP_SUCCESS;
 108 }
 109
 110 void trps_clear_rtable(TRPS_INSTANCE *trps)
 111 {
 112   trp_rtable_clear(trps->rtable);
 113 }
 114
 115 void trps_free (TRPS_INSTANCE *trps)
 116 {
 117   if (trps!=NULL)
 118     talloc_free(trps);
 119 }
 120
 121 TR_MQ_MSG *trps_mq_pop(TRPS_INSTANCE *trps)
 122 {
 123   return tr_mq_pop(trps->mq, 0);
 124 }
 125
 126 void trps_mq_add(TRPS_INSTANCE *trps, TR_MQ_MSG *msg)
 127 {
 128   tr_mq_add(trps->mq, msg);
 129 }
 130
 131 unsigned int trps_get_connect_interval(TRPS_INSTANCE *trps)
 132 {
 133   return trps->connect_interval.tv_sec;
 134 }
 135
 136 void trps_set_connect_interval(TRPS_INSTANCE *trps, unsigned int interval)
 137 {
 138   trps->connect_interval.tv_sec=interval;
 139   trps->connect_interval.tv_usec=0;
 140 }
 141
 142 unsigned int trps_get_update_interval(TRPS_INSTANCE *trps)
 143 {
 144   return trps->update_interval.tv_sec;
 145 }
 146
 147 void trps_set_update_interval(TRPS_INSTANCE *trps, unsigned int interval)
 148 {
 149   trps->update_interval.tv_sec=interval;
 150   trps->update_interval.tv_usec=0;
 151 }
 152
 153 unsigned int trps_get_sweep_interval(TRPS_INSTANCE *trps)
 154 {
 155   return trps->sweep_interval.tv_sec;
 156 }
 157
 158 void trps_set_sweep_interval(TRPS_INSTANCE *trps, unsigned int interval)
 159 {
 160   trps->sweep_interval.tv_sec=interval;
 161   trps->sweep_interval.tv_usec=0;
 162 }
 163
 164 void trps_set_ctable(TRPS_INSTANCE *trps, TR_COMM_TABLE *comm)
 165 {
 166   trps->ctable=comm;
 167 }
 168
 169 void trps_set_ptable(TRPS_INSTANCE *trps, TRP_PTABLE *ptable)
 170 {
 171   if (trps->ptable!=NULL)
 172     trp_ptable_free(trps->ptable);
 173   trps->ptable=ptable;
 174 }
 175
 176 void trps_set_peer_status_callback(TRPS_INSTANCE *trps, void (*cb)(TRP_PEER *, void *), void *cookie)
 177 {
 178   TRP_PTABLE_ITER *iter=NULL;
 179   TRP_PEER *peer=NULL;
 180   if (trps->ptable==NULL)
 181     return;
 182
 183   iter=trp_ptable_iter_new(NULL);
 184   for (peer=trp_ptable_iter_first(iter, trps->ptable); peer!=NULL; peer=trp_ptable_iter_next(iter))
 185     trp_peer_set_conn_status_cb(peer, cb, cookie);
 186   trp_ptable_iter_free(iter);
 187 }
 188
 189 /* Get the label peers will know us by - needs to match trp_peer_get_label() output.
 190  * There is no get, only dup, because we don't store the label except when requested. */
 191 TR_NAME *trps_dup_label(TRPS_INSTANCE *trps)
 192 {
 193   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 194   TR_NAME *label=NULL;
 195   char *s=talloc_asprintf(tmp_ctx, "%s:%u", trps->hostname, trps->port);
 196   if (s==NULL)
 197     goto cleanup;
 198   label=tr_new_name(s);
 199
 200 cleanup:
 201   talloc_free(tmp_ctx);
 202   return label;
 203 }
 204
 205 TRPC_INSTANCE *trps_find_trpc(TRPS_INSTANCE *trps, TRP_PEER *peer)
 206 {
 207   TRPC_INSTANCE *cur=NULL;
 208   TR_NAME *name=NULL;
 209   TR_NAME *peer_servicename=trp_peer_get_servicename(peer);
 210
 211   for (cur=trps->trpc; cur!=NULL; cur=trpc_get_next(cur)) {
 212     name=trpc_get_gssname(cur);
 213     if ((name!=NULL) && (0==tr_name_cmp(peer_servicename, name))) {
 214       break;
 215     }
 216   }
 217   return cur;
 218 }
 219
 220 void trps_add_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *new)
 221 {
 222   if (trps->conn==NULL)
 223     trps->conn=new;
 224   else
 225     trp_connection_append(trps->conn, new);
 226
 227   talloc_steal(trps, new);
 228 }
 229
 230 /* ok to call more than once; guarantees connection no longer in the list.
 231  * Caller is responsible for freeing the removed element afterwards.  */
 232 void trps_remove_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *remove)
 233 {
 234   trps->conn=trp_connection_remove(trps->conn, remove);
 235 }
 236
 237 void trps_add_trpc(TRPS_INSTANCE *trps, TRPC_INSTANCE *trpc)
 238 {
 239   if (trps->trpc==NULL)
 240     trps->trpc=trpc;
 241   else
 242     trpc_append(trps->trpc, trpc);
 243
 244   talloc_steal(trps, trpc);
 245 }
 246
 247 /* ok to call more than once; guarantees trpc no longer in the list.
 248  * Caller is responsible for freeing the removed element afterwards.  */
 249 void trps_remove_trpc(TRPS_INSTANCE *trps, TRPC_INSTANCE *remove)
 250 {
 251   trps->trpc=trpc_remove(trps->trpc, remove);
 252 }
 253
 254 TRP_RC trps_send_msg(TRPS_INSTANCE *trps, TRP_PEER *peer, const char *msg)
 255 {
 256   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 257   TR_MQ_MSG *mq_msg=NULL;
 258   char *msg_dup=NULL;
 259   TRP_RC rc=TRP_ERROR;
 260   TRPC_INSTANCE *trpc=NULL;
 261
 262   /* get the connection for this peer */
 263   trpc=trps_find_trpc(trps, peer);
 264   /* instead, let's let that happen and then clear the queue when an attempt to
 265    * connect fails */
 266   if (trpc==NULL) {
 267     tr_warning("trps_send_msg: skipping message queued for missing TRP client entry.");
 268   } else {
 269     mq_msg=tr_mq_msg_new(tmp_ctx, TR_MQMSG_TRPC_SEND, TR_MQ_PRIO_NORMAL);
 270     msg_dup=talloc_strdup(mq_msg, msg); /* get local copy in mq_msg context */
 271     tr_mq_msg_set_payload(mq_msg, msg_dup, NULL); /* no need for a free() func */
 272     trpc_mq_add(trpc, mq_msg);
 273     rc=TRP_SUCCESS;
 274   }
 275   talloc_free(tmp_ctx);
 276   return rc;
 277 }
 278
 279 /* Listens on all interfaces. Returns number of sockets opened. Their
 280  * descriptors are stored in *fd_out, which should point to space for
 281  * up to max_fd of them. */
 282 static size_t trps_listen(TRPS_INSTANCE *trps, int port, int *fd_out, size_t max_fd)
 283 {
 284   int rc = 0;
 285   int conn = -1;
 286   int optval=0;
 287   struct addrinfo *ai=NULL;
 288   struct addrinfo *ai_head=NULL;
 289   struct addrinfo hints={.ai_flags=AI_PASSIVE,
 290                          .ai_family=AF_UNSPEC,
 291                          .ai_socktype=SOCK_STREAM,
 292                          .ai_protocol=IPPROTO_TCP};
 293   char *port_str=NULL;
 294   size_t n_opened=0;
 295
 296   port_str=talloc_asprintf(NULL, "%d", port);
 297   if (port_str==NULL) {
 298     tr_debug("trps_listen: unable to allocate port.");
 299     return -1;
 300   }
 301   getaddrinfo(NULL, port_str, &hints, &ai_head);
 302   talloc_free(port_str);
 303
 304   for (ai=ai_head,n_opened=0; (ai!=NULL)&&(n_opened<max_fd); ai=ai->ai_next) {
 305     if (0 > (conn = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol))) {
 306       tr_debug("trps_listen: unable to open socket.");
 307       continue;
 308     }
 309
 310     optval=1;
 311     if (0!=setsockopt(conn, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval)))
 312       tr_debug("trps_listen: unable to set SO_REUSEADDR."); /* not fatal? */
 313
 314     if (ai->ai_family==AF_INET6) {
 315       /* don't allow IPv4-mapped IPv6 addresses (per RFC4942, not sure
 316        * if still relevant) */
 317       if (0!=setsockopt(conn, IPPROTO_IPV6, IPV6_V6ONLY, &optval, sizeof(optval))) {
 318         tr_debug("trps_listen: unable to set IPV6_V6ONLY. Skipping interface.");
 319         close(conn);
 320         continue;
 321       }
 322     }
 323
 324     rc=bind(conn, ai->ai_addr, ai->ai_addrlen);
 325     if (rc<0) {
 326       tr_debug("trps_listen: unable to bind to socket.");
 327       close(conn);
 328       continue;
 329     }
 330
 331     if (0>listen(conn, 512)) {
 332       tr_debug("trps_listen: unable to listen on bound socket.");
 333       close(conn);
 334       continue;
 335     }
 336
 337     /* ok, this one worked. Save it */
 338     fd_out[n_opened++]=conn;
 339   }
 340   freeaddrinfo(ai_head);
 341
 342   if (n_opened==0) {
 343     tr_debug("trps_listen: no addresses available for listening.");
 344     return -1;
 345   }
 346
 347   tr_debug("trps_listen: TRP Server listening on port %d on %d socket%s",
 348            port,
 349            n_opened,
 350            (n_opened==1)?"":"s");
 351
 352   return n_opened;
 353 }
 354
 355 /* get the currently selected route if available */
 356 TRP_ROUTE *trps_get_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer)
 357 {
 358   return trp_rtable_get_entry(trps->rtable, comm, realm, peer);
 359 }
 360
 361 TRP_ROUTE *trps_get_selected_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm)
 362 {
 363   tr_debug("trps_get_selected_route: entered. trps=%p, comm=%p, realm=%p", trps, comm, realm);
 364   return trp_rtable_get_selected_entry(trps->rtable, comm, realm);
 365 }
 366
 367 /* copy the result if you want to keep it */
 368 TR_NAME *trps_get_next_hop(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm)
 369 {
 370   TRP_ROUTE *route=trps_get_selected_route(trps, comm, realm);
 371   if (route==NULL)
 372     return NULL;
 373
 374   return trp_route_get_next_hop(route);
 375 }
 376
 377
 378 /* mark a route as retracted */
 379 static void trps_retract_route(TRPS_INSTANCE *trps, TRP_ROUTE *entry)
 380 {
 381   trp_route_set_metric(entry, TRP_METRIC_INFINITY);
 382   trp_route_set_triggered(entry, 1);
 383 }
 384
 385 /* is this route retracted? */
 386 static int trps_route_retracted(TRPS_INSTANCE *trps, TRP_ROUTE *entry)
 387 {
 388   return (trp_metric_is_infinite(trp_route_get_metric(entry)));
 389 }
 390
 391 static TRP_RC trps_read_message(TRPS_INSTANCE *trps, TRP_CONNECTION *conn, TR_MSG **msg)
 392 {
 393   int err=0;
 394   char *buf=NULL;
 395   size_t buflen = 0;
 396   TRP_PEER *peer=NULL; /* entry in the peer table */
 397   TR_NAME *conn_peer=NULL; /* name from the TRP_CONN, which comes from the gss context */
 398
 399   tr_debug("trps_read_message: started");
 400   if (err = gsscon_read_encrypted_token(trp_connection_get_fd(conn),
 401                                        *(trp_connection_get_gssctx(conn)),
 402                                        &buf,
 403                                        &buflen)) {
 404     tr_debug("trps_read_message: error");
 405     if (buf)
 406       free(buf);
 407     return TRP_ERROR;
 408   }
 409
 410   tr_debug("trps_read_message: message received, %u bytes.", (unsigned) buflen);
 411   tr_debug("trps_read_message: %.*s", buflen, buf);
 412
 413   *msg=tr_msg_decode(buf, buflen);
 414   free(buf);
 415   if (*msg==NULL)
 416     return TRP_NOPARSE;
 417
 418   conn_peer=trp_connection_get_peer(conn);
 419   if (conn_peer==NULL) {
 420     tr_err("trps_read_message: connection has no peer name");
 421     return TRP_ERROR;
 422   }
 423
 424   peer=trps_get_peer_by_gssname(trps, conn_peer);
 425   if (peer==NULL) {
 426     tr_err("trps_read_message: could not find peer with gssname=%s", trp_connection_get_gssname(conn));
 427     return TRP_ERROR;
 428   }
 429
 430   /* verify we received a message we support, otherwise drop it now */
 431   switch (tr_msg_get_msg_type(*msg)) {
 432   case TRP_UPDATE:
 433     trp_upd_set_peer(tr_msg_get_trp_upd(*msg), tr_dup_name(conn_peer));
 434     trp_upd_set_next_hop(tr_msg_get_trp_upd(*msg), trp_peer_get_server(peer), 0); /* TODO: 0 should be the configured TID port */
 435     /* update provenance if necessary */
 436     trp_upd_add_to_provenance(tr_msg_get_trp_upd(*msg), trp_peer_get_label(peer));
 437     break;
 438
 439   case TRP_REQUEST:
 440     trp_req_set_peer(tr_msg_get_trp_req(*msg), tr_dup_name(conn_peer));
 441     break;
 442
 443   default:
 444     tr_debug("trps_read_message: received unsupported message from %.*s", conn_peer->len, conn_peer->buf);
 445     tr_msg_free_decoded(*msg);
 446     *msg=NULL;
 447     return TRP_UNSUPPORTED;
 448   }
 449
 450   return TRP_SUCCESS;
 451 }
 452
 453 int trps_get_listener(TRPS_INSTANCE *trps,
 454                       TRPS_MSG_FUNC msg_handler,
 455                       TRP_AUTH_FUNC auth_handler,
 456                       const char *hostname,
 457                       unsigned int port,
 458                       void *cookie,
 459                       int *fd_out,
 460                       size_t max_fd)
 461 {
 462   size_t n_fd=0;
 463   size_t ii=0;
 464
 465   n_fd=trps_listen(trps, port, fd_out, max_fd);
 466   if (n_fd==0)
 467     tr_debug("trps_get_listener: Error opening port %d.");
 468   else {
 469     /* opening port succeeded */
 470     tr_debug("trps_get_listener: Opened port %d.", port);
 471
 472     /* make the sockets non-blocking */
 473     for (ii=0; ii<n_fd; ii++) {
 474       if (0 != fcntl(fd_out[ii], F_SETFL, O_NONBLOCK)) {
 475         tr_debug("trps_get_listener: Error setting O_NONBLOCK.");
 476         for (ii=0; ii<n_fd; ii++) {
 477           close(fd_out[ii]);
 478           fd_out[ii]=-1;
 479         }
 480         n_fd=0;
 481         break;
 482       }
 483     }
 484   }
 485
 486   if (n_fd>0) {
 487     /* store the caller's request handler & cookie */
 488     trps->msg_handler = msg_handler;
 489     trps->auth_handler = auth_handler;
 490     trps->hostname = talloc_strdup(trps, hostname);
 491     trps->port = port;
 492     trps->cookie = cookie;
 493   }
 494
 495   return n_fd;
 496 }
 497
 498 TRP_RC trps_authorize_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *conn)
 499 {
 500   /* try to establish a GSS context */
 501   if (0!=trp_connection_auth(conn, trps->auth_handler, trps->cookie)) {
 502     tr_notice("trps_authorize_connection: failed to authorize connection");
 503     trp_connection_close(conn);
 504     return TRP_ERROR;
 505   }
 506   tr_notice("trps_authorize_connection: authorized connection");
 507   return TRP_SUCCESS;
 508 }
 509
 510 void trps_handle_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *conn)
 511 {
 512   TR_MSG *msg=NULL;
 513   TRP_RC rc=TRP_ERROR;
 514
 515   /* loop as long as the connection exists */
 516   while (trp_connection_get_status(conn)==TRP_CONNECTION_UP) {
 517     rc=trps_read_message(trps, conn, &msg);
 518     switch(rc) {
 519     case TRP_SUCCESS:
 520       trps->msg_handler(trps, conn, msg); /* send the TR_MSG off to the callback */
 521       break;
 522
 523     case TRP_ERROR:
 524       trp_connection_close(conn);
 525       break;
 526
 527     default:
 528       tr_debug("trps_handle_connection: trps_read_message failed (%d)", rc);
 529     }
 530   }
 531
 532   tr_debug("trps_handle_connection: connection closed.");
 533 }
 534
 535 /* TODO: check realm/comm, now part of the update instead of inforec */
 536 static TRP_RC trps_validate_update(TRPS_INSTANCE *trps, TRP_UPD *upd)
 537 {
 538   if (upd==NULL) {
 539     tr_notice("trps_validate_update: null TRP update.");
 540     return TRP_BADARG;
 541   }
 542
 543   if (trp_upd_get_realm(upd)==NULL) {
 544     tr_notice("trps_validate_update: received TRP update without realm.");
 545     return TRP_ERROR;
 546   }
 547
 548   if (trp_upd_get_comm(upd)==NULL) {
 549     tr_notice("trps_validate_update: received TRP update without community.");
 550     return TRP_ERROR;
 551   }
 552
 553   if (trp_upd_get_inforec(upd)==NULL) {
 554     tr_notice("trps_validate_update: received TRP update with no info records.");
 555     return TRP_ERROR;
 556   }
 557
 558   if (trp_upd_get_peer(upd)==NULL) {
 559     tr_notice("trps_validate_update: received TRP update without origin peer information.");
 560     return TRP_ERROR;
 561   }
 562
 563
 564   return TRP_SUCCESS;
 565 }
 566
 567 /* ensure that the update could be accepted if feasible */
 568 static TRP_RC trps_validate_inforec(TRPS_INSTANCE *trps, TRP_INFOREC *rec)
 569 {
 570   switch(trp_inforec_get_type(rec)) {
 571   case TRP_INFOREC_TYPE_ROUTE:
 572     if ((trp_inforec_get_trust_router(rec)==NULL)
 573        || (trp_inforec_get_next_hop(rec)==NULL)) {
 574       tr_debug("trps_validate_inforec: missing record info.");
 575       return TRP_ERROR;
 576     }
 577
 578     /* check for valid metric */
 579     if (trp_metric_is_invalid(trp_inforec_get_metric(rec))) {
 580       tr_debug("trps_validate_inforec: invalid metric (%u).", trp_inforec_get_metric(rec));
 581       return TRP_ERROR;
 582     }
 583
 584     /* check for valid interval */
 585     if (trp_inforec_get_interval(rec)==TRP_INTERVAL_INVALID) {
 586       tr_debug("trps_validate_inforec: invalid interval.");
 587       return TRP_ERROR;
 588     }
 589     break;
 590
 591   case TRP_INFOREC_TYPE_COMMUNITY:
 592     /* TODO: validate community updates */
 593     break;
 594
 595   default:
 596     tr_notice("trps_validate_inforec: unsupported record type.");
 597     return TRP_UNSUPPORTED;
 598   }
 599
 600   return TRP_SUCCESS;
 601 }
 602
 603 /* link cost to a peer */
 604 static unsigned int trps_cost(TRPS_INSTANCE *trps, TR_NAME *peer)
 605 {
 606   return 1;
 607 }
 608
 609 static unsigned int trps_advertised_metric(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer)
 610 {
 611   TRP_ROUTE *entry=trp_rtable_get_entry(trps->rtable, comm, realm, peer);
 612   if (entry==NULL)
 613     return TRP_METRIC_INFINITY;
 614   return trp_route_get_metric(entry) + trps_cost(trps, peer);
 615 }
 616
 617 static int trps_check_feasibility(TRPS_INSTANCE *trps, TR_NAME *realm, TR_NAME *comm, TRP_INFOREC *rec)
 618 {
 619   unsigned int rec_metric=trp_inforec_get_metric(rec);
 620   unsigned int new_metric=0;
 621   unsigned int current_metric=0;
 622   TR_NAME *next_hop=NULL;
 623
 624   /* we check these in the validation stage, but just in case... */
 625   if (trp_metric_is_invalid(rec_metric))
 626     return 0;
 627
 628   /* retractions (aka infinite metrics) are always feasible */
 629   if (trp_metric_is_infinite(rec_metric))
 630     return 1;
 631
 632   /* updates from our current next hop are always feasible*/
 633   next_hop=trps_get_next_hop(trps, comm, realm);
 634   if ((next_hop!=NULL)
 635      && (0==tr_name_cmp(next_hop,trp_inforec_get_next_hop(rec)))) {
 636     return 1;
 637   }
 638
 639
 640   /* compare the existing metric we advertise to what we would advertise
 641    * if we accept this update */
 642   current_metric=trps_advertised_metric(trps, comm, realm, trp_inforec_get_next_hop(rec));
 643   new_metric=rec_metric + trps_cost(trps, trp_inforec_get_next_hop(rec));
 644   if (new_metric <= current_metric)
 645     return 1;
 646   else
 647     return 0;
 648 }
 649
 650 /* uses memory pointed to by *ts, also returns that value. On error, its contents are {0,0} */
 651 static struct timespec *trps_compute_expiry(TRPS_INSTANCE *trps, unsigned int interval, struct timespec *ts)
 652 {
 653   const unsigned int small_factor=3; /* how many intervals we wait before expiring */
 654   if (0!=clock_gettime(TRP_CLOCK, ts)) {
 655     tr_err("trps_compute_expiry: could not read realtime clock.");
 656     ts->tv_sec=0;
 657     ts->tv_nsec=0;
 658   }
 659   tr_debug("trps_compute_expiry: tv_sec=%u, interval=%u, small_factor*interval=%u", ts->tv_sec, interval, small_factor*interval);
 660   ts->tv_sec += small_factor*interval;
 661   return ts;
 662 }
 663
 664 static TRP_RC trps_accept_update(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC *rec)
 665 {
 666   TRP_ROUTE *entry=NULL;
 667
 668   entry=trp_rtable_get_entry(trps->rtable,
 669                              trp_upd_get_comm(upd),
 670                              trp_upd_get_realm(upd),
 671                              trp_inforec_get_next_hop(rec));
 672   if (entry==NULL) {
 673     entry=trp_route_new(NULL);
 674     if (entry==NULL) {
 675       tr_err("trps_accept_update: unable to allocate new entry.");
 676       return TRP_NOMEM;
 677     }
 678
 679     trp_route_set_comm(entry, trp_upd_dup_comm(upd));
 680     trp_route_set_realm(entry, trp_upd_dup_realm(upd));
 681     trp_route_set_peer(entry, trp_upd_dup_peer(upd));
 682     trp_route_set_trust_router(entry, trp_inforec_dup_trust_router(rec));
 683     trp_route_set_next_hop(entry, trp_inforec_dup_next_hop(rec));
 684     /* TODO: pass next hop port (now defaults to TID_PORT) --jlr */
 685     if ((trp_route_get_comm(entry)==NULL)
 686        ||(trp_route_get_realm(entry)==NULL)
 687        ||(trp_route_get_peer(entry)==NULL)
 688        ||(trp_route_get_trust_router(entry)==NULL)
 689        ||(trp_route_get_next_hop(entry)==NULL)) {
 690       /* at least one field could not be allocated */
 691       tr_err("trps_accept_update: unable to allocate all fields for entry.");
 692       trp_route_free(entry);
 693       return TRP_NOMEM;
 694     }
 695     trp_rtable_add(trps->rtable, entry);
 696   }
 697
 698   /* We now have an entry in the table, whether it's new or not. Update metric and expiry, unless
 699    * the metric is infinity. An infinite metric can only occur here if we just retracted an existing
 700    * route (we never accept retractions as new routes), so there is no risk of leaving the expiry
 701    * time unset on a new route entry. */
 702   tr_debug("trps_accept_update: accepting route update.");
 703   trp_route_set_metric(entry, trp_inforec_get_metric(rec));
 704   trp_route_set_interval(entry, trp_inforec_get_interval(rec));
 705
 706   /* check whether the trust router has changed */
 707   if (0!=tr_name_cmp(trp_route_get_trust_router(entry),
 708                      trp_inforec_get_trust_router(rec))) {
 709     /* The name changed. Set this route as triggered. */
 710     tr_debug("trps_accept_update: trust router for route changed.");
 711     trp_route_set_triggered(entry, 1);
 712     trp_route_set_trust_router(entry, trp_inforec_dup_trust_router(rec)); /* frees old name */
 713   }
 714   if (!trps_route_retracted(trps, entry)) {
 715     tr_debug("trps_accept_update: route not retracted, setting expiry timer.");
 716     trp_route_set_expiry(entry, trps_compute_expiry(trps,
 717                                                      trp_route_get_interval(entry),
 718                                                      trp_route_get_expiry(entry)));
 719   }
 720   return TRP_SUCCESS;
 721 }
 722
 723
 724 static TRP_RC trps_handle_inforec_route(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC *rec)
 725 {
 726   TRP_ROUTE *route=NULL;
 727   unsigned int feas=0;
 728
 729   /* determine feasibility */
 730   feas=trps_check_feasibility(trps, trp_upd_get_realm(upd), trp_upd_get_comm(upd), rec);
 731   tr_debug("trps_handle_update: record feasibility=%d", feas);
 732
 733   /* do we have an existing route? */
 734   route=trps_get_route(trps,
 735                        trp_upd_get_comm(upd),
 736                        trp_upd_get_realm(upd),
 737                        trp_upd_get_peer(upd));
 738   if (route!=NULL) {
 739     /* there was a route table entry already */
 740     tr_debug("trps_handle_updates: route entry already exists.");
 741     if (feas) {
 742       /* Update is feasible. Accept it. */
 743       trps_accept_update(trps, upd, rec);
 744     } else {
 745       /* Update is infeasible. Ignore it unless the trust router has changed. */
 746       if (0!=tr_name_cmp(trp_route_get_trust_router(route),
 747                          trp_inforec_get_trust_router(rec))) {
 748         /* the trust router associated with the route has changed, treat update as a retraction */
 749         trps_retract_route(trps, route);
 750       }
 751     }
 752   } else {
 753     /* No existing route table entry. Ignore it unless it is feasible and not a retraction. */
 754     tr_debug("trps_handle_update: no route entry exists yet.");
 755     if (feas && trp_metric_is_finite(trp_inforec_get_metric(rec)))
 756       trps_accept_update(trps, upd, rec);
 757   }
 758
 759   return TRP_SUCCESS;
 760 }
 761
 762 static int trps_name_in_provenance(TR_NAME *name, json_t *prov)
 763 {
 764   size_t ii=0;
 765   TR_NAME *this_name=NULL;
 766   const char *s=NULL;
 767
 768   if (prov==NULL)
 769     return 0; /* no provenance list, so it has no names in it */
 770
 771   /* now check to see if name is in the provenance */
 772   for (ii=0; ii<json_array_size(prov); ii++) {
 773     s=json_string_value(json_array_get(prov, ii));
 774     if (s==NULL) {
 775       tr_debug("trps_name_in_provenance: empty entry in provenance list.");
 776       continue;
 777     }
 778
 779     this_name=tr_new_name(s);
 780     if (this_name==NULL) {
 781       tr_debug("trps_name_in_provenance: unable to allocate name.");
 782       return -1;
 783     }
 784     if (0==tr_name_cmp(name, this_name)) {
 785       tr_free_name(this_name);
 786       return 1;
 787     }
 788     tr_free_name(this_name);
 789   }
 790   return 0;
 791 }
 792
 793 static TR_COMM *trps_create_new_comm(TALLOC_CTX *mem_ctx, TR_NAME *comm_id, TRP_INFOREC *rec)
 794 {
 795   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 796   TR_COMM *comm=tr_comm_new(tmp_ctx);
 797
 798   if (comm==NULL) {
 799     tr_debug("trps_create_new_comm: unable to allocate new community.");
 800     goto cleanup;
 801   }
 802   /* fill in the community with info */
 803   tr_comm_set_id(comm, tr_dup_name(comm_id));
 804   if (tr_comm_get_id(comm)==NULL) {
 805     tr_debug("trps_create_new_comm: unable to allocate community name.");
 806     comm=NULL;
 807     goto cleanup;
 808   }
 809   tr_comm_set_type(comm, trp_inforec_get_comm_type(rec));
 810   if (trp_inforec_get_apcs(rec)!=NULL) {
 811     tr_comm_set_apcs(comm, tr_apc_dup(tmp_ctx, trp_inforec_get_apcs(rec)));
 812     if (tr_comm_get_apcs(comm)==NULL) {
 813       tr_debug("trps_create_new_comm: unable to allocate APC list.");
 814       comm=NULL;
 815       goto cleanup;
 816     }
 817   }
 818   if (trp_inforec_get_owner_realm(rec)!=NULL) {
 819     tr_comm_set_owner_realm(comm, tr_dup_name(trp_inforec_get_owner_realm(rec)));
 820     if (tr_comm_get_owner_realm(comm)==NULL) {
 821       tr_debug("trps_create_new_comm: unable to allocate owner realm name.");
 822       comm=NULL;
 823       goto cleanup;
 824     }
 825   }
 826   if (trp_inforec_get_owner_contact(rec)!=NULL) {
 827     tr_comm_set_owner_contact(comm, tr_dup_name(trp_inforec_get_owner_contact(rec)));
 828     if (tr_comm_get_owner_contact(comm)==NULL) {
 829       tr_debug("trps_create_new_comm: unable to allocate owner contact.");
 830       comm=NULL;
 831       goto cleanup;
 832     }
 833   }
 834   comm->expiration_interval=trp_inforec_get_exp_interval(rec);
 835   talloc_steal(mem_ctx, comm);
 836
 837 cleanup:
 838   talloc_free(tmp_ctx);
 839   return comm;
 840 }
 841
 842 static TR_RP_REALM *trps_create_new_rp_realm(TALLOC_CTX *mem_ctx, TR_NAME *realm_id, TRP_INFOREC *rec)
 843 {
 844   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 845   TR_RP_REALM *rp=tr_rp_realm_new(tmp_ctx);
 846
 847   if (rp==NULL) {
 848     tr_debug("trps_create_new_rp_realm: unable to allocate new realm.");
 849     goto cleanup;
 850   }
 851   /* fill in the realm */
 852   tr_rp_realm_set_id(rp, tr_dup_name(realm_id));
 853   if (tr_rp_realm_get_id(rp)==NULL) {
 854     tr_debug("trps_create_new_rp_realm: unable to allocate realm name.");
 855     rp=NULL;
 856     goto cleanup;
 857   }
 858   talloc_steal(mem_ctx, rp);
 859
 860 cleanup:
 861   talloc_free(tmp_ctx);
 862   return rp;
 863 }
 864
 865 static TR_IDP_REALM *trps_create_new_idp_realm(TALLOC_CTX *mem_ctx, TR_NAME *realm_id, TRP_INFOREC *rec)
 866 {
 867   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 868   TR_IDP_REALM *idp=tr_idp_realm_new(tmp_ctx);
 869
 870   if (idp==NULL) {
 871     tr_debug("trps_create_new_idp_realm: unable to allocate new realm.");
 872     goto cleanup;
 873   }
 874   /* fill in the realm */
 875   tr_idp_realm_set_id(idp, tr_dup_name(realm_id));
 876   if (tr_idp_realm_get_id(idp)==NULL) {
 877     tr_debug("trps_create_new_idp_realm: unable to allocate realm name.");
 878     idp=NULL;
 879     goto cleanup;
 880   }
 881   if (trp_inforec_get_apcs(rec)!=NULL) {
 882     tr_idp_realm_set_apcs(idp, tr_apc_dup(tmp_ctx, trp_inforec_get_apcs(rec)));
 883     if (tr_idp_realm_get_apcs(idp)==NULL) {
 884       tr_debug("trps_create_new_idp_realm: unable to allocate APC list.");
 885       idp=NULL;
 886       goto cleanup;
 887     }
 888   }
 889   idp->origin=TR_REALM_DISCOVERED;
 890
 891   talloc_steal(mem_ctx, idp);
 892
 893 cleanup:
 894   talloc_free(tmp_ctx);
 895   return idp;
 896 }
 897
 898 static TRP_RC trps_handle_inforec_comm(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC *rec)
 899 {
 900   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
 901   TR_NAME *comm_id=trp_upd_get_comm(upd);
 902   TR_NAME *realm_id=trp_upd_get_realm(upd);
 903   TR_NAME *origin_id=NULL;
 904   TR_NAME *our_peer_label=NULL;
 905   TR_COMM *comm=NULL;
 906   TR_RP_REALM *rp_realm=NULL;
 907   TR_IDP_REALM *idp_realm=NULL;
 908   struct timespec expiry={0,0};
 909   TRP_RC rc=TRP_ERROR;
 910
 911   if ((comm_id==NULL) || (realm_id==NULL))
 912     goto cleanup;
 913
 914   origin_id=trp_inforec_dup_origin(rec);
 915   if (origin_id==NULL)
 916     goto cleanup;
 917
 918   /* see whether we want to add this */
 919   our_peer_label=trps_dup_label(trps);
 920   if (our_peer_label==NULL) {
 921     tr_debug("trps_handle_inforec_comm: unable to allocate peer label.");
 922     goto cleanup;
 923   }
 924
 925   if (trps_name_in_provenance(our_peer_label, trp_inforec_get_provenance(rec)))
 926     tr_debug("trps_handle_inforec_comm: rejecting community inforec to avoid provenance loop.");
 927   else {
 928     /* no loop occurring, accept the update */
 929     comm=tr_comm_table_find_comm(trps->ctable, comm_id);
 930     if (comm==NULL) {
 931       tr_debug("trps_handle_inforec_comm: unknown community %.*s in inforec, creating it.",
 932                comm_id->len, comm_id->buf);
 933       comm=trps_create_new_comm(tmp_ctx, comm_id, rec);
 934       if (comm==NULL) {
 935         tr_debug("trps_handle_inforec_comm: unable to create new community.");
 936         goto cleanup;
 937       }
 938       tr_comm_table_add_comm(trps->ctable, comm);
 939     }
 940     /* TODO: see if other comm data match the new inforec and update or complain */
 941
 942     trps_compute_expiry(trps, trp_inforec_get_interval(rec), &expiry);
 943     if ((expiry.tv_sec==0)&&(expiry.tv_nsec==0))
 944       goto cleanup;
 945
 946     switch (trp_inforec_get_role(rec)) {
 947     case TR_ROLE_RP:
 948       rp_realm=tr_rp_realm_lookup(trps->ctable->rp_realms, realm_id);
 949       if (rp_realm==NULL) {
 950         tr_debug("trps_handle_inforec_comm: unknown RP realm %.*s in inforec, creating it.",
 951                  realm_id->len, realm_id->buf);
 952         rp_realm=trps_create_new_rp_realm(tmp_ctx, realm_id, rec);
 953         if (rp_realm==NULL) {
 954           tr_debug("trps_handle_inforec_comm: unable to create new RP realm.");
 955           /* we may leave an unused community in the table, but it will only last until
 956            * the next table sweep if it does not get any realms before that happens */
 957           goto cleanup;
 958         }
 959         tr_comm_table_add_rp_realm(trps->ctable, rp_realm);
 960       }
 961       /* TODO: if realm existed, see if data match the new inforec and update or complain */
 962       tr_comm_add_rp_realm(trps->ctable, comm, rp_realm, trp_inforec_get_interval(rec), trp_inforec_get_provenance(rec), &expiry);
 963       tr_debug("trps_handle_inforec_comm: added RP realm %.*s to comm %.*s (origin %.*s).",
 964                realm_id->len, realm_id->buf,
 965                comm_id->len, comm_id->buf,
 966                origin_id->len, origin_id->buf);
 967       break;
 968     case TR_ROLE_IDP:
 969       idp_realm=tr_idp_realm_lookup(trps->ctable->idp_realms, realm_id);
 970       if (idp_realm==NULL) {
 971         tr_debug("trps_handle_inforec_comm: unknown IDP realm %.*s in inforec, creating it.",
 972                  realm_id->len, realm_id->buf);
 973         idp_realm=trps_create_new_idp_realm(tmp_ctx, realm_id, rec);
 974         if (idp_realm==NULL) {
 975           tr_debug("trps_handle_inforec_comm: unable to create new IDP realm.");
 976           /* we may leave an unused community in the table, but it will only last until
 977            * the next table sweep if it does not get any realms before that happens */
 978           goto cleanup;
 979         }
 980         tr_comm_table_add_idp_realm(trps->ctable, idp_realm);
 981       }
 982       /* TODO: if realm existed, see if data match the new inforec and update or complain */
 983       tr_comm_add_idp_realm(trps->ctable, comm, idp_realm, trp_inforec_get_interval(rec), trp_inforec_get_provenance(rec), &expiry);
 984       tr_debug("trps_handle_inforec_comm: added IDP realm %.*s to comm %.*s (origin %.*s).",
 985                realm_id->len, realm_id->buf,
 986                comm_id->len, comm_id->buf,
 987                origin_id->len, origin_id->buf);
 988       break;
 989     default:
 990       tr_debug("trps_handle_inforec_comm: unable to add realm.");
 991       goto cleanup;
 992     }
 993   }
 994
 995   rc=TRP_SUCCESS;
 996
 997 cleanup:
 998   if (our_peer_label!=NULL)
 999     tr_free_name(our_peer_label);
1000   if (origin_id!=NULL)
1001     tr_free_name(origin_id);
1002   talloc_free(tmp_ctx);
1003   return rc;
1004 }
1005
1006 /**
1007  * Apply applicable TRP_INBOUND filters to an inforec. Rejects everything if peer has no filters.
1008  *
1009  * @param trps Active TRPS instance
1010  * @param peer_name Name of peer that sent this inforec
1011  * @param rec Inforec to filter
1012  * @return 1 if accepted by the filter, 0 otherwise
1013  */
1014 static int trps_filter_inbound_inforec(TRPS_INSTANCE *trps, TR_NAME *peer_name, TRP_INFOREC *rec)
1015 {
1016   TRP_PEER *peer=NULL;
1017   TR_FILTER_ACTION action=TR_FILTER_ACTION_REJECT;
1018
1019   /* Look up the peer. For inbound messages, the peer is identified by its GSS name */
1020   peer=trps_get_peer_by_gssname(trps, peer_name);
1021   if (peer==NULL) {
1022     tr_err("trps_filter_inbound_inforec: received inforec from unknown peer (%.*s), rejecting.",
1023            peer_name->len,
1024            peer_name->buf);
1025     return 0;
1026   }
1027
1028   /* tr_filter_apply() and tr_filter_set_get() handle null filter sets/filters by rejecting */
1029   if ((TR_FILTER_NO_MATCH==tr_filter_apply(rec,
1030                                            tr_filter_set_get(peer->filters, TR_FILTER_TYPE_TRP_INBOUND),
1031                                            NULL,
1032                                            &action))
1033       || (action!=TR_FILTER_ACTION_ACCEPT)) {
1034     /* either the filter did not match or it matched a reject rule */
1035     return 0;
1036   }
1037
1038   /* filter matched an accept rule */
1039   return 1;
1040 }
1041
1042
1043 static TRP_RC trps_handle_update(TRPS_INSTANCE *trps, TRP_UPD *upd)
1044 {
1045   TRP_INFOREC *rec=NULL;
1046
1047   if (trps_validate_update(trps, upd) != TRP_SUCCESS) {
1048     tr_notice("trps_handle_update: received invalid TRP update.");
1049     return TRP_ERROR;
1050   }
1051
1052   for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
1053     /* validate/sanity check the record update */
1054     if (trps_validate_inforec(trps, rec) != TRP_SUCCESS) {
1055       tr_notice("trps_handle_update: invalid inforec in TRP update, discarding entire update.");
1056       return TRP_ERROR;
1057     }
1058   }
1059
1060   for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
1061     if (!trps_filter_inbound_inforec(trps, trp_upd_get_peer(upd), rec)) {
1062       tr_debug("trps_handle_update: inforec rejected by filter.");
1063       continue; /* just go on to the next record */
1064     }
1065
1066     switch (trp_inforec_get_type(rec)) {
1067     case TRP_INFOREC_TYPE_ROUTE:
1068       tr_debug("trps_handle_update: handling route inforec.");
1069       if (TRP_SUCCESS!=trps_handle_inforec_route(trps, upd, rec))
1070         tr_notice("trps_handle_update: error handling route inforec.");
1071       break;
1072     case TRP_INFOREC_TYPE_COMMUNITY:
1073       tr_debug("trps_handle_update: handling community inforec.");
1074       if (TRP_SUCCESS!=trps_handle_inforec_comm(trps, upd, rec))
1075         tr_notice("trps_handle_update: error handling community inforec.");
1076
1077       break;
1078     default:
1079       tr_notice("trps_handle_update: unsupported inforec in TRP update.");
1080       break;
1081     }
1082   }
1083   return TRP_SUCCESS;
1084 }
1085
1086 static TRP_RC trps_validate_request(TRPS_INSTANCE *trps, TRP_REQ *req)
1087 {
1088   if (req==NULL) {
1089     tr_notice("trps_validate_request: null TRP request.");
1090     return TRP_BADARG;
1091   }
1092
1093   if (trp_req_get_comm(req)==NULL) {
1094     tr_notice("trps_validate_request: received TRP request with null community.");
1095     return TRP_ERROR;
1096   }
1097
1098   if (trp_req_get_realm(req)==NULL) {
1099     tr_notice("trps_validate_request: received TRP request with null realm.");
1100     return TRP_ERROR;
1101   }
1102
1103   if (trp_req_get_peer(req)==NULL) {
1104     tr_notice("trps_validate_request: received TRP request without origin peer information.");
1105     return TRP_ERROR;
1106   }
1107
1108   return TRP_SUCCESS;
1109 }
1110
1111 /* choose the best route to comm/realm, optionally excluding routes to a particular peer */
1112 static TRP_ROUTE *trps_find_best_route(TRPS_INSTANCE *trps,
1113                                         TR_NAME *comm,
1114                                         TR_NAME *realm,
1115                                         TR_NAME *exclude_peer)
1116 {
1117   TRP_ROUTE **entry=NULL;
1118   TRP_ROUTE *best=NULL;
1119   size_t n_entry=0;
1120   unsigned int kk=0;
1121   unsigned int kk_min=0;
1122   unsigned int min_metric=TRP_METRIC_INFINITY;
1123
1124   entry=trp_rtable_get_realm_entries(trps->rtable, comm, realm, &n_entry);
1125   for (kk=0; kk<n_entry; kk++) {
1126     if (trp_route_get_metric(entry[kk]) < min_metric) {
1127       if ((exclude_peer==NULL) || (0!=tr_name_cmp(trp_route_get_peer(entry[kk]),
1128                                                   exclude_peer))) {
1129         kk_min=kk;
1130         min_metric=trp_route_get_metric(entry[kk]);
1131       }
1132     }
1133   }
1134   if (trp_metric_is_finite(min_metric))
1135     best=entry[kk_min];
1136
1137   talloc_free(entry);
1138   return best;
1139 }
1140
1141 /* TODO: think this through more carefully. At least ought to add hysteresis
1142  * to avoid flapping between routers or routes. */
1143 TRP_RC trps_update_active_routes(TRPS_INSTANCE *trps)
1144 {
1145   size_t n_comm=0, ii=0;
1146   TR_NAME **comm=trp_rtable_get_comms(trps->rtable, &n_comm);
1147   size_t n_realm=0, jj=0;
1148   TR_NAME **realm=NULL;
1149   TRP_ROUTE *best_route=NULL, *cur_route=NULL;
1150   unsigned int best_metric=0, cur_metric=0;
1151
1152   for (ii=0; ii<n_comm; ii++) {
1153     realm=trp_rtable_get_comm_realms(trps->rtable, comm[ii], &n_realm);
1154     for (jj=0; jj<n_realm; jj++) {
1155       best_route=trps_find_best_route(trps, comm[ii], realm[jj], NULL);
1156       if (best_route==NULL)
1157         best_metric=TRP_METRIC_INFINITY;
1158       else
1159         best_metric=trp_route_get_metric(best_route);
1160
1161       cur_route=trps_get_selected_route(trps, comm[ii], realm[jj]);
1162       if (cur_route!=NULL) {
1163         cur_metric=trp_route_get_metric(cur_route);
1164         if ((best_metric < cur_metric) && (trp_metric_is_finite(best_metric))) {
1165           /* The new route has a lower metric than the previous, and is finite. Accept. */
1166           trp_route_set_selected(cur_route, 0);
1167           trp_route_set_selected(best_route, 1);
1168         } else if (!trp_metric_is_finite(cur_metric)) /* rejects infinite or invalid metrics */
1169           trp_route_set_selected(cur_route, 0);
1170       } else if (trp_metric_is_finite(best_metric)) {
1171         trp_route_set_selected(best_route, 1);
1172       }
1173     }
1174     if (realm!=NULL)
1175       talloc_free(realm);
1176     realm=NULL; n_realm=0;
1177   }
1178   if (comm!=NULL)
1179     talloc_free(comm);
1180   comm=NULL; n_comm=0;
1181
1182   return TRP_SUCCESS;
1183 }
1184
1185 /* true if curtime >= expiry */
1186 static int trps_expired(struct timespec *expiry, struct timespec *curtime)
1187 {
1188   return (tr_cmp_timespec(curtime, expiry) >= 0);
1189 }
1190
1191 /* Sweep for expired routes. For each expired route, if its metric is infinite, the route is flushed.
1192  * If its metric is finite, the metric is set to infinite and the route's expiration time is updated. */
1193 TRP_RC trps_sweep_routes(TRPS_INSTANCE *trps)
1194 {
1195   struct timespec sweep_time={0,0};
1196   TRP_ROUTE **entry=NULL;
1197   size_t n_entry=0;
1198   size_t ii=0;
1199
1200   /* use a single time for the entire sweep */
1201   if (0!=clock_gettime(TRP_CLOCK, &sweep_time)) {
1202     tr_err("trps_sweep_routes: could not read realtime clock.");
1203     sweep_time.tv_sec=0;
1204     sweep_time.tv_nsec=0;
1205     return TRP_ERROR;
1206   }
1207
1208   entry=trp_rtable_get_entries(trps->rtable, &n_entry); /* must talloc_free *entry */
1209
1210   /* loop over the entries */
1211   for (ii=0; ii<n_entry; ii++) {
1212     if (!trp_route_is_local(entry[ii]) && trps_expired(trp_route_get_expiry(entry[ii]), &sweep_time)) {
1213       tr_debug("trps_sweep_routes: route expired.");
1214       if (!trp_metric_is_finite(trp_route_get_metric(entry[ii]))) {
1215         /* flush route */
1216         tr_debug("trps_sweep_routes: metric was infinity, flushing route.");
1217         trp_rtable_remove(trps->rtable, entry[ii]); /* entry[ii] is no longer valid */
1218         entry[ii]=NULL;
1219       } else {
1220         /* set metric to infinity and reset timer */
1221         tr_debug("trps_sweep_routes: setting metric to infinity and resetting expiry.");
1222         trp_route_set_metric(entry[ii], TRP_METRIC_INFINITY);
1223         trp_route_set_expiry(entry[ii], trps_compute_expiry(trps,
1224                                                              trp_route_get_interval(entry[ii]),
1225                                                              trp_route_get_expiry(entry[ii])));
1226       }
1227     }
1228   }
1229
1230   talloc_free(entry);
1231   return TRP_SUCCESS;
1232 }
1233
1234
1235 static char *timespec_to_str(struct timespec *ts)
1236 {
1237   struct tm tm;
1238   char *s=NULL;
1239
1240   if (localtime_r(&(ts->tv_sec), &tm)==NULL)
1241     return NULL;
1242
1243   s=malloc(40); /* long enough to contain strftime result */
1244   if (s==NULL)
1245     return NULL;
1246
1247   if (strftime(s, 40, "%F %T", &tm)==0) {
1248     free(s);
1249     return NULL;
1250   }
1251   return s;
1252 }
1253
1254
1255 /* Sweep for expired communities/realms/memberships. */
1256 TRP_RC trps_sweep_ctable(TRPS_INSTANCE *trps)
1257 {
1258   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1259   struct timespec sweep_time={0,0};
1260   TR_COMM_MEMB *memb=NULL;
1261   TR_COMM_ITER *iter=NULL;
1262   TRP_RC rc=TRP_ERROR;
1263
1264   /* use a single time for the entire sweep */
1265   if (0!=clock_gettime(TRP_CLOCK, &sweep_time)) {
1266     tr_err("trps_sweep_ctable: could not read realtime clock.");
1267     sweep_time.tv_sec=0;
1268     sweep_time.tv_nsec=0;
1269     goto cleanup;
1270   }
1271
1272   /* iterate all memberships */
1273   iter=tr_comm_iter_new(tmp_ctx);
1274   if (iter==NULL) {
1275     tr_err("trps_sweep_ctable: unable to allocate iterator.");
1276     rc=TRP_NOMEM;
1277     goto cleanup;
1278   }
1279   for (memb=tr_comm_memb_iter_all_first(iter, trps->ctable);
1280        memb!=NULL;
1281        memb=tr_comm_memb_iter_all_next(iter)) {
1282     if (tr_comm_memb_get_origin(memb)==NULL)
1283       continue; /* do not expire local entries */
1284
1285     if (tr_comm_memb_is_expired(memb, &sweep_time)) {
1286       if (tr_comm_memb_get_times_expired(memb)>0) {
1287         /* Already expired once; flush. */
1288         tr_debug("trps_sweep_ctable: flushing expired community membership (%.*s in %.*s, origin %.*s, expired %s).",
1289                  tr_comm_memb_get_realm_id(memb)->len, tr_comm_memb_get_realm_id(memb)->buf,
1290                  tr_comm_get_id(tr_comm_memb_get_comm(memb))->len, tr_comm_get_id(tr_comm_memb_get_comm(memb))->buf,
1291                  tr_comm_memb_get_origin(memb)->len, tr_comm_memb_get_origin(memb)->buf,
1292                  timespec_to_str(tr_comm_memb_get_expiry(memb)));
1293         tr_comm_table_remove_memb(trps->ctable, memb);
1294         tr_comm_memb_free(memb);
1295       } else {
1296         /* This is the first expiration. Note this and reset the expiry time. */
1297         tr_comm_memb_expire(memb);
1298         trps_compute_expiry(trps, tr_comm_memb_get_interval(memb), tr_comm_memb_get_expiry(memb));
1299         tr_debug("trps_sweep_ctable: community membership expired at %s, resetting expiry to %s (%.*s in %.*s, origin %.*s).",
1300                  timespec_to_str(&sweep_time),
1301                  timespec_to_str(tr_comm_memb_get_expiry(memb)),
1302                  tr_comm_memb_get_realm_id(memb)->len, tr_comm_memb_get_realm_id(memb)->buf,
1303                  tr_comm_get_id(tr_comm_memb_get_comm(memb))->len, tr_comm_get_id(tr_comm_memb_get_comm(memb))->buf,
1304                  tr_comm_memb_get_origin(memb)->len, tr_comm_memb_get_origin(memb)->buf);
1305       }
1306     }
1307   }
1308
1309   /* get rid of any unreferenced realms, etc */
1310   tr_comm_table_sweep(trps->ctable);
1311
1312 cleanup:
1313   talloc_free(tmp_ctx);
1314   return rc;
1315 }
1316
1317 /* add metrics */
1318 static unsigned int trps_metric_add(unsigned int m1, unsigned int m2)
1319 {
1320   if (trp_metric_is_invalid(m1) || trp_metric_is_invalid(m2))
1321     return TRP_METRIC_INVALID;
1322
1323   if (trp_metric_is_infinite(m1) || trp_metric_is_infinite(m2))
1324     return TRP_METRIC_INFINITY;
1325
1326   if (trp_metric_is_finite(m1+m2))
1327     return m1+m2;
1328   else
1329     return TRP_METRIC_INFINITY;
1330 }
1331
1332 /* convert an rentry into a new trp update info record */
1333 static TRP_INFOREC *trps_route_to_inforec(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TRP_ROUTE *route)
1334 {
1335   TRP_INFOREC *rec=trp_inforec_new(mem_ctx, TRP_INFOREC_TYPE_ROUTE);
1336   unsigned int linkcost=0;
1337
1338   if (rec!=NULL) {
1339     if (trp_route_is_local(route))
1340       linkcost=0;
1341     else {
1342       linkcost=trp_peer_get_linkcost(trps_get_peer_by_gssname(trps,
1343                                                               trp_route_get_peer(route)));
1344     }
1345
1346     /* Note that we leave the next hop empty since the recipient fills that in.
1347      * This is where we add the link cost (currently always 1) to the next peer. */
1348     if ((trp_inforec_set_trust_router(rec, trp_route_dup_trust_router(route)) != TRP_SUCCESS)
1349        ||(trp_inforec_set_metric(rec,
1350                                  trps_metric_add(trp_route_get_metric(route),
1351                                                  linkcost)) != TRP_SUCCESS)
1352        ||(trp_inforec_set_interval(rec, trps_get_update_interval(trps)) != TRP_SUCCESS)) {
1353       tr_err("trps_route_to_inforec: error creating route update.");
1354       talloc_free(rec);
1355       rec=NULL;
1356     }
1357   }
1358   return rec;
1359 }
1360
1361 static TRP_UPD *trps_route_to_upd(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TRP_ROUTE *route)
1362 {
1363   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1364   TRP_UPD *upd=trp_upd_new(tmp_ctx);
1365   TRP_INFOREC *rec=NULL;
1366
1367   if (upd==NULL) {
1368     tr_err("trps_route_to_upd: could not create update message.");
1369     goto cleanup;
1370   }
1371   trp_upd_set_realm(upd, trp_route_dup_realm(route));
1372   if (trp_upd_get_realm(upd)==NULL) {
1373     tr_err("trps_route_to_upd: could not copy realm.");
1374     upd=NULL; /* it's still in tmp_ctx, so it will be freed */
1375     goto cleanup;
1376   }
1377   trp_upd_set_comm(upd, trp_route_dup_comm(route));
1378   if (trp_upd_get_comm(upd)==NULL) {
1379     tr_err("trps_route_to_upd: could not copy comm.");
1380     upd=NULL; /* it's still in tmp_ctx, so it will be freed */
1381     goto cleanup;
1382   }
1383   rec=trps_route_to_inforec(tmp_ctx, trps, route);
1384   if (rec==NULL) {
1385     tr_err("trps_route_to_upd: could not create route info record for realm %.*s in comm %.*s.",
1386            trp_route_get_realm(route)->len, trp_route_get_realm(route)->buf,
1387            trp_route_get_comm(route)->len, trp_route_get_comm(route)->buf);
1388     upd=NULL; /* it's till in tmp_ctx, so it will be freed */
1389     goto cleanup;
1390   }
1391   trp_upd_add_inforec(upd, rec);
1392
1393   /* sucess */
1394   talloc_steal(mem_ctx, upd);
1395
1396 cleanup:
1397   talloc_free(tmp_ctx);
1398   return upd;
1399 }
1400
1401 /* select the correct route to comm/realm to be announced to peer */
1402 static TRP_ROUTE *trps_select_realm_update(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer_gssname)
1403 {
1404   TRP_ROUTE *route;
1405
1406   /* Take the currently selected route unless it is through the peer we're sending the update to.
1407    * I.e., enforce the split horizon rule. */
1408   route=trp_rtable_get_selected_entry(trps->rtable, comm, realm);
1409   if (route==NULL) {
1410     /* No selected route, this should only happen if the only route has been retracted,
1411      * in which case we do not want to advertise it. */
1412     return NULL;
1413   }
1414   tr_debug("trps_select_realm_update: %s vs %s", peer_gssname->buf,
1415            trp_route_get_peer(route)->buf);
1416   if (0==tr_name_cmp(peer_gssname, trp_route_get_peer(route))) {
1417     tr_debug("trps_select_realm_update: matched, finding alternate route");
1418     /* the selected entry goes through the peer we're reporting to, choose an alternate */
1419     route=trps_find_best_route(trps, comm, realm, peer_gssname);
1420     if ((route==NULL) || (!trp_metric_is_finite(trp_route_get_metric(route))))
1421       return NULL; /* don't advertise a nonexistent or retracted route */
1422   }
1423   return route;
1424 }
1425
1426 /* Add TRP_UPD msgs to the updates GPtrArray. Caller needs to arrange for these to be freed. */
1427 static TRP_RC trps_select_route_updates_for_peer(TALLOC_CTX *mem_ctx,
1428                                                  GPtrArray *updates,
1429                                                  TRPS_INSTANCE *trps,
1430                                                  TR_NAME *peer_gssname,
1431                                                  int triggered)
1432 {
1433   size_t n_comm=0;
1434   TR_NAME **comm=trp_rtable_get_comms(trps->rtable, &n_comm);
1435   TR_NAME **realm=NULL;
1436   size_t n_realm=0;
1437   size_t ii=0, jj=0;
1438   TRP_ROUTE *best=NULL;
1439   TRP_UPD *upd=NULL;
1440
1441   if (updates==NULL)
1442     return TRP_BADARG;
1443
1444   for (ii=0; ii<n_comm; ii++) {
1445     realm=trp_rtable_get_comm_realms(trps->rtable, comm[ii], &n_realm);
1446     for (jj=0; jj<n_realm; jj++) {
1447       best=trps_select_realm_update(trps, comm[ii], realm[jj], peer_gssname);
1448       /* If we found a route, add it to the list. If triggered!=0, then only
1449        * add triggered routes. */
1450       if ((best!=NULL) && ((!triggered) || trp_route_is_triggered(best))) {
1451         upd=trps_route_to_upd(mem_ctx, trps, best);
1452         if (upd==NULL) {
1453           tr_err("trps_select_route_updates_for_peer: unable to create update message.");
1454           continue;
1455         }
1456         g_ptr_array_add(updates, upd);
1457       }
1458     }
1459
1460     if (realm!=NULL)
1461       talloc_free(realm);
1462     realm=NULL;
1463     n_realm=0;
1464   }
1465
1466   if (comm!=NULL)
1467     talloc_free(comm);
1468
1469   return TRP_SUCCESS;
1470 }
1471
1472 static TRP_INFOREC *trps_memb_to_inforec(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TR_COMM_MEMB *memb)
1473 {
1474   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1475   TRP_INFOREC *rec=NULL;
1476   TR_COMM *comm=NULL;
1477
1478   if (memb==NULL)
1479     goto cleanup;
1480
1481   comm=tr_comm_memb_get_comm(memb);
1482   rec=trp_inforec_new(tmp_ctx, TRP_INFOREC_TYPE_COMMUNITY);
1483   if (rec==NULL)
1484     goto cleanup;
1485
1486   if (TRP_SUCCESS!=trp_inforec_set_comm_type(rec, tr_comm_get_type(comm))) {
1487     rec=NULL;
1488     goto cleanup;
1489   }
1490
1491   if (TRP_SUCCESS!=trp_inforec_set_role(rec, tr_comm_memb_get_role(memb))) {
1492     rec=NULL;
1493     goto cleanup;
1494   }
1495
1496   if ((NULL!=tr_comm_get_apcs(comm)) &&
1497       ( (TRP_SUCCESS!=trp_inforec_set_apcs(rec,
1498                                            tr_apc_dup(rec, tr_comm_get_apcs(comm)))) ||
1499         (NULL==trp_inforec_get_apcs(rec)))) {
1500     rec=NULL;
1501     goto cleanup;
1502   }
1503
1504   if ((NULL!=tr_comm_get_owner_realm(comm)) &&
1505       ( (TRP_SUCCESS!=trp_inforec_set_owner_realm(rec, tr_dup_name(tr_comm_get_owner_realm(comm)))) ||
1506         (NULL==trp_inforec_get_owner_realm(rec)))) {
1507     rec=NULL;
1508     goto cleanup;
1509   }
1510
1511   if ((NULL!=tr_comm_get_owner_contact(comm)) &&
1512       ( (TRP_SUCCESS!=trp_inforec_set_owner_contact(rec, tr_dup_name(tr_comm_get_owner_contact(comm)))) ||
1513         (NULL==trp_inforec_get_owner_contact(rec)))) {
1514     rec=NULL;
1515     goto cleanup;
1516   }
1517
1518   if ((NULL!=tr_comm_memb_get_provenance(memb)) &&
1519       (TRP_SUCCESS!=trp_inforec_set_provenance(rec, tr_comm_memb_get_provenance(memb)))) {
1520     rec=NULL;
1521     goto cleanup;
1522   }
1523
1524   if (TRP_SUCCESS!=trp_inforec_set_interval(rec, trps_get_update_interval(trps))) {
1525     rec=NULL;
1526     goto cleanup;
1527   }
1528
1529   /* success! */
1530   talloc_steal(mem_ctx, rec);
1531
1532 cleanup:
1533   talloc_free(tmp_ctx);
1534   return rec;
1535 }
1536
1537 /* construct an update with all the inforecs for comm/realm/role to be sent to peer */
1538 static TRP_UPD *trps_comm_update(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TR_NAME *peer_gssname, TR_COMM *comm, TR_REALM *realm)
1539 {
1540   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1541   TRP_UPD *upd=trp_upd_new(tmp_ctx);
1542   TRP_INFOREC *rec=NULL;
1543   TR_COMM_ITER *iter=NULL;
1544   TR_COMM_MEMB *memb=NULL;
1545
1546   if (upd==NULL)
1547     goto cleanup;
1548
1549   trp_upd_set_comm(upd, tr_comm_dup_id(comm));
1550   trp_upd_set_realm(upd, tr_realm_dup_id(realm));
1551   /* leave peer empty */
1552
1553   iter=tr_comm_iter_new(tmp_ctx);
1554   if (iter==NULL) {
1555     tr_err("trps_comm_update: unable to allocate iterator.");
1556     upd=NULL;
1557     goto cleanup;
1558   }
1559
1560   /* now add inforecs */
1561   switch (realm->role) {
1562   case TR_ROLE_IDP:
1563     memb=tr_comm_table_find_idp_memb(trps->ctable,
1564                                      tr_realm_get_id(realm),
1565                                      tr_comm_get_id(comm));
1566     break;
1567   case TR_ROLE_RP:
1568     memb=tr_comm_table_find_rp_memb(trps->ctable,
1569                                     tr_realm_get_id(realm),
1570                                     tr_comm_get_id(comm));
1571     break;
1572   default:
1573     break;
1574   }
1575   if (memb!=NULL) {
1576     for (memb=tr_comm_memb_iter_first(iter, memb);
1577          memb!=NULL;
1578          memb=tr_comm_memb_iter_next(iter)) {
1579       rec=trps_memb_to_inforec(tmp_ctx, trps, memb);
1580       if (rec==NULL) {
1581         tr_err("trps_comm_update: unable to allocate inforec.");
1582         upd=NULL;
1583         goto cleanup;
1584       }
1585       trp_upd_add_inforec(upd, rec);
1586     }
1587   }
1588
1589   if (trp_upd_get_inforec(upd)==NULL)
1590     upd=NULL; /* no inforecs, no reason to send the update */
1591   else
1592     talloc_steal(mem_ctx, upd); /* success! */
1593
1594 cleanup:
1595   talloc_free(tmp_ctx);
1596   return upd;
1597 }
1598
1599 /* Find all community updates to send to a peer and add these as TR_UPD records
1600  * to the updates GPtrArray. */
1601 static TRP_RC trps_select_comm_updates_for_peer(TALLOC_CTX *mem_ctx,
1602                                                 GPtrArray *updates,
1603                                                 TRPS_INSTANCE *trps,
1604                                                 TR_NAME *peer_gssname,
1605                                                 int triggered)
1606 {
1607   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1608   TR_COMM_ITER *comm_iter=NULL;
1609   TR_COMM *comm=NULL;
1610   TR_COMM_ITER *realm_iter=NULL;
1611   TR_REALM *realm=NULL;
1612   TRP_UPD *upd=NULL;
1613   TRP_RC rc=TRP_ERROR;
1614
1615   /* currently do not send any communities on triggered updates */
1616   if (triggered) {
1617     rc=TRP_SUCCESS;
1618     goto cleanup;
1619   }
1620
1621   comm_iter=tr_comm_iter_new(tmp_ctx);
1622   realm_iter=tr_comm_iter_new(tmp_ctx);
1623   if ((comm_iter==NULL) || (realm_iter==NULL)) {
1624     tr_err("trps_select_comm_updates_for_peer: unable to allocate iterator.");
1625     rc=TRP_NOMEM;
1626     goto cleanup;
1627   }
1628
1629   /* do every community */
1630   for (comm=tr_comm_table_iter_first(comm_iter, trps->ctable);
1631        comm!=NULL;
1632        comm=tr_comm_table_iter_next(comm_iter)) {
1633     /* do every realm in this community */
1634     tr_debug("trps_select_comm_updates_for_peer: looking through community %.*s",
1635              tr_comm_get_id(comm)->len,
1636              tr_comm_get_id(comm)->buf);
1637     for (realm=tr_realm_iter_first(realm_iter, trps->ctable, tr_comm_get_id(comm));
1638          realm!=NULL;
1639          realm=tr_realm_iter_next(realm_iter)) {
1640       /* get the update for this comm/realm */
1641       tr_debug("trps_select_comm_updates_for_peer: adding realm %.*s",
1642                tr_realm_get_id(realm)->len,
1643                tr_realm_get_id(realm)->buf);
1644       upd=trps_comm_update(mem_ctx, trps, peer_gssname, comm, realm);
1645       if (upd!=NULL)
1646         g_ptr_array_add(updates, upd);
1647     }
1648   }
1649
1650 cleanup:
1651   talloc_free(tmp_ctx);
1652   return rc;
1653 }
1654
1655
1656 /* helper for trps_update_one_peer. Frees the TRP_UPD pointed to by a GPtrArray element */
1657 static void trps_trp_upd_destroy(gpointer data)
1658 {
1659   trp_upd_free((TRP_UPD *)data);
1660 }
1661
1662 /* all routes/communities to a single peer, unless comm/realm are specified (both or neither must be NULL) */
1663 static TRP_RC trps_update_one_peer(TRPS_INSTANCE *trps,
1664                                    TRP_PEER *peer,
1665                                    TRP_UPDATE_TYPE update_type,
1666                                    TR_NAME *comm,
1667                                    TR_NAME *realm)
1668 {
1669   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1670   TR_MSG msg; /* not a pointer! */
1671   TRP_UPD *upd=NULL;
1672   TRP_ROUTE *route=NULL;
1673   size_t ii=0;
1674   char *encoded=NULL;
1675   TRP_RC rc=TRP_ERROR;
1676   TR_NAME *peer_label=trp_peer_get_label(peer);
1677   GPtrArray *updates=g_ptr_array_new_with_free_func(trps_trp_upd_destroy);
1678
1679   if (updates==NULL) {
1680     tr_err("trps_update_one_peer: unable to allocate updates array.");
1681     rc=TRP_NOMEM;
1682     goto cleanup;
1683   }
1684
1685   switch (update_type) {
1686   case TRP_UPDATE_TRIGGERED:
1687     tr_debug("trps_update_one_peer: preparing triggered update for %.*s",
1688              peer_label->len, peer_label->buf);
1689     break;
1690   case TRP_UPDATE_SCHEDULED:
1691     tr_debug("trps_update_one_peer: preparing scheduled update for %.*s",
1692              peer_label->len, peer_label->buf);
1693     break;
1694   case TRP_UPDATE_REQUESTED:
1695     tr_debug("trps_update_one_peer: preparing requested update for %.*s",
1696              peer_label->len, peer_label->buf);
1697     break;
1698   default:
1699     tr_err("trps_update_one_peer: invalid update type requested.");
1700     rc=TRP_BADARG;
1701     goto cleanup;
1702   }
1703
1704   /* First, gather route updates. */
1705   tr_debug("trps_update_one_peer: selecting route updates for %.*s.", peer_label->len, peer_label->buf);
1706   if ((comm==NULL) && (realm==NULL)) {
1707     /* do all realms */
1708     rc=trps_select_route_updates_for_peer(tmp_ctx,
1709                                           updates,
1710                                           trps,
1711                                           peer_label,
1712                                           update_type==TRP_UPDATE_TRIGGERED);
1713   } else if ((comm!=NULL) && (realm!=NULL)) {
1714     /* a single community/realm was requested */
1715     route=trps_select_realm_update(trps, comm, realm, peer_label);
1716     if (route==NULL) {
1717       /* we have no actual update to send back, MUST send a retraction */
1718       tr_debug("trps_update_one_peer: community/realm without route requested, sending mandatory retraction.");
1719       route=trp_route_new(tmp_ctx);
1720       trp_route_set_comm(route, tr_dup_name(comm));
1721       trp_route_set_realm(route, tr_dup_name(realm));
1722       trp_route_set_peer(route, tr_new_name(""));
1723       trp_route_set_metric(route, TRP_METRIC_INFINITY);
1724       trp_route_set_trust_router(route, tr_new_name(""));
1725       trp_route_set_next_hop(route, tr_new_name(""));
1726     }
1727     upd=trps_route_to_upd(tmp_ctx, trps, route);
1728     if (upd==NULL) {
1729       tr_err("trps_update_one_peer: unable to allocate route update.");
1730       rc=TRP_NOMEM;
1731       goto cleanup;
1732     }
1733     g_ptr_array_add(updates, upd);
1734   } else {
1735     tr_err("trps_update_one_peer: error: only comm or realm was specified. Need both or neither.");
1736     rc=TRP_ERROR;
1737     goto cleanup;
1738   }
1739
1740   /* Second, gather community updates */
1741   tr_debug("trps_update_one_peer: selecting community updates for %.*s.", peer_label->len, peer_label->buf);
1742   rc=trps_select_comm_updates_for_peer(tmp_ctx, updates, trps, peer_label, update_type==TRP_UPDATE_TRIGGERED);
1743
1744   /* see if we have anything to send */
1745   if (updates->len<=0)
1746     tr_debug("trps_update_one_peer: no updates for %.*s", peer_label->len, peer_label->buf);
1747   else {
1748     tr_debug("trps_update_one_peer: sending %d update messages.", updates->len);
1749     for (ii=0; ii<updates->len; ii++) {
1750       upd=(TRP_UPD *)g_ptr_array_index(updates, ii);
1751       /* now encode the update message */
1752       tr_msg_set_trp_upd(&msg, upd);
1753       encoded=tr_msg_encode(&msg);
1754       if (encoded==NULL) {
1755         tr_err("trps_update_one_peer: error encoding update.");
1756         rc=TRP_ERROR;
1757         goto cleanup;
1758       }
1759
1760       tr_debug("trps_update_one_peer: adding message to queue.");
1761       if (trps_send_msg(trps, peer, encoded) != TRP_SUCCESS)
1762         tr_err("trps_update_one_peer: error queueing update.");
1763       else
1764         tr_debug("trps_update_one_peer: update queued successfully.");
1765
1766       tr_msg_free_encoded(encoded);
1767       encoded=NULL;
1768     }
1769   }
1770
1771   rc=TRP_SUCCESS;
1772
1773 cleanup:
1774   if (updates!=NULL)
1775     g_ptr_array_free(updates, TRUE); /* frees any TRP_UPD records */
1776   talloc_free(tmp_ctx);
1777   return rc;
1778 }
1779
1780 /* all routes/communities to all peers */
1781 TRP_RC trps_update(TRPS_INSTANCE *trps, TRP_UPDATE_TYPE update_type)
1782 {
1783   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1784   TRP_PTABLE_ITER *iter=trp_ptable_iter_new(tmp_ctx);
1785   TRP_PEER *peer=NULL;
1786   TRP_RC rc=TRP_SUCCESS;
1787
1788   if (trps->ptable==NULL)
1789     return TRP_SUCCESS; /* no peers, nothing to do */
1790
1791   if (iter==NULL) {
1792     tr_err("trps_update: failed to allocate peer table iterator.");
1793     talloc_free(tmp_ctx);
1794     return TRP_NOMEM;
1795   }
1796
1797   for (peer=trp_ptable_iter_first(iter, trps->ptable);
1798        (peer!=NULL) && (rc==TRP_SUCCESS);
1799        peer=trp_ptable_iter_next(iter))
1800   {
1801     if (!trps_peer_connected(trps, peer)) {
1802       TR_NAME *peer_label=trp_peer_get_label(peer);
1803       tr_debug("trps_update: no TRP connection to %.*s, skipping.",
1804                peer_label->len, peer_label->buf);
1805       continue;
1806     }
1807     rc=trps_update_one_peer(trps, peer, update_type, NULL, NULL);
1808   }
1809
1810   tr_debug("trps_update: rc=%u after attempting update.", rc);
1811   trp_ptable_iter_free(iter);
1812   trp_rtable_clear_triggered(trps->rtable); /* don't re-send triggered updates */
1813   talloc_free(tmp_ctx);
1814   return rc;
1815 }
1816
1817 TRP_RC trps_add_route(TRPS_INSTANCE *trps, TRP_ROUTE *route)
1818 {
1819   trp_rtable_add(trps->rtable, route); /* should return status */
1820   return TRP_SUCCESS;
1821 }
1822
1823 /* steals the peer object */
1824 TRP_RC trps_add_peer(TRPS_INSTANCE *trps, TRP_PEER *peer)
1825 {
1826   if (trps->ptable==NULL) {
1827     trps->ptable=trp_ptable_new(trps);
1828     if (trps->ptable==NULL)
1829       return TRP_NOMEM;
1830   }
1831   return trp_ptable_add(trps->ptable, peer);
1832 }
1833
1834 TRP_PEER *trps_get_peer_by_gssname(TRPS_INSTANCE *trps, TR_NAME *gssname)
1835 {
1836   if (trps->ptable==NULL)
1837     return NULL;
1838
1839   return trp_ptable_find_gss_name(trps->ptable, gssname);
1840 }
1841
1842 TRP_PEER *trps_get_peer_by_servicename(TRPS_INSTANCE *trps, TR_NAME *servicename)
1843 {
1844   if (trps->ptable==NULL)
1845     return NULL;
1846
1847   return trp_ptable_find_servicename(trps->ptable, servicename);
1848 }
1849
1850 int trps_peer_connected(TRPS_INSTANCE *trps, TRP_PEER *peer)
1851 {
1852   TRPC_INSTANCE *trpc=trps_find_trpc(trps, peer);
1853   if (trpc==NULL)
1854     return 0;
1855
1856   if (trpc_get_status(trpc)==TRP_CONNECTION_UP)
1857     return 1;
1858   else
1859     return 0;
1860 }
1861
1862
1863 static TRP_RC trps_handle_request(TRPS_INSTANCE *trps, TRP_REQ *req)
1864 {
1865   TR_NAME *comm=NULL;
1866   TR_NAME *realm=NULL;
1867
1868   tr_debug("trps_handle_request: handling TRP request.");
1869
1870   if (trps_validate_request(trps, req) != TRP_SUCCESS) {
1871     tr_notice("trps_handle_request: received invalid TRP request.");
1872     return TRP_ERROR;
1873   }
1874
1875   if (!trp_req_is_wildcard(req)) {
1876     comm=trp_req_get_comm(req);
1877     realm=trp_req_get_realm(req);
1878     tr_debug("trps_handle_request: route for %.*s/%.*s requested.",
1879              comm->len, comm->buf, realm->len, realm->buf);
1880   } else {
1881     tr_debug("trps_handle_request: all routes requested.");
1882     /* leave comm/realm NULL */
1883   }
1884   return trps_update_one_peer(trps,
1885                               trps_get_peer_by_gssname(trps, trp_req_get_peer(req)),
1886                               TRP_UPDATE_REQUESTED,
1887                               comm,
1888                               realm);
1889 }
1890
1891
1892 TRP_RC trps_handle_tr_msg(TRPS_INSTANCE *trps, TR_MSG *tr_msg)
1893 {
1894   TRP_RC rc=TRP_ERROR;
1895
1896   switch (tr_msg_get_msg_type(tr_msg)) {
1897   case TRP_UPDATE:
1898     rc=trps_handle_update(trps, tr_msg_get_trp_upd(tr_msg));
1899     if (rc==TRP_SUCCESS) {
1900       rc=trps_update_active_routes(trps);
1901       trps_update(trps, TRP_UPDATE_TRIGGERED); /* send any triggered routes */
1902     }
1903     return rc;
1904
1905   case TRP_REQUEST:
1906     rc=trps_handle_request(trps, tr_msg_get_trp_req(tr_msg));
1907     return rc;
1908
1909   default:
1910     /* unknown error or one we don't care about (e.g., TID messages) */
1911     return TRP_ERROR;
1912   }
1913 }
1914
1915 /* send wildcard route request to a peer */
1916 TRP_RC trps_wildcard_route_req(TRPS_INSTANCE *trps, TR_NAME *peer_servicename)
1917 {
1918   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1919   TRP_PEER *peer=trps_get_peer_by_servicename(trps, peer_servicename);
1920   TR_MSG msg; /* not a pointer */
1921   TRP_REQ *req=trp_req_new(tmp_ctx);
1922   char *encoded=NULL;
1923   TRP_RC rc=TRP_ERROR;
1924
1925   if (peer==NULL) {
1926     tr_err("trps_wildcard_route_req: unknown peer (%.*s).", peer_servicename->len, peer_servicename->buf);
1927     rc=TRP_BADARG;
1928     goto cleanup;
1929   }
1930   if ((req==NULL) || (trp_req_make_wildcard(req)!=TRP_SUCCESS)) {
1931     tr_err("trps_wildcard_route_req: unable to create wildcard TRP request.");
1932     rc=TRP_NOMEM;
1933     goto cleanup;
1934   }
1935
1936   tr_msg_set_trp_req(&msg, req);
1937   encoded=tr_msg_encode(&msg);
1938   if (encoded==NULL) {
1939     tr_err("trps_wildcard_route_req: error encoding wildcard TRP request.");
1940     rc=TRP_ERROR;
1941     goto cleanup;
1942   }
1943
1944   tr_debug("trps_wildcard_route_req: adding message to queue.");
1945   if (trps_send_msg(trps, peer, encoded) != TRP_SUCCESS) {
1946     tr_err("trps_wildcard_route_req: error queueing request.");
1947     rc=TRP_ERROR;
1948   } else {
1949     tr_debug("trps_wildcard_route_req: request queued successfully.");
1950     rc=TRP_SUCCESS;
1951   }
1952
1953 cleanup:
1954   if (encoded!=NULL)
1955     tr_msg_free_encoded(encoded);
1956   if (req!=NULL)
1957     trp_req_free(req);
1958
1959   talloc_free(tmp_ctx);
1960   return rc;
1961 }