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