Allow inforec filter to have access to realm and community
[trust_router.git] / 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  * @param realm Name of realm
1013  * @param comm Name of community
1014  * @return 1 if accepted by the filter, 0 otherwise
1015  */
1016 static int trps_filter_inbound_inforec(TRPS_INSTANCE *trps, TR_NAME *peer_name, TRP_INFOREC *rec, TR_NAME *realm, TR_NAME *comm)
1017 {
1018   TRP_PEER *peer=NULL;
1019   TR_FILTER_ACTION action=TR_FILTER_ACTION_REJECT;
1020   TR_FILTER_TARGET *target=NULL;
1021   int retval=0;
1022
1023   /* Look up the peer. For inbound messages, the peer is identified by its GSS name */
1024   peer=trps_get_peer_by_gssname(trps, peer_name);
1025   if (peer==NULL) {
1026     tr_err("trps_filter_inbound_inforec: received inforec from unknown peer (%.*s), rejecting.",
1027            peer_name->len,
1028            peer_name->buf);
1029     return 0;
1030   }
1031
1032   /* tr_filter_apply() and tr_filter_set_get() handle null filter sets/filters by rejecting */
1033   target=tr_filter_target_trp_inforec(NULL, rec, realm, comm);
1034   if (target==NULL) {
1035     /* TODO: signal that filtering failed. Until then, just filter everything and give an error message. */
1036     tr_crit("trps_filter_inbound_inforec: Unable to allocate filter target, cannot apply filter!");
1037   }
1038   if ((target==NULL)
1039       || (TR_FILTER_NO_MATCH==tr_filter_apply(target,
1040                                               tr_filter_set_get(peer->filters, TR_FILTER_TYPE_TRP_INBOUND),
1041                                               NULL,
1042                                               &action))
1043       || (action!=TR_FILTER_ACTION_ACCEPT)) {
1044     /* either the filter did not match or it matched a reject rule or allocating the target failed */
1045     retval=0;
1046   } else
1047     retval=1;
1048   if (target!=NULL)
1049     tr_filter_target_free(target);
1050
1051   /* filter matched an accept rule */
1052   return retval;
1053 }
1054
1055
1056 static TRP_RC trps_handle_update(TRPS_INSTANCE *trps, TRP_UPD *upd)
1057 {
1058   TRP_INFOREC *rec=NULL;
1059
1060   if (trps_validate_update(trps, upd) != TRP_SUCCESS) {
1061     tr_notice("trps_handle_update: received invalid TRP update.");
1062     return TRP_ERROR;
1063   }
1064
1065   for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
1066     /* validate/sanity check the record update */
1067     if (trps_validate_inforec(trps, rec) != TRP_SUCCESS) {
1068       tr_notice("trps_handle_update: invalid inforec in TRP update, discarding entire update.");
1069       return TRP_ERROR;
1070     }
1071   }
1072
1073   for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
1074     if (!trps_filter_inbound_inforec(trps,
1075                                      trp_upd_get_peer(upd),
1076                                      rec,
1077                                      trp_upd_get_realm(upd),
1078                                      trp_upd_get_comm(upd))) {
1079       tr_debug("trps_handle_update: inforec rejected by filter.");
1080       continue; /* just go on to the next record */
1081     }
1082
1083     switch (trp_inforec_get_type(rec)) {
1084     case TRP_INFOREC_TYPE_ROUTE:
1085       tr_debug("trps_handle_update: handling route inforec.");
1086       if (TRP_SUCCESS!=trps_handle_inforec_route(trps, upd, rec))
1087         tr_notice("trps_handle_update: error handling route inforec.");
1088       break;
1089     case TRP_INFOREC_TYPE_COMMUNITY:
1090       tr_debug("trps_handle_update: handling community inforec.");
1091       if (TRP_SUCCESS!=trps_handle_inforec_comm(trps, upd, rec))
1092         tr_notice("trps_handle_update: error handling community inforec.");
1093
1094       break;
1095     default:
1096       tr_notice("trps_handle_update: unsupported inforec in TRP update.");
1097       break;
1098     }
1099   }
1100   return TRP_SUCCESS;
1101 }
1102
1103 static TRP_RC trps_validate_request(TRPS_INSTANCE *trps, TRP_REQ *req)
1104 {
1105   if (req==NULL) {
1106     tr_notice("trps_validate_request: null TRP request.");
1107     return TRP_BADARG;
1108   }
1109
1110   if (trp_req_get_comm(req)==NULL) {
1111     tr_notice("trps_validate_request: received TRP request with null community.");
1112     return TRP_ERROR;
1113   }
1114   
1115   if (trp_req_get_realm(req)==NULL) {
1116     tr_notice("trps_validate_request: received TRP request with null realm.");
1117     return TRP_ERROR;
1118   }
1119   
1120   if (trp_req_get_peer(req)==NULL) {
1121     tr_notice("trps_validate_request: received TRP request without origin peer information.");
1122     return TRP_ERROR;
1123   }
1124   
1125   return TRP_SUCCESS;
1126 }
1127
1128 /* choose the best route to comm/realm, optionally excluding routes to a particular peer */
1129 static TRP_ROUTE *trps_find_best_route(TRPS_INSTANCE *trps,
1130                                         TR_NAME *comm,
1131                                         TR_NAME *realm,
1132                                         TR_NAME *exclude_peer)
1133 {
1134   TRP_ROUTE **entry=NULL;
1135   TRP_ROUTE *best=NULL;
1136   size_t n_entry=0;
1137   unsigned int kk=0;
1138   unsigned int kk_min=0;
1139   unsigned int min_metric=TRP_METRIC_INFINITY;
1140
1141   entry=trp_rtable_get_realm_entries(trps->rtable, comm, realm, &n_entry);
1142   for (kk=0; kk<n_entry; kk++) {
1143     if (trp_route_get_metric(entry[kk]) < min_metric) {
1144       if ((exclude_peer==NULL) || (0!=tr_name_cmp(trp_route_get_peer(entry[kk]),
1145                                                   exclude_peer))) {
1146         kk_min=kk;
1147         min_metric=trp_route_get_metric(entry[kk]);
1148       } 
1149     }
1150   }
1151   if (trp_metric_is_finite(min_metric))
1152     best=entry[kk_min];
1153   
1154   talloc_free(entry);
1155   return best;
1156 }
1157
1158 /* TODO: think this through more carefully. At least ought to add hysteresis
1159  * to avoid flapping between routers or routes. */
1160 TRP_RC trps_update_active_routes(TRPS_INSTANCE *trps)
1161 {
1162   size_t n_comm=0, ii=0;
1163   TR_NAME **comm=trp_rtable_get_comms(trps->rtable, &n_comm);
1164   size_t n_realm=0, jj=0;
1165   TR_NAME **realm=NULL;
1166   TRP_ROUTE *best_route=NULL, *cur_route=NULL;
1167   unsigned int best_metric=0, cur_metric=0;
1168
1169   for (ii=0; ii<n_comm; ii++) {
1170     realm=trp_rtable_get_comm_realms(trps->rtable, comm[ii], &n_realm);
1171     for (jj=0; jj<n_realm; jj++) {
1172       best_route=trps_find_best_route(trps, comm[ii], realm[jj], NULL);
1173       if (best_route==NULL)
1174         best_metric=TRP_METRIC_INFINITY;
1175       else
1176         best_metric=trp_route_get_metric(best_route);
1177
1178       cur_route=trps_get_selected_route(trps, comm[ii], realm[jj]);
1179       if (cur_route!=NULL) {
1180         cur_metric=trp_route_get_metric(cur_route);
1181         if ((best_metric < cur_metric) && (trp_metric_is_finite(best_metric))) {
1182           /* The new route has a lower metric than the previous, and is finite. Accept. */
1183           trp_route_set_selected(cur_route, 0);
1184           trp_route_set_selected(best_route, 1);
1185         } else if (!trp_metric_is_finite(cur_metric)) /* rejects infinite or invalid metrics */
1186           trp_route_set_selected(cur_route, 0);
1187       } else if (trp_metric_is_finite(best_metric)) {
1188         trp_route_set_selected(best_route, 1);
1189       }
1190     }
1191     if (realm!=NULL)
1192       talloc_free(realm);
1193     realm=NULL; n_realm=0;
1194   }
1195   if (comm!=NULL)
1196     talloc_free(comm);
1197   comm=NULL; n_comm=0;
1198
1199   return TRP_SUCCESS;
1200 }
1201
1202 /* true if curtime >= expiry */
1203 static int trps_expired(struct timespec *expiry, struct timespec *curtime)
1204 {
1205   return (tr_cmp_timespec(curtime, expiry) >= 0);
1206 }
1207
1208 /* Sweep for expired routes. For each expired route, if its metric is infinite, the route is flushed.
1209  * If its metric is finite, the metric is set to infinite and the route's expiration time is updated. */
1210 TRP_RC trps_sweep_routes(TRPS_INSTANCE *trps)
1211 {
1212   struct timespec sweep_time={0,0};
1213   TRP_ROUTE **entry=NULL;
1214   size_t n_entry=0;
1215   size_t ii=0;
1216
1217   /* use a single time for the entire sweep */
1218   if (0!=clock_gettime(TRP_CLOCK, &sweep_time)) {
1219     tr_err("trps_sweep_routes: could not read realtime clock.");
1220     sweep_time.tv_sec=0;
1221     sweep_time.tv_nsec=0;
1222     return TRP_ERROR;
1223   }
1224
1225   entry=trp_rtable_get_entries(trps->rtable, &n_entry); /* must talloc_free *entry */
1226
1227   /* loop over the entries */
1228   for (ii=0; ii<n_entry; ii++) {
1229     if (!trp_route_is_local(entry[ii]) && trps_expired(trp_route_get_expiry(entry[ii]), &sweep_time)) {
1230       tr_debug("trps_sweep_routes: route expired.");
1231       if (!trp_metric_is_finite(trp_route_get_metric(entry[ii]))) {
1232         /* flush route */
1233         tr_debug("trps_sweep_routes: metric was infinity, flushing route.");
1234         trp_rtable_remove(trps->rtable, entry[ii]); /* entry[ii] is no longer valid */
1235         entry[ii]=NULL;
1236       } else {
1237         /* set metric to infinity and reset timer */
1238         tr_debug("trps_sweep_routes: setting metric to infinity and resetting expiry.");
1239         trp_route_set_metric(entry[ii], TRP_METRIC_INFINITY);
1240         trp_route_set_expiry(entry[ii], trps_compute_expiry(trps,
1241                                                              trp_route_get_interval(entry[ii]),
1242                                                              trp_route_get_expiry(entry[ii])));
1243       }
1244     }
1245   }
1246
1247   talloc_free(entry);
1248   return TRP_SUCCESS;
1249 }
1250
1251
1252 static char *timespec_to_str(struct timespec *ts)
1253 {
1254   struct tm tm;
1255   char *s=NULL;
1256
1257   if (localtime_r(&(ts->tv_sec), &tm)==NULL)
1258     return NULL;
1259
1260   s=malloc(40); /* long enough to contain strftime result */
1261   if (s==NULL)
1262     return NULL;
1263
1264   if (strftime(s, 40, "%F %T", &tm)==0) {
1265     free(s);
1266     return NULL;
1267   }
1268   return s;
1269 }
1270
1271
1272 /* Sweep for expired communities/realms/memberships. */
1273 TRP_RC trps_sweep_ctable(TRPS_INSTANCE *trps)
1274 {
1275   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1276   struct timespec sweep_time={0,0};
1277   TR_COMM_MEMB *memb=NULL;
1278   TR_COMM_ITER *iter=NULL;
1279   TRP_RC rc=TRP_ERROR;
1280
1281   /* use a single time for the entire sweep */
1282   if (0!=clock_gettime(TRP_CLOCK, &sweep_time)) {
1283     tr_err("trps_sweep_ctable: could not read realtime clock.");
1284     sweep_time.tv_sec=0;
1285     sweep_time.tv_nsec=0;
1286     goto cleanup;
1287   }
1288
1289   /* iterate all memberships */
1290   iter=tr_comm_iter_new(tmp_ctx);
1291   if (iter==NULL) {
1292     tr_err("trps_sweep_ctable: unable to allocate iterator.");
1293     rc=TRP_NOMEM;
1294     goto cleanup;
1295   }
1296   for (memb=tr_comm_memb_iter_all_first(iter, trps->ctable);
1297        memb!=NULL;
1298        memb=tr_comm_memb_iter_all_next(iter)) {
1299     if (tr_comm_memb_get_origin(memb)==NULL)
1300       continue; /* do not expire local entries */
1301
1302     if (tr_comm_memb_is_expired(memb, &sweep_time)) {
1303       if (tr_comm_memb_get_times_expired(memb)>0) {
1304         /* Already expired once; flush. */
1305         tr_debug("trps_sweep_ctable: flushing expired community membership (%.*s in %.*s, origin %.*s, expired %s).",
1306                  tr_comm_memb_get_realm_id(memb)->len, tr_comm_memb_get_realm_id(memb)->buf,
1307                  tr_comm_get_id(tr_comm_memb_get_comm(memb))->len, tr_comm_get_id(tr_comm_memb_get_comm(memb))->buf,
1308                  tr_comm_memb_get_origin(memb)->len, tr_comm_memb_get_origin(memb)->buf,
1309                  timespec_to_str(tr_comm_memb_get_expiry(memb)));
1310         tr_comm_table_remove_memb(trps->ctable, memb);
1311         tr_comm_memb_free(memb);
1312       } else {
1313         /* This is the first expiration. Note this and reset the expiry time. */
1314         tr_comm_memb_expire(memb);
1315         trps_compute_expiry(trps, tr_comm_memb_get_interval(memb), tr_comm_memb_get_expiry(memb));
1316         tr_debug("trps_sweep_ctable: community membership expired at %s, resetting expiry to %s (%.*s in %.*s, origin %.*s).",
1317                  timespec_to_str(&sweep_time),
1318                  timespec_to_str(tr_comm_memb_get_expiry(memb)),
1319                  tr_comm_memb_get_realm_id(memb)->len, tr_comm_memb_get_realm_id(memb)->buf,
1320                  tr_comm_get_id(tr_comm_memb_get_comm(memb))->len, tr_comm_get_id(tr_comm_memb_get_comm(memb))->buf,
1321                  tr_comm_memb_get_origin(memb)->len, tr_comm_memb_get_origin(memb)->buf);
1322       }
1323     }
1324   }
1325
1326   /* get rid of any unreferenced realms, etc */
1327   tr_comm_table_sweep(trps->ctable);
1328
1329 cleanup:
1330   talloc_free(tmp_ctx);
1331   return rc;
1332 }
1333
1334 /* add metrics */
1335 static unsigned int trps_metric_add(unsigned int m1, unsigned int m2)
1336 {
1337   if (trp_metric_is_invalid(m1) || trp_metric_is_invalid(m2))
1338     return TRP_METRIC_INVALID;
1339
1340   if (trp_metric_is_infinite(m1) || trp_metric_is_infinite(m2))
1341     return TRP_METRIC_INFINITY;
1342
1343   if (trp_metric_is_finite(m1+m2))
1344     return m1+m2;
1345   else
1346     return TRP_METRIC_INFINITY;
1347 }
1348
1349 /* convert an rentry into a new trp update info record */
1350 static TRP_INFOREC *trps_route_to_inforec(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TRP_ROUTE *route)
1351 {
1352   TRP_INFOREC *rec=trp_inforec_new(mem_ctx, TRP_INFOREC_TYPE_ROUTE);
1353   unsigned int linkcost=0;
1354
1355   if (rec!=NULL) {
1356     if (trp_route_is_local(route))
1357       linkcost=0;
1358     else {
1359       linkcost=trp_peer_get_linkcost(trps_get_peer_by_gssname(trps,
1360                                                               trp_route_get_peer(route)));
1361     }
1362
1363     /* Note that we leave the next hop empty since the recipient fills that in.
1364      * This is where we add the link cost (currently always 1) to the next peer. */
1365     if ((trp_inforec_set_trust_router(rec, trp_route_dup_trust_router(route)) != TRP_SUCCESS)
1366        ||(trp_inforec_set_metric(rec,
1367                                  trps_metric_add(trp_route_get_metric(route),
1368                                                  linkcost)) != TRP_SUCCESS)
1369        ||(trp_inforec_set_interval(rec, trps_get_update_interval(trps)) != TRP_SUCCESS)) {
1370       tr_err("trps_route_to_inforec: error creating route update.");
1371       talloc_free(rec);
1372       rec=NULL;
1373     }
1374   }
1375   return rec;
1376 }
1377
1378 static TRP_UPD *trps_route_to_upd(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TRP_ROUTE *route)
1379 {
1380   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1381   TRP_UPD *upd=trp_upd_new(tmp_ctx);
1382   TRP_INFOREC *rec=NULL;
1383
1384   if (upd==NULL) {
1385     tr_err("trps_route_to_upd: could not create update message.");
1386     goto cleanup;
1387   }
1388   trp_upd_set_realm(upd, trp_route_dup_realm(route));
1389   if (trp_upd_get_realm(upd)==NULL) {
1390     tr_err("trps_route_to_upd: could not copy realm.");
1391     upd=NULL; /* it's still in tmp_ctx, so it will be freed */
1392     goto cleanup;
1393   }
1394   trp_upd_set_comm(upd, trp_route_dup_comm(route));
1395   if (trp_upd_get_comm(upd)==NULL) {
1396     tr_err("trps_route_to_upd: could not copy comm.");
1397     upd=NULL; /* it's still in tmp_ctx, so it will be freed */
1398     goto cleanup;
1399   }
1400   rec=trps_route_to_inforec(tmp_ctx, trps, route);
1401   if (rec==NULL) {
1402     tr_err("trps_route_to_upd: could not create route info record for realm %.*s in comm %.*s.",
1403            trp_route_get_realm(route)->len, trp_route_get_realm(route)->buf,
1404            trp_route_get_comm(route)->len, trp_route_get_comm(route)->buf);
1405     upd=NULL; /* it's till in tmp_ctx, so it will be freed */
1406     goto cleanup;
1407   }
1408   trp_upd_add_inforec(upd, rec);
1409
1410   /* sucess */
1411   talloc_steal(mem_ctx, upd);
1412
1413 cleanup:
1414   talloc_free(tmp_ctx);
1415   return upd;
1416 }
1417
1418 /* select the correct route to comm/realm to be announced to peer */
1419 static TRP_ROUTE *trps_select_realm_update(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer_gssname)
1420 {
1421   TRP_ROUTE *route;
1422
1423   /* Take the currently selected route unless it is through the peer we're sending the update to.
1424    * I.e., enforce the split horizon rule. */
1425   route=trp_rtable_get_selected_entry(trps->rtable, comm, realm);
1426   if (route==NULL) {
1427     /* No selected route, this should only happen if the only route has been retracted,
1428      * in which case we do not want to advertise it. */
1429     return NULL;
1430   }
1431   tr_debug("trps_select_realm_update: %s vs %s", peer_gssname->buf,
1432            trp_route_get_peer(route)->buf);
1433   if (0==tr_name_cmp(peer_gssname, trp_route_get_peer(route))) {
1434     tr_debug("trps_select_realm_update: matched, finding alternate route");
1435     /* the selected entry goes through the peer we're reporting to, choose an alternate */
1436     route=trps_find_best_route(trps, comm, realm, peer_gssname);
1437     if ((route==NULL) || (!trp_metric_is_finite(trp_route_get_metric(route))))
1438       return NULL; /* don't advertise a nonexistent or retracted route */
1439   }
1440   return route;
1441 }
1442
1443 /* Add TRP_UPD msgs to the updates GPtrArray. Caller needs to arrange for these to be freed. */
1444 static TRP_RC trps_select_route_updates_for_peer(TALLOC_CTX *mem_ctx,
1445                                                  GPtrArray *updates,
1446                                                  TRPS_INSTANCE *trps,
1447                                                  TR_NAME *peer_gssname,
1448                                                  int triggered)
1449 {
1450   size_t n_comm=0;
1451   TR_NAME **comm=trp_rtable_get_comms(trps->rtable, &n_comm);
1452   TR_NAME **realm=NULL;
1453   size_t n_realm=0;
1454   size_t ii=0, jj=0;
1455   TRP_ROUTE *best=NULL;
1456   TRP_UPD *upd=NULL;
1457
1458   if (updates==NULL)
1459     return TRP_BADARG;
1460
1461   for (ii=0; ii<n_comm; ii++) {
1462     realm=trp_rtable_get_comm_realms(trps->rtable, comm[ii], &n_realm);
1463     for (jj=0; jj<n_realm; jj++) {
1464       best=trps_select_realm_update(trps, comm[ii], realm[jj], peer_gssname);
1465       /* If we found a route, add it to the list. If triggered!=0, then only
1466        * add triggered routes. */
1467       if ((best!=NULL) && ((!triggered) || trp_route_is_triggered(best))) {
1468         upd=trps_route_to_upd(mem_ctx, trps, best);
1469         if (upd==NULL) {
1470           tr_err("trps_select_route_updates_for_peer: unable to create update message.");
1471           continue;
1472         }
1473         g_ptr_array_add(updates, upd);
1474       }
1475     }
1476     
1477     if (realm!=NULL)
1478       talloc_free(realm);
1479     realm=NULL;
1480     n_realm=0;
1481   }
1482
1483   if (comm!=NULL)
1484     talloc_free(comm);
1485   
1486   return TRP_SUCCESS;
1487 }
1488
1489 static TRP_INFOREC *trps_memb_to_inforec(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TR_COMM_MEMB *memb)
1490 {
1491   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1492   TRP_INFOREC *rec=NULL;
1493   TR_COMM *comm=NULL;
1494
1495   if (memb==NULL)
1496     goto cleanup;
1497
1498   comm=tr_comm_memb_get_comm(memb);
1499   rec=trp_inforec_new(tmp_ctx, TRP_INFOREC_TYPE_COMMUNITY);
1500   if (rec==NULL)
1501     goto cleanup;
1502   
1503   if (TRP_SUCCESS!=trp_inforec_set_comm_type(rec, tr_comm_get_type(comm))) {
1504     rec=NULL;
1505     goto cleanup;
1506   }
1507   
1508   if (TRP_SUCCESS!=trp_inforec_set_role(rec, tr_comm_memb_get_role(memb))) {
1509     rec=NULL;
1510     goto cleanup;
1511   }
1512
1513   if ((NULL!=tr_comm_get_apcs(comm)) &&
1514       ( (TRP_SUCCESS!=trp_inforec_set_apcs(rec,
1515                                            tr_apc_dup(rec, tr_comm_get_apcs(comm)))) ||
1516         (NULL==trp_inforec_get_apcs(rec)))) {
1517     rec=NULL;
1518     goto cleanup;
1519   }
1520
1521   if ((NULL!=tr_comm_get_owner_realm(comm)) &&
1522       ( (TRP_SUCCESS!=trp_inforec_set_owner_realm(rec, tr_dup_name(tr_comm_get_owner_realm(comm)))) ||
1523         (NULL==trp_inforec_get_owner_realm(rec)))) {
1524     rec=NULL;
1525     goto cleanup;
1526   }
1527
1528   if ((NULL!=tr_comm_get_owner_contact(comm)) &&
1529       ( (TRP_SUCCESS!=trp_inforec_set_owner_contact(rec, tr_dup_name(tr_comm_get_owner_contact(comm)))) ||
1530         (NULL==trp_inforec_get_owner_contact(rec)))) {
1531     rec=NULL;
1532     goto cleanup;
1533   }
1534
1535   if ((NULL!=tr_comm_memb_get_provenance(memb)) &&
1536       (TRP_SUCCESS!=trp_inforec_set_provenance(rec, tr_comm_memb_get_provenance(memb)))) {
1537     rec=NULL;
1538     goto cleanup;
1539   }
1540
1541   if (TRP_SUCCESS!=trp_inforec_set_interval(rec, trps_get_update_interval(trps))) {
1542     rec=NULL;
1543     goto cleanup;
1544   }
1545
1546   /* success! */
1547   talloc_steal(mem_ctx, rec);
1548
1549 cleanup:
1550   talloc_free(tmp_ctx);
1551   return rec;
1552 }
1553
1554 /* construct an update with all the inforecs for comm/realm/role to be sent to peer */
1555 static TRP_UPD *trps_comm_update(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TR_NAME *peer_gssname, TR_COMM *comm, TR_REALM *realm)
1556 {
1557   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1558   TRP_UPD *upd=trp_upd_new(tmp_ctx);
1559   TRP_INFOREC *rec=NULL;
1560   TR_COMM_ITER *iter=NULL;
1561   TR_COMM_MEMB *memb=NULL;
1562
1563   if (upd==NULL)
1564     goto cleanup;
1565   
1566   trp_upd_set_comm(upd, tr_comm_dup_id(comm));
1567   trp_upd_set_realm(upd, tr_realm_dup_id(realm));
1568   /* leave peer empty */
1569
1570   iter=tr_comm_iter_new(tmp_ctx);
1571   if (iter==NULL) {
1572     tr_err("trps_comm_update: unable to allocate iterator.");
1573     upd=NULL;
1574     goto cleanup;
1575   }
1576   
1577   /* now add inforecs */
1578   switch (realm->role) {
1579   case TR_ROLE_IDP:
1580     memb=tr_comm_table_find_idp_memb(trps->ctable,
1581                                      tr_realm_get_id(realm),
1582                                      tr_comm_get_id(comm));
1583     break;
1584   case TR_ROLE_RP:
1585     memb=tr_comm_table_find_rp_memb(trps->ctable,
1586                                     tr_realm_get_id(realm),
1587                                     tr_comm_get_id(comm));
1588     break;
1589   default:
1590     break;
1591   }
1592   if (memb!=NULL) {
1593     for (memb=tr_comm_memb_iter_first(iter, memb);
1594          memb!=NULL;
1595          memb=tr_comm_memb_iter_next(iter)) {
1596       rec=trps_memb_to_inforec(tmp_ctx, trps, memb);
1597       if (rec==NULL) {
1598         tr_err("trps_comm_update: unable to allocate inforec.");
1599         upd=NULL;
1600         goto cleanup;
1601       }
1602       trp_upd_add_inforec(upd, rec);
1603     }
1604   }
1605
1606   if (trp_upd_get_inforec(upd)==NULL)
1607     upd=NULL; /* no inforecs, no reason to send the update */
1608   else
1609     talloc_steal(mem_ctx, upd); /* success! */
1610
1611 cleanup:
1612   talloc_free(tmp_ctx);
1613   return upd;
1614 }
1615
1616 /* Find all community updates to send to a peer and add these as TR_UPD records
1617  * to the updates GPtrArray. */
1618 static TRP_RC trps_select_comm_updates_for_peer(TALLOC_CTX *mem_ctx,
1619                                                 GPtrArray *updates,
1620                                                 TRPS_INSTANCE *trps,
1621                                                 TR_NAME *peer_gssname,
1622                                                 int triggered)
1623 {
1624   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1625   TR_COMM_ITER *comm_iter=NULL;
1626   TR_COMM *comm=NULL;
1627   TR_COMM_ITER *realm_iter=NULL;
1628   TR_REALM *realm=NULL;
1629   TRP_UPD *upd=NULL;
1630   TRP_RC rc=TRP_ERROR;
1631
1632   /* currently do not send any communities on triggered updates */
1633   if (triggered) {
1634     rc=TRP_SUCCESS;
1635     goto cleanup;
1636   }
1637
1638   comm_iter=tr_comm_iter_new(tmp_ctx);
1639   realm_iter=tr_comm_iter_new(tmp_ctx);
1640   if ((comm_iter==NULL) || (realm_iter==NULL)) {
1641     tr_err("trps_select_comm_updates_for_peer: unable to allocate iterator.");
1642     rc=TRP_NOMEM;
1643     goto cleanup;
1644   }
1645
1646   /* do every community */
1647   for (comm=tr_comm_table_iter_first(comm_iter, trps->ctable);
1648        comm!=NULL;
1649        comm=tr_comm_table_iter_next(comm_iter)) {
1650     /* do every realm in this community */
1651     tr_debug("trps_select_comm_updates_for_peer: looking through community %.*s",
1652              tr_comm_get_id(comm)->len,
1653              tr_comm_get_id(comm)->buf);
1654     for (realm=tr_realm_iter_first(realm_iter, trps->ctable, tr_comm_get_id(comm));
1655          realm!=NULL;
1656          realm=tr_realm_iter_next(realm_iter)) {
1657       /* get the update for this comm/realm */
1658       tr_debug("trps_select_comm_updates_for_peer: adding realm %.*s",
1659                tr_realm_get_id(realm)->len,
1660                tr_realm_get_id(realm)->buf);
1661       upd=trps_comm_update(mem_ctx, trps, peer_gssname, comm, realm);
1662       if (upd!=NULL)
1663         g_ptr_array_add(updates, upd);
1664     }
1665   }
1666
1667 cleanup:
1668   talloc_free(tmp_ctx);
1669   return rc;
1670 }
1671
1672 /**
1673  * Filter the inforecs in a single update
1674  *
1675  * @param filt The filter to apply
1676  * @param upd The update to filter
1677  */
1678 static void trps_filter_one_outbound_update(TR_FILTER *filt, TRP_UPD *upd)
1679 {
1680   TRP_INFOREC *this=NULL, *next=NULL;
1681   TR_FILTER_ACTION action=TR_FILTER_ACTION_REJECT;
1682   TR_FILTER_TARGET *target=NULL;
1683
1684   for(this=trp_upd_get_inforec(upd); this!=NULL; this=next) {
1685     next=this->next;
1686     target=tr_filter_target_trp_inforec(NULL, this, trp_upd_get_realm(upd), trp_upd_get_comm(upd));
1687     if (target==NULL) {
1688       /* TODO: signal that filtering failed. Until then, just filter everything and give an error message. */
1689       tr_crit("trps_filter_one_outbound_update: Unable to allocate filter target, cannot apply filter!");
1690     }
1691     if ((target==NULL)
1692         || (TR_FILTER_NO_MATCH==tr_filter_apply(target, filt, NULL, &action))
1693         || (action!=TR_FILTER_ACTION_ACCEPT)) {
1694       /* Either no filter matched or one matched and rejected this record.
1695        * Also filter out record if we were unable to allocate a target. */
1696       trp_upd_remove_inforec(upd, this); /* "this" is now invalid */
1697     }
1698     if (target!=NULL)
1699       tr_filter_target_free(target);
1700   }
1701 }
1702
1703 /**
1704  * May shuffle the update list.
1705  *
1706  * @param filters The filter set for the relevant TRP peer
1707  * @param updates GPtrArray of updates to filter
1708  */
1709 static void trps_filter_outbound_updates(TR_FILTER_SET *filters, GPtrArray *updates)
1710 {
1711   TRP_UPD *upd=NULL;
1712   int ii=0;
1713
1714   /* walk backward through the array so we can remove elements */
1715   for (ii=updates->len-1; ii>=0; ii--) {
1716     upd=g_ptr_array_index(updates, ii);
1717     trps_filter_one_outbound_update(tr_filter_set_get(filters, TR_FILTER_TYPE_TRP_OUTBOUND), upd);
1718     /* see if we removed all the records from this update */
1719     if (trp_upd_num_inforecs(upd)==0)
1720       g_ptr_array_remove_index_fast(updates, ii); /* does not preserve order at index ii or higher */
1721   }
1722 }
1723
1724 /* helper for trps_update_one_peer. Frees the TRP_UPD pointed to by a GPtrArray element */
1725 static void trps_trp_upd_destroy(gpointer data)
1726 {
1727   trp_upd_free((TRP_UPD *)data);
1728 }
1729
1730 /* all routes/communities to a single peer, unless comm/realm are specified (both or neither must be NULL) */
1731 static TRP_RC trps_update_one_peer(TRPS_INSTANCE *trps,
1732                                    TRP_PEER *peer,
1733                                    TRP_UPDATE_TYPE update_type,
1734                                    TR_NAME *realm,
1735                                    TR_NAME *comm)
1736 {
1737   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1738   TR_MSG msg; /* not a pointer! */
1739   TRP_UPD *upd=NULL;
1740   TRP_ROUTE *route=NULL;
1741   size_t ii=0;
1742   char *encoded=NULL;
1743   TRP_RC rc=TRP_ERROR;
1744   TR_NAME *peer_label=trp_peer_get_label(peer);
1745   GPtrArray *updates=g_ptr_array_new_with_free_func(trps_trp_upd_destroy);
1746
1747   if (updates==NULL) {
1748     tr_err("trps_update_one_peer: unable to allocate updates array.");
1749     rc=TRP_NOMEM;
1750     goto cleanup;
1751   }
1752
1753   switch (update_type) {
1754   case TRP_UPDATE_TRIGGERED:
1755     tr_debug("trps_update_one_peer: preparing triggered update for %.*s",
1756              peer_label->len, peer_label->buf);
1757     break;
1758   case TRP_UPDATE_SCHEDULED:
1759     tr_debug("trps_update_one_peer: preparing scheduled update for %.*s",
1760              peer_label->len, peer_label->buf);
1761     break;
1762   case TRP_UPDATE_REQUESTED:
1763     tr_debug("trps_update_one_peer: preparing requested update for %.*s",
1764              peer_label->len, peer_label->buf);
1765     break;
1766   default:
1767     tr_err("trps_update_one_peer: invalid update type requested.");
1768     rc=TRP_BADARG;
1769     goto cleanup;
1770   }
1771
1772   /* First, gather route updates. */
1773   tr_debug("trps_update_one_peer: selecting route updates for %.*s.", peer_label->len, peer_label->buf);
1774   if ((comm==NULL) && (realm==NULL)) {
1775     /* do all realms */
1776     rc=trps_select_route_updates_for_peer(tmp_ctx,
1777                                           updates,
1778                                           trps,
1779                                           peer_label,
1780                                           update_type==TRP_UPDATE_TRIGGERED);
1781   } else if ((comm!=NULL) && (realm!=NULL)) {
1782     /* a single community/realm was requested */
1783     route=trps_select_realm_update(trps, comm, realm, peer_label);
1784     if (route==NULL) {
1785       /* we have no actual update to send back, MUST send a retraction */
1786       tr_debug("trps_update_one_peer: community/realm without route requested, sending mandatory retraction.");
1787       route=trp_route_new(tmp_ctx);
1788       trp_route_set_comm(route, tr_dup_name(comm));
1789       trp_route_set_realm(route, tr_dup_name(realm));
1790       trp_route_set_peer(route, tr_new_name(""));
1791       trp_route_set_metric(route, TRP_METRIC_INFINITY);
1792       trp_route_set_trust_router(route, tr_new_name(""));
1793       trp_route_set_next_hop(route, tr_new_name(""));
1794     }
1795     upd=trps_route_to_upd(tmp_ctx, trps, route);
1796     if (upd==NULL) {
1797       tr_err("trps_update_one_peer: unable to allocate route update.");
1798       rc=TRP_NOMEM;
1799       goto cleanup;
1800     }
1801     g_ptr_array_add(updates, upd);
1802   } else {
1803     tr_err("trps_update_one_peer: error: only comm or realm was specified. Need both or neither.");
1804     rc=TRP_ERROR;
1805     goto cleanup;
1806   }
1807
1808   /* Second, gather community updates */
1809   tr_debug("trps_update_one_peer: selecting community updates for %.*s.", peer_label->len, peer_label->buf);
1810   rc=trps_select_comm_updates_for_peer(tmp_ctx, updates, trps, peer_label, update_type==TRP_UPDATE_TRIGGERED);
1811
1812   /* see if we have anything to send */
1813   if (updates->len<=0)
1814     tr_debug("trps_update_one_peer: no updates for %.*s", peer_label->len, peer_label->buf);
1815   else {
1816     /* Apply outbound TRP filters for this peer */
1817     trps_filter_outbound_updates(peer->filters, updates);
1818
1819     if (updates->len<=0)
1820       tr_debug("trps_update_one_peer: no updates for %.*s after filtering.", peer_label->len, peer_label->buf);
1821     else {
1822       tr_debug("trps_update_one_peer: sending %d update messages.", updates->len);
1823       for (ii=0; ii<updates->len; ii++) {
1824         upd = (TRP_UPD *) g_ptr_array_index(updates, ii);
1825         /* now encode the update message */
1826         tr_msg_set_trp_upd(&msg, upd);
1827         encoded = tr_msg_encode(&msg);
1828         if (encoded == NULL) {
1829           tr_err("trps_update_one_peer: error encoding update.");
1830           rc = TRP_ERROR;
1831           goto cleanup;
1832         }
1833
1834         tr_debug("trps_update_one_peer: adding message to queue.");
1835         if (trps_send_msg(trps, peer, encoded) != TRP_SUCCESS)
1836           tr_err("trps_update_one_peer: error queueing update.");
1837         else
1838           tr_debug("trps_update_one_peer: update queued successfully.");
1839
1840         tr_msg_free_encoded(encoded);
1841         encoded = NULL;
1842       }
1843     }
1844   }
1845
1846   rc=TRP_SUCCESS;
1847
1848 cleanup:
1849   if (updates!=NULL)
1850     g_ptr_array_free(updates, TRUE); /* frees any TRP_UPD records */
1851   talloc_free(tmp_ctx);
1852   return rc;
1853 }
1854
1855 /* all routes/communities to all peers */
1856 TRP_RC trps_update(TRPS_INSTANCE *trps, TRP_UPDATE_TYPE update_type)
1857 {
1858   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1859   TRP_PTABLE_ITER *iter=trp_ptable_iter_new(tmp_ctx);
1860   TRP_PEER *peer=NULL;
1861   TRP_RC rc=TRP_SUCCESS;
1862
1863   if (trps->ptable==NULL)
1864     return TRP_SUCCESS; /* no peers, nothing to do */
1865
1866   if (iter==NULL) {
1867     tr_err("trps_update: failed to allocate peer table iterator.");
1868     talloc_free(tmp_ctx);
1869     return TRP_NOMEM;
1870   }
1871
1872   for (peer=trp_ptable_iter_first(iter, trps->ptable);
1873        (peer!=NULL) && (rc==TRP_SUCCESS);
1874        peer=trp_ptable_iter_next(iter))
1875   {
1876     if (!trps_peer_connected(trps, peer)) {
1877       TR_NAME *peer_label=trp_peer_get_label(peer);
1878       tr_debug("trps_update: no TRP connection to %.*s, skipping.",
1879                peer_label->len, peer_label->buf);
1880       continue;
1881     }
1882     rc=trps_update_one_peer(trps, peer, update_type, NULL, NULL);
1883   }
1884
1885   tr_debug("trps_update: rc=%u after attempting update.", rc);
1886   trp_ptable_iter_free(iter);
1887   trp_rtable_clear_triggered(trps->rtable); /* don't re-send triggered updates */
1888   talloc_free(tmp_ctx);
1889   return rc;
1890 }        
1891
1892 TRP_RC trps_add_route(TRPS_INSTANCE *trps, TRP_ROUTE *route)
1893 {
1894   trp_rtable_add(trps->rtable, route); /* should return status */
1895   return TRP_SUCCESS; 
1896 }
1897
1898 /* steals the peer object */
1899 TRP_RC trps_add_peer(TRPS_INSTANCE *trps, TRP_PEER *peer)
1900 {
1901   if (trps->ptable==NULL) {
1902     trps->ptable=trp_ptable_new(trps);
1903     if (trps->ptable==NULL)
1904       return TRP_NOMEM;
1905   }
1906   return trp_ptable_add(trps->ptable, peer);
1907 }
1908
1909 TRP_PEER *trps_get_peer_by_gssname(TRPS_INSTANCE *trps, TR_NAME *gssname)
1910 {
1911   if (trps->ptable==NULL)
1912     return NULL;
1913
1914   return trp_ptable_find_gss_name(trps->ptable, gssname);
1915 }
1916
1917 TRP_PEER *trps_get_peer_by_servicename(TRPS_INSTANCE *trps, TR_NAME *servicename)
1918 {
1919   if (trps->ptable==NULL)
1920     return NULL;
1921
1922   return trp_ptable_find_servicename(trps->ptable, servicename);
1923 }
1924
1925 int trps_peer_connected(TRPS_INSTANCE *trps, TRP_PEER *peer)
1926 {
1927   TRPC_INSTANCE *trpc=trps_find_trpc(trps, peer);
1928   if (trpc==NULL)
1929     return 0;
1930
1931   if (trpc_get_status(trpc)==TRP_CONNECTION_UP)
1932     return 1;
1933   else
1934     return 0;
1935 }
1936
1937
1938 static TRP_RC trps_handle_request(TRPS_INSTANCE *trps, TRP_REQ *req)
1939 {
1940   TR_NAME *comm=NULL;
1941   TR_NAME *realm=NULL;
1942
1943   tr_debug("trps_handle_request: handling TRP request.");
1944
1945   if (trps_validate_request(trps, req) != TRP_SUCCESS) {
1946     tr_notice("trps_handle_request: received invalid TRP request.");
1947     return TRP_ERROR;
1948   }
1949
1950   if (!trp_req_is_wildcard(req)) {
1951     comm=trp_req_get_comm(req);
1952     realm=trp_req_get_realm(req);
1953     tr_debug("trps_handle_request: route for %.*s/%.*s requested.",
1954              comm->len, comm->buf, realm->len, realm->buf);
1955   } else {
1956     tr_debug("trps_handle_request: all routes requested.");
1957     /* leave comm/realm NULL */
1958   }
1959   return trps_update_one_peer(trps,
1960                               trps_get_peer_by_gssname(trps, trp_req_get_peer(req)),
1961                               TRP_UPDATE_REQUESTED,
1962                               realm,
1963                               comm);
1964 }
1965
1966
1967 TRP_RC trps_handle_tr_msg(TRPS_INSTANCE *trps, TR_MSG *tr_msg)
1968 {
1969   TRP_RC rc=TRP_ERROR;
1970
1971   switch (tr_msg_get_msg_type(tr_msg)) {
1972   case TRP_UPDATE:
1973     rc=trps_handle_update(trps, tr_msg_get_trp_upd(tr_msg));
1974     if (rc==TRP_SUCCESS) {
1975       rc=trps_update_active_routes(trps);
1976       trps_update(trps, TRP_UPDATE_TRIGGERED); /* send any triggered routes */
1977     }
1978     return rc;
1979
1980   case TRP_REQUEST:
1981     rc=trps_handle_request(trps, tr_msg_get_trp_req(tr_msg));
1982     return rc;
1983
1984   default:
1985     /* unknown error or one we don't care about (e.g., TID messages) */
1986     return TRP_ERROR;
1987   }
1988 }
1989
1990 /* send wildcard route request to a peer */
1991 TRP_RC trps_wildcard_route_req(TRPS_INSTANCE *trps, TR_NAME *peer_servicename)
1992 {
1993   TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1994   TRP_PEER *peer=trps_get_peer_by_servicename(trps, peer_servicename);
1995   TR_MSG msg; /* not a pointer */
1996   TRP_REQ *req=trp_req_new(tmp_ctx);
1997   char *encoded=NULL;
1998   TRP_RC rc=TRP_ERROR;
1999
2000   if (peer==NULL) {
2001     tr_err("trps_wildcard_route_req: unknown peer (%.*s).", peer_servicename->len, peer_servicename->buf);
2002     rc=TRP_BADARG;
2003     goto cleanup;
2004   }
2005   if ((req==NULL) || (trp_req_make_wildcard(req)!=TRP_SUCCESS)) {
2006     tr_err("trps_wildcard_route_req: unable to create wildcard TRP request.");
2007     rc=TRP_NOMEM;
2008     goto cleanup;
2009   }
2010
2011   tr_msg_set_trp_req(&msg, req);
2012   encoded=tr_msg_encode(&msg);
2013   if (encoded==NULL) {
2014     tr_err("trps_wildcard_route_req: error encoding wildcard TRP request.");
2015     rc=TRP_ERROR;
2016     goto cleanup;
2017   }
2018
2019   tr_debug("trps_wildcard_route_req: adding message to queue.");
2020   if (trps_send_msg(trps, peer, encoded) != TRP_SUCCESS) {
2021     tr_err("trps_wildcard_route_req: error queueing request.");
2022     rc=TRP_ERROR;
2023   } else {
2024     tr_debug("trps_wildcard_route_req: request queued successfully.");
2025     rc=TRP_SUCCESS;
2026   }
2027
2028 cleanup:
2029   if (encoded!=NULL)
2030     tr_msg_free_encoded(encoded);
2031   if (req!=NULL)
2032     trp_req_free(req);
2033
2034   talloc_free(tmp_ctx);
2035   return rc;
2036 }