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