2 * Copyright (c) 2016, JANET(UK)
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
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.
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.
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.
47 #include <trust_router/tr_name.h>
48 #include <trp_internal.h>
50 #include <trp_ptable.h>
51 #include <trp_rtable.h>
55 static int trps_destructor(void *object)
57 TRPS_INSTANCE *trps=talloc_get_type_abort(object, TRPS_INSTANCE);
58 if (trps->rtable!=NULL)
59 trp_rtable_free(trps->rtable);
63 TRPS_INSTANCE *trps_new (TALLOC_CTX *mem_ctx)
65 TRPS_INSTANCE *trps=talloc(mem_ctx, TRPS_INSTANCE);
72 trps->update_interval=(struct timeval){0,0};
73 trps->sweep_interval=(struct timeval){0,0};
76 trps->mq=tr_mq_new(trps);
78 /* failed to allocate mq */
84 if (trps_init_rtable(trps) != TRP_SUCCESS) {
85 /* failed to allocate rtable */
90 talloc_set_destructor((void *)trps, trps_destructor);
95 /* create a new route table, first discarding an old one if necessary */
96 TRP_RC trps_init_rtable(TRPS_INSTANCE *trps)
98 if (trps->rtable != NULL) {
99 trp_rtable_free(trps->rtable);
103 trps->rtable=trp_rtable_new();
104 if (trps->rtable==NULL) {
110 void trps_clear_rtable(TRPS_INSTANCE *trps)
112 trp_rtable_clear(trps->rtable);
115 void trps_free (TRPS_INSTANCE *trps)
121 TR_MQ_MSG *trps_mq_pop(TRPS_INSTANCE *trps)
123 return tr_mq_pop(trps->mq, 0);
126 void trps_mq_add(TRPS_INSTANCE *trps, TR_MQ_MSG *msg)
128 tr_mq_add(trps->mq, msg);
131 unsigned int trps_get_connect_interval(TRPS_INSTANCE *trps)
133 return trps->connect_interval.tv_sec;
136 void trps_set_connect_interval(TRPS_INSTANCE *trps, unsigned int interval)
138 trps->connect_interval.tv_sec=interval;
139 trps->connect_interval.tv_usec=0;
142 unsigned int trps_get_update_interval(TRPS_INSTANCE *trps)
144 return trps->update_interval.tv_sec;
147 void trps_set_update_interval(TRPS_INSTANCE *trps, unsigned int interval)
149 trps->update_interval.tv_sec=interval;
150 trps->update_interval.tv_usec=0;
153 unsigned int trps_get_sweep_interval(TRPS_INSTANCE *trps)
155 return trps->sweep_interval.tv_sec;
158 void trps_set_sweep_interval(TRPS_INSTANCE *trps, unsigned int interval)
160 trps->sweep_interval.tv_sec=interval;
161 trps->sweep_interval.tv_usec=0;
164 void trps_set_ctable(TRPS_INSTANCE *trps, TR_COMM_TABLE *comm)
169 void trps_set_ptable(TRPS_INSTANCE *trps, TRP_PTABLE *ptable)
171 if (trps->ptable!=NULL)
172 trp_ptable_free(trps->ptable);
176 void trps_set_peer_status_callback(TRPS_INSTANCE *trps, void (*cb)(TRP_PEER *, void *), void *cookie)
178 TRP_PTABLE_ITER *iter=NULL;
180 if (trps->ptable==NULL)
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);
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)
193 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
195 char *s=talloc_asprintf(tmp_ctx, "%s:%u", trps->hostname, trps->port);
198 label=tr_new_name(s);
201 talloc_free(tmp_ctx);
205 TRPC_INSTANCE *trps_find_trpc(TRPS_INSTANCE *trps, TRP_PEER *peer)
207 TRPC_INSTANCE *cur=NULL;
209 TR_NAME *peer_servicename=trp_peer_get_servicename(peer);
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))) {
220 void trps_add_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *new)
222 if (trps->conn==NULL)
225 trp_connection_append(trps->conn, new);
227 talloc_steal(trps, new);
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)
234 trps->conn=trp_connection_remove(trps->conn, remove);
237 void trps_add_trpc(TRPS_INSTANCE *trps, TRPC_INSTANCE *trpc)
239 if (trps->trpc==NULL)
242 trpc_append(trps->trpc, trpc);
244 talloc_steal(trps, trpc);
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)
251 trps->trpc=trpc_remove(trps->trpc, remove);
254 TRP_RC trps_send_msg(TRPS_INSTANCE *trps, TRP_PEER *peer, const char *msg)
256 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
257 TR_MQ_MSG *mq_msg=NULL;
260 TRPC_INSTANCE *trpc=NULL;
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
267 tr_warning("trps_send_msg: skipping message queued for missing TRP client entry.");
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);
275 talloc_free(tmp_ctx);
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)
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};
296 port_str=talloc_asprintf(NULL, "%d", port);
297 if (port_str==NULL) {
298 tr_debug("trps_listen: unable to allocate port.");
301 getaddrinfo(NULL, port_str, &hints, &ai_head);
302 talloc_free(port_str);
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.");
311 if (0!=setsockopt(conn, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval)))
312 tr_debug("trps_listen: unable to set SO_REUSEADDR."); /* not fatal? */
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.");
324 rc=bind(conn, ai->ai_addr, ai->ai_addrlen);
326 tr_debug("trps_listen: unable to bind to socket.");
331 if (0>listen(conn, 512)) {
332 tr_debug("trps_listen: unable to listen on bound socket.");
337 /* ok, this one worked. Save it */
338 fd_out[n_opened++]=conn;
340 freeaddrinfo(ai_head);
343 tr_debug("trps_listen: no addresses available for listening.");
347 tr_debug("trps_listen: TRP Server listening on port %d on %d socket%s",
350 (n_opened==1)?"":"s");
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)
358 return trp_rtable_get_entry(trps->rtable, comm, realm, peer);
361 TRP_ROUTE *trps_get_selected_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm)
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);
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)
370 TRP_ROUTE *route=trps_get_selected_route(trps, comm, realm);
374 return trp_route_get_next_hop(route);
378 /* mark a route as retracted */
379 static void trps_retract_route(TRPS_INSTANCE *trps, TRP_ROUTE *entry)
381 trp_route_set_metric(entry, TRP_METRIC_INFINITY);
382 trp_route_set_triggered(entry, 1);
385 /* is this route retracted? */
386 static int trps_route_retracted(TRPS_INSTANCE *trps, TRP_ROUTE *entry)
388 return (trp_metric_is_infinite(trp_route_get_metric(entry)));
391 static TRP_RC trps_read_message(TRPS_INSTANCE *trps, TRP_CONNECTION *conn, TR_MSG **msg)
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 */
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)),
404 tr_debug("trps_read_message: error");
410 tr_debug("trps_read_message: message received, %u bytes.", (unsigned) buflen);
411 tr_debug("trps_read_message: %.*s", buflen, buf);
413 *msg=tr_msg_decode(buf, buflen);
418 conn_peer=trp_connection_get_peer(conn);
419 if (conn_peer==NULL) {
420 tr_err("trps_read_message: connection has no peer name");
424 peer=trps_get_peer_by_gssname(trps, conn_peer);
426 tr_err("trps_read_message: could not find peer with gssname=%s", trp_connection_get_gssname(conn));
430 /* verify we received a message we support, otherwise drop it now */
431 switch (tr_msg_get_msg_type(*msg)) {
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));
440 trp_req_set_peer(tr_msg_get_trp_req(*msg), tr_dup_name(conn_peer));
444 tr_debug("trps_read_message: received unsupported message from %.*s", conn_peer->len, conn_peer->buf);
445 tr_msg_free_decoded(*msg);
447 return TRP_UNSUPPORTED;
453 int trps_get_listener(TRPS_INSTANCE *trps,
454 TRPS_MSG_FUNC msg_handler,
455 TRP_AUTH_FUNC auth_handler,
456 const char *hostname,
465 n_fd=trps_listen(trps, port, fd_out, max_fd);
467 tr_debug("trps_get_listener: Error opening port %d.");
469 /* opening port succeeded */
470 tr_debug("trps_get_listener: Opened port %d.", port);
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++) {
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);
492 trps->cookie = cookie;
498 TRP_RC trps_authorize_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *conn)
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);
506 tr_notice("trps_authorize_connection: authorized connection");
510 void trps_handle_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *conn)
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);
520 trps->msg_handler(trps, conn, msg); /* send the TR_MSG off to the callback */
524 trp_connection_close(conn);
528 tr_debug("trps_handle_connection: trps_read_message failed (%d)", rc);
532 tr_debug("trps_handle_connection: connection closed.");
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)
539 tr_notice("trps_validate_update: null TRP update.");
543 if (trp_upd_get_realm(upd)==NULL) {
544 tr_notice("trps_validate_update: received TRP update without realm.");
548 if (trp_upd_get_comm(upd)==NULL) {
549 tr_notice("trps_validate_update: received TRP update without community.");
553 if (trp_upd_get_inforec(upd)==NULL) {
554 tr_notice("trps_validate_update: received TRP update with no info records.");
558 if (trp_upd_get_peer(upd)==NULL) {
559 tr_notice("trps_validate_update: received TRP update without origin peer information.");
567 /* ensure that the update could be accepted if feasible */
568 static TRP_RC trps_validate_inforec(TRPS_INSTANCE *trps, TRP_INFOREC *rec)
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.");
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));
584 /* check for valid interval */
585 if (trp_inforec_get_interval(rec)==TRP_INTERVAL_INVALID) {
586 tr_debug("trps_validate_inforec: invalid interval.");
591 case TRP_INFOREC_TYPE_COMMUNITY:
592 /* TODO: validate community updates */
596 tr_notice("trps_validate_inforec: unsupported record type.");
597 return TRP_UNSUPPORTED;
603 /* link cost to a peer */
604 static unsigned int trps_cost(TRPS_INSTANCE *trps, TR_NAME *peer)
609 static unsigned int trps_advertised_metric(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer)
611 TRP_ROUTE *entry=trp_rtable_get_entry(trps->rtable, comm, realm, peer);
613 return TRP_METRIC_INFINITY;
614 return trp_route_get_metric(entry) + trps_cost(trps, peer);
617 static int trps_check_feasibility(TRPS_INSTANCE *trps, TR_NAME *realm, TR_NAME *comm, TRP_INFOREC *rec)
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;
624 /* we check these in the validation stage, but just in case... */
625 if (trp_metric_is_invalid(rec_metric))
628 /* retractions (aka infinite metrics) are always feasible */
629 if (trp_metric_is_infinite(rec_metric))
632 /* updates from our current next hop are always feasible*/
633 next_hop=trps_get_next_hop(trps, comm, realm);
635 && (0==tr_name_cmp(next_hop,trp_inforec_get_next_hop(rec)))) {
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)
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)
653 const unsigned int small_factor=3; /* how many intervals we wait before expiring */
654 if (0!=clock_gettime(CLOCK_REALTIME, ts)) {
655 tr_err("trps_compute_expiry: could not read realtime clock.");
659 ts->tv_sec += small_factor*interval;
663 static TRP_RC trps_accept_update(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC *rec)
665 TRP_ROUTE *entry=NULL;
667 entry=trp_rtable_get_entry(trps->rtable,
668 trp_upd_get_comm(upd),
669 trp_upd_get_realm(upd),
670 trp_inforec_get_next_hop(rec));
672 entry=trp_route_new(NULL);
674 tr_err("trps_accept_update: unable to allocate new entry.");
678 trp_route_set_comm(entry, trp_upd_dup_comm(upd));
679 trp_route_set_realm(entry, trp_upd_dup_realm(upd));
680 trp_route_set_peer(entry, trp_upd_dup_peer(upd));
681 trp_route_set_trust_router(entry, trp_inforec_dup_trust_router(rec));
682 trp_route_set_next_hop(entry, trp_inforec_dup_next_hop(rec));
683 /* TODO: pass next hop port (now defaults to TID_PORT) --jlr */
684 if ((trp_route_get_comm(entry)==NULL)
685 ||(trp_route_get_realm(entry)==NULL)
686 ||(trp_route_get_peer(entry)==NULL)
687 ||(trp_route_get_trust_router(entry)==NULL)
688 ||(trp_route_get_next_hop(entry)==NULL)) {
689 /* at least one field could not be allocated */
690 tr_err("trps_accept_update: unable to allocate all fields for entry.");
691 trp_route_free(entry);
694 trp_rtable_add(trps->rtable, entry);
697 /* We now have an entry in the table, whether it's new or not. Update metric and expiry, unless
698 * the metric is infinity. An infinite metric can only occur here if we just retracted an existing
699 * route (we never accept retractions as new routes), so there is no risk of leaving the expiry
700 * time unset on a new route entry. */
701 tr_debug("trps_accept_update: accepting route update.");
702 trp_route_set_metric(entry, trp_inforec_get_metric(rec));
703 trp_route_set_interval(entry, trp_inforec_get_interval(rec));
705 /* check whether the trust router has changed */
706 if (0!=tr_name_cmp(trp_route_get_trust_router(entry),
707 trp_inforec_get_trust_router(rec))) {
708 /* The name changed. Set this route as triggered. */
709 tr_debug("trps_accept_update: trust router for route changed.");
710 trp_route_set_triggered(entry, 1);
711 trp_route_set_trust_router(entry, trp_inforec_dup_trust_router(rec)); /* frees old name */
713 if (!trps_route_retracted(trps, entry)) {
714 tr_debug("trps_accept_update: route not retracted, setting expiry timer.");
715 trp_route_set_expiry(entry, trps_compute_expiry(trps,
716 trp_route_get_interval(entry),
717 trp_route_get_expiry(entry)));
723 static TRP_RC trps_handle_inforec_route(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC *rec)
725 TRP_ROUTE *route=NULL;
728 /* determine feasibility */
729 feas=trps_check_feasibility(trps, trp_upd_get_realm(upd), trp_upd_get_comm(upd), rec);
730 tr_debug("trps_handle_update: record feasibility=%d", feas);
732 /* do we have an existing route? */
733 route=trps_get_route(trps,
734 trp_upd_get_comm(upd),
735 trp_upd_get_realm(upd),
736 trp_upd_get_peer(upd));
738 /* there was a route table entry already */
739 tr_debug("trps_handle_updates: route entry already exists.");
741 /* Update is feasible. Accept it. */
742 trps_accept_update(trps, upd, rec);
744 /* Update is infeasible. Ignore it unless the trust router has changed. */
745 if (0!=tr_name_cmp(trp_route_get_trust_router(route),
746 trp_inforec_get_trust_router(rec))) {
747 /* the trust router associated with the route has changed, treat update as a retraction */
748 trps_retract_route(trps, route);
752 /* No existing route table entry. Ignore it unless it is feasible and not a retraction. */
753 tr_debug("trps_handle_update: no route entry exists yet.");
754 if (feas && trp_metric_is_finite(trp_inforec_get_metric(rec)))
755 trps_accept_update(trps, upd, rec);
761 static int trps_name_in_provenance(TR_NAME *name, json_t *prov)
764 TR_NAME *this_name=NULL;
768 return 0; /* no provenance list, so it has no names in it */
770 /* now check to see if name is in the provenance */
771 for (ii=0; ii<json_array_size(prov); ii++) {
772 s=json_string_value(json_array_get(prov, ii));
774 tr_debug("trps_name_in_provenance: empty entry in provenance list.");
778 this_name=tr_new_name(s);
779 if (this_name==NULL) {
780 tr_debug("trps_name_in_provenance: unable to allocate name.");
783 if (0==tr_name_cmp(name, this_name)) {
784 tr_free_name(this_name);
787 tr_free_name(this_name);
792 static TR_COMM *trps_create_new_comm(TALLOC_CTX *mem_ctx, TR_NAME *comm_id, TRP_INFOREC *rec)
794 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
795 TR_COMM *comm=tr_comm_new(tmp_ctx);
798 tr_debug("trps_create_new_comm: unable to allocate new community.");
801 /* fill in the community with info */
802 tr_comm_set_id(comm, tr_dup_name(comm_id));
803 if (tr_comm_get_id(comm)==NULL) {
804 tr_debug("trps_create_new_comm: unable to allocate community name.");
808 tr_comm_set_type(comm, trp_inforec_get_comm_type(rec));
809 if (trp_inforec_get_apcs(rec)!=NULL) {
810 tr_comm_set_apcs(comm, tr_apc_dup(tmp_ctx, trp_inforec_get_apcs(rec)));
811 if (tr_comm_get_apcs(comm)==NULL) {
812 tr_debug("trps_create_new_comm: unable to allocate APC list.");
817 if (trp_inforec_get_owner_realm(rec)!=NULL) {
818 tr_comm_set_owner_realm(comm, tr_dup_name(trp_inforec_get_owner_realm(rec)));
819 if (tr_comm_get_owner_realm(comm)==NULL) {
820 tr_debug("trps_create_new_comm: unable to allocate owner realm name.");
825 if (trp_inforec_get_owner_contact(rec)!=NULL) {
826 tr_comm_set_owner_contact(comm, tr_dup_name(trp_inforec_get_owner_contact(rec)));
827 if (tr_comm_get_owner_contact(comm)==NULL) {
828 tr_debug("trps_create_new_comm: unable to allocate owner contact.");
833 comm->expiration_interval=trp_inforec_get_exp_interval(rec);
834 talloc_steal(mem_ctx, comm);
837 talloc_free(tmp_ctx);
841 static TR_RP_REALM *trps_create_new_rp_realm(TALLOC_CTX *mem_ctx, TR_NAME *realm_id, TRP_INFOREC *rec)
843 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
844 TR_RP_REALM *rp=tr_rp_realm_new(tmp_ctx);
847 tr_debug("trps_create_new_rp_realm: unable to allocate new realm.");
850 /* fill in the realm */
851 tr_rp_realm_set_id(rp, tr_dup_name(realm_id));
852 if (tr_rp_realm_get_id(rp)==NULL) {
853 tr_debug("trps_create_new_rp_realm: unable to allocate realm name.");
857 talloc_steal(mem_ctx, rp);
860 talloc_free(tmp_ctx);
864 static TR_IDP_REALM *trps_create_new_idp_realm(TALLOC_CTX *mem_ctx, TR_NAME *realm_id, TRP_INFOREC *rec)
866 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
867 TR_IDP_REALM *idp=tr_idp_realm_new(tmp_ctx);
870 tr_debug("trps_create_new_idp_realm: unable to allocate new realm.");
873 /* fill in the realm */
874 tr_idp_realm_set_id(idp, tr_dup_name(realm_id));
875 if (tr_idp_realm_get_id(idp)==NULL) {
876 tr_debug("trps_create_new_idp_realm: unable to allocate realm name.");
880 if (trp_inforec_get_apcs(rec)!=NULL) {
881 tr_idp_realm_set_apcs(idp, tr_apc_dup(tmp_ctx, trp_inforec_get_apcs(rec)));
882 if (tr_idp_realm_get_apcs(idp)==NULL) {
883 tr_debug("trps_create_new_idp_realm: unable to allocate APC list.");
888 idp->origin=TR_REALM_DISCOVERED;
890 talloc_steal(mem_ctx, idp);
893 talloc_free(tmp_ctx);
897 static TRP_RC trps_handle_inforec_comm(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC *rec)
899 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
900 TR_NAME *comm_id=trp_upd_get_comm(upd);
901 TR_NAME *realm_id=trp_upd_get_realm(upd);
902 TR_NAME *origin_id=NULL;
903 TR_NAME *our_peer_label=NULL;
905 TR_RP_REALM *rp_realm=NULL;
906 TR_IDP_REALM *idp_realm=NULL;
907 struct timespec expiry={0,0};
910 if ((comm_id==NULL) || (realm_id==NULL))
913 origin_id=trp_inforec_dup_origin(rec);
917 /* see whether we want to add this */
918 our_peer_label=trps_dup_label(trps);
919 if (our_peer_label==NULL) {
920 tr_debug("trps_handle_inforec_comm: unable to allocate peer label.");
924 if (trps_name_in_provenance(our_peer_label, trp_inforec_get_provenance(rec)))
925 tr_debug("trps_handle_inforec_comm: rejecting community inforec to avoid loop.");
927 /* no loop occurring, accept the update */
928 comm=tr_comm_table_find_comm(trps->ctable, comm_id);
930 tr_debug("trps_handle_inforec_comm: unknown community %.*s in inforec, creating it.",
931 comm_id->len, comm_id->buf);
932 comm=trps_create_new_comm(tmp_ctx, comm_id, rec);
934 tr_debug("trps_handle_inforec_comm: unable to create new community.");
937 tr_comm_table_add_comm(trps->ctable, comm);
939 /* TODO: see if other comm data match the new inforec and update or complain */
941 trps_compute_expiry(trps, trp_inforec_get_interval(rec), &expiry);
942 if ((expiry.tv_sec==0)&&(expiry.tv_nsec==0))
945 switch (trp_inforec_get_role(rec)) {
947 rp_realm=tr_rp_realm_lookup(trps->ctable->rp_realms, realm_id);
948 if (rp_realm==NULL) {
949 tr_debug("trps_handle_inforec_comm: unknown RP realm %.*s in inforec, creating it.",
950 realm_id->len, realm_id->buf);
951 rp_realm=trps_create_new_rp_realm(tmp_ctx, realm_id, rec);
952 if (rp_realm==NULL) {
953 tr_debug("trps_handle_inforec_comm: unable to create new RP realm.");
954 /* we may leave an unused community in the table, but it will only last until
955 * the next table sweep if it does not get any realms before that happens */
958 tr_comm_table_add_rp_realm(trps->ctable, rp_realm);
960 /* TODO: if realm existed, see if data match the new inforec and update or complain */
961 tr_comm_add_rp_realm(trps->ctable, comm, rp_realm, trp_inforec_get_interval(rec), trp_inforec_get_provenance(rec), &expiry);
962 tr_debug("trps_handle_inforec_comm: added RP realm %.*s to comm %.*s (origin %.*s).",
963 realm_id->len, realm_id->buf,
964 comm_id->len, comm_id->buf,
965 origin_id->len, origin_id->buf);
968 idp_realm=tr_idp_realm_lookup(trps->ctable->idp_realms, realm_id);
969 if (idp_realm==NULL) {
970 tr_debug("trps_handle_inforec_comm: unknown IDP realm %.*s in inforec, creating it.",
971 realm_id->len, realm_id->buf);
972 idp_realm=trps_create_new_idp_realm(tmp_ctx, realm_id, rec);
973 if (idp_realm==NULL) {
974 tr_debug("trps_handle_inforec_comm: unable to create new IDP realm.");
975 /* we may leave an unused community in the table, but it will only last until
976 * the next table sweep if it does not get any realms before that happens */
979 tr_comm_table_add_idp_realm(trps->ctable, idp_realm);
981 /* TODO: if realm existed, see if data match the new inforec and update or complain */
982 tr_comm_add_idp_realm(trps->ctable, comm, idp_realm, trp_inforec_get_interval(rec), trp_inforec_get_provenance(rec), &expiry);
983 tr_debug("trps_handle_inforec_comm: added IDP realm %.*s to comm %.*s (origin %.*s).",
984 realm_id->len, realm_id->buf,
985 comm_id->len, comm_id->buf,
986 origin_id->len, origin_id->buf);
989 tr_debug("trps_handle_inforec_comm: unable to add realm.");
997 if (our_peer_label!=NULL)
998 tr_free_name(our_peer_label);
1000 tr_free_name(origin_id);
1001 talloc_free(tmp_ctx);
1005 static TRP_RC trps_handle_update(TRPS_INSTANCE *trps, TRP_UPD *upd)
1007 TRP_INFOREC *rec=NULL;
1009 if (trps_validate_update(trps, upd) != TRP_SUCCESS) {
1010 tr_notice("trps_handle_update: received invalid TRP update.");
1014 for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
1015 /* validate/sanity check the record update */
1016 if (trps_validate_inforec(trps, rec) != TRP_SUCCESS) {
1017 tr_notice("trps_handle_update: invalid inforec in TRP update, discarding entire update.");
1022 for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
1023 switch (trp_inforec_get_type(rec)) {
1024 case TRP_INFOREC_TYPE_ROUTE:
1025 tr_debug("trps_handle_update: handling route inforec.");
1026 if (TRP_SUCCESS!=trps_handle_inforec_route(trps, upd, rec))
1027 tr_notice("trps_handle_update: error handling route inforec.");
1029 case TRP_INFOREC_TYPE_COMMUNITY:
1030 tr_debug("trps_handle_update: handling community inforec.");
1031 if (TRP_SUCCESS!=trps_handle_inforec_comm(trps, upd, rec))
1032 tr_notice("trps_handle_update: error handling community inforec.");
1034 tr_comm_table_print(stdout, trps->ctable); /* for debugging TODO remove */
1038 tr_notice("trps_handle_update: unsupported inforec in TRP update.");
1045 static TRP_RC trps_validate_request(TRPS_INSTANCE *trps, TRP_REQ *req)
1048 tr_notice("trps_validate_request: null TRP request.");
1052 if (trp_req_get_comm(req)==NULL) {
1053 tr_notice("trps_validate_request: received TRP request with null community.");
1057 if (trp_req_get_realm(req)==NULL) {
1058 tr_notice("trps_validate_request: received TRP request with null realm.");
1062 if (trp_req_get_peer(req)==NULL) {
1063 tr_notice("trps_validate_request: received TRP request without origin peer information.");
1070 /* choose the best route to comm/realm, optionally excluding routes to a particular peer */
1071 static TRP_ROUTE *trps_find_best_route(TRPS_INSTANCE *trps,
1074 TR_NAME *exclude_peer)
1076 TRP_ROUTE **entry=NULL;
1077 TRP_ROUTE *best=NULL;
1080 unsigned int kk_min=0;
1081 unsigned int min_metric=TRP_METRIC_INFINITY;
1083 entry=trp_rtable_get_realm_entries(trps->rtable, comm, realm, &n_entry);
1084 for (kk=0; kk<n_entry; kk++) {
1085 if (trp_route_get_metric(entry[kk]) < min_metric) {
1086 if ((exclude_peer==NULL) || (0!=tr_name_cmp(trp_route_get_peer(entry[kk]),
1089 min_metric=trp_route_get_metric(entry[kk]);
1093 if (trp_metric_is_finite(min_metric))
1100 /* TODO: think this through more carefully. At least ought to add hysteresis
1101 * to avoid flapping between routers or routes. */
1102 TRP_RC trps_update_active_routes(TRPS_INSTANCE *trps)
1104 size_t n_comm=0, ii=0;
1105 TR_NAME **comm=trp_rtable_get_comms(trps->rtable, &n_comm);
1106 size_t n_realm=0, jj=0;
1107 TR_NAME **realm=NULL;
1108 TRP_ROUTE *best_route=NULL, *cur_route=NULL;
1109 unsigned int best_metric=0, cur_metric=0;
1111 for (ii=0; ii<n_comm; ii++) {
1112 realm=trp_rtable_get_comm_realms(trps->rtable, comm[ii], &n_realm);
1113 for (jj=0; jj<n_realm; jj++) {
1114 best_route=trps_find_best_route(trps, comm[ii], realm[jj], NULL);
1115 if (best_route==NULL)
1116 best_metric=TRP_METRIC_INFINITY;
1118 best_metric=trp_route_get_metric(best_route);
1120 cur_route=trps_get_selected_route(trps, comm[ii], realm[jj]);
1121 if (cur_route!=NULL) {
1122 cur_metric=trp_route_get_metric(cur_route);
1123 if ((best_metric < cur_metric) && (trp_metric_is_finite(best_metric))) {
1124 /* The new route has a lower metric than the previous, and is finite. Accept. */
1125 trp_route_set_selected(cur_route, 0);
1126 trp_route_set_selected(best_route, 1);
1127 } else if (!trp_metric_is_finite(cur_metric)) /* rejects infinite or invalid metrics */
1128 trp_route_set_selected(cur_route, 0);
1129 } else if (trp_metric_is_finite(best_metric)) {
1130 trp_route_set_selected(best_route, 1);
1135 realm=NULL; n_realm=0;
1139 comm=NULL; n_comm=0;
1144 /* true if curtime >= expiry */
1145 static int trps_expired(struct timespec *expiry, struct timespec *curtime)
1147 return (tr_cmp_timespec(curtime, expiry) >= 0);
1150 /* Sweep for expired routes. For each expired route, if its metric is infinite, the route is flushed.
1151 * If its metric is finite, the metric is set to infinite and the route's expiration time is updated. */
1152 TRP_RC trps_sweep_routes(TRPS_INSTANCE *trps)
1154 struct timespec sweep_time={0,0};
1155 TRP_ROUTE **entry=NULL;
1159 /* use a single time for the entire sweep */
1160 if (0!=clock_gettime(CLOCK_REALTIME, &sweep_time)) {
1161 tr_err("trps_sweep_routes: could not read realtime clock.");
1162 sweep_time.tv_sec=0;
1163 sweep_time.tv_nsec=0;
1167 entry=trp_rtable_get_entries(trps->rtable, &n_entry); /* must talloc_free *entry */
1169 /* loop over the entries */
1170 for (ii=0; ii<n_entry; ii++) {
1171 if (!trp_route_is_local(entry[ii]) && trps_expired(trp_route_get_expiry(entry[ii]), &sweep_time)) {
1172 tr_debug("trps_sweep_routes: route expired.");
1173 if (!trp_metric_is_finite(trp_route_get_metric(entry[ii]))) {
1175 tr_debug("trps_sweep_routes: metric was infinity, flushing route.");
1176 trp_rtable_remove(trps->rtable, entry[ii]); /* entry[ii] is no longer valid */
1179 /* set metric to infinity and reset timer */
1180 tr_debug("trps_sweep_routes: setting metric to infinity and resetting expiry.");
1181 trp_route_set_metric(entry[ii], TRP_METRIC_INFINITY);
1182 trp_route_set_expiry(entry[ii], trps_compute_expiry(trps,
1183 trp_route_get_interval(entry[ii]),
1184 trp_route_get_expiry(entry[ii])));
1194 static char *timespec_to_str(struct timespec *ts)
1199 if (localtime_r(&(ts->tv_sec), &tm)==NULL)
1202 s=malloc(40); /* long enough to contain strftime result */
1206 if (strftime(s, 40, "%F %T", &tm)==0) {
1214 /* Sweep for expired communities/realms/memberships. */
1215 TRP_RC trps_sweep_ctable(TRPS_INSTANCE *trps)
1217 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1218 struct timespec sweep_time={0,0};
1219 TR_COMM_MEMB *memb=NULL;
1220 TR_COMM_ITER *iter=NULL;
1221 TRP_RC rc=TRP_ERROR;
1223 /* use a single time for the entire sweep */
1224 if (0!=clock_gettime(CLOCK_REALTIME, &sweep_time)) {
1225 tr_err("trps_sweep_ctable: could not read realtime clock.");
1226 sweep_time.tv_sec=0;
1227 sweep_time.tv_nsec=0;
1231 /* iterate all memberships */
1232 iter=tr_comm_iter_new(tmp_ctx);
1234 tr_err("trps_sweep_ctable: unable to allocate iterator.");
1238 for (memb=tr_comm_memb_iter_all_first(iter, trps->ctable);
1240 memb=tr_comm_memb_iter_all_next(iter)) {
1241 if (tr_comm_memb_get_origin(memb)==NULL)
1242 continue; /* do not expire local entries */
1244 if (tr_comm_memb_is_expired(memb, &sweep_time)) {
1245 if (tr_comm_memb_get_times_expired(memb)>0) {
1246 /* Already expired once; flush. */
1247 tr_debug("trps_sweep_ctable: flushing expired community membership (%.*s in %.*s, origin %.*s, expired %s).",
1248 tr_comm_memb_get_realm_id(memb)->len, tr_comm_memb_get_realm_id(memb)->buf,
1249 tr_comm_get_id(tr_comm_memb_get_comm(memb))->len, tr_comm_get_id(tr_comm_memb_get_comm(memb))->buf,
1250 tr_comm_memb_get_origin(memb)->len, tr_comm_memb_get_origin(memb)->buf,
1251 timespec_to_str(tr_comm_memb_get_expiry(memb)));
1252 tr_comm_table_remove_memb(trps->ctable, memb);
1253 tr_comm_memb_free(memb);
1255 /* This is the first expiration. Note this and reset the expiry time. */
1256 tr_comm_memb_expire(memb);
1257 trps_compute_expiry(trps, tr_comm_memb_get_interval(memb), tr_comm_memb_get_expiry(memb));
1258 tr_debug("trps_sweep_ctable: community membership expired, resetting expiry to %s (%.*s in %.*s, origin %.*s).",
1259 timespec_to_str(tr_comm_memb_get_expiry(memb)),
1260 tr_comm_memb_get_realm_id(memb)->len, tr_comm_memb_get_realm_id(memb)->buf,
1261 tr_comm_get_id(tr_comm_memb_get_comm(memb))->len, tr_comm_get_id(tr_comm_memb_get_comm(memb))->buf,
1262 tr_comm_memb_get_origin(memb)->len, tr_comm_memb_get_origin(memb)->buf);
1267 /* get rid of any unreferenced realms, etc */
1268 tr_comm_table_sweep(trps->ctable);
1271 talloc_free(tmp_ctx);
1276 static unsigned int trps_metric_add(unsigned int m1, unsigned int m2)
1278 if (trp_metric_is_invalid(m1) || trp_metric_is_invalid(m2))
1279 return TRP_METRIC_INVALID;
1281 if (trp_metric_is_infinite(m1) || trp_metric_is_infinite(m2))
1282 return TRP_METRIC_INFINITY;
1284 if (trp_metric_is_finite(m1+m2))
1287 return TRP_METRIC_INFINITY;
1290 /* convert an rentry into a new trp update info record */
1291 static TRP_INFOREC *trps_route_to_inforec(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TRP_ROUTE *route)
1293 TRP_INFOREC *rec=trp_inforec_new(mem_ctx, TRP_INFOREC_TYPE_ROUTE);
1294 unsigned int linkcost=0;
1297 if (trp_route_is_local(route))
1300 linkcost=trp_peer_get_linkcost(trps_get_peer_by_gssname(trps,
1301 trp_route_get_peer(route)));
1304 /* Note that we leave the next hop empty since the recipient fills that in.
1305 * This is where we add the link cost (currently always 1) to the next peer. */
1306 if ((trp_inforec_set_trust_router(rec, trp_route_dup_trust_router(route)) != TRP_SUCCESS)
1307 ||(trp_inforec_set_metric(rec,
1308 trps_metric_add(trp_route_get_metric(route),
1309 linkcost)) != TRP_SUCCESS)
1310 ||(trp_inforec_set_interval(rec, trps_get_update_interval(trps)) != TRP_SUCCESS)) {
1311 tr_err("trps_route_to_inforec: error creating route update.");
1319 static TRP_UPD *trps_route_to_upd(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TRP_ROUTE *route)
1321 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1322 TRP_UPD *upd=trp_upd_new(tmp_ctx);
1323 TRP_INFOREC *rec=NULL;
1326 tr_err("trps_route_to_upd: could not create update message.");
1329 trp_upd_set_realm(upd, trp_route_dup_realm(route));
1330 if (trp_upd_get_realm(upd)==NULL) {
1331 tr_err("trps_route_to_upd: could not copy realm.");
1332 upd=NULL; /* it's still in tmp_ctx, so it will be freed */
1335 trp_upd_set_comm(upd, trp_route_dup_comm(route));
1336 if (trp_upd_get_comm(upd)==NULL) {
1337 tr_err("trps_route_to_upd: could not copy comm.");
1338 upd=NULL; /* it's still in tmp_ctx, so it will be freed */
1341 rec=trps_route_to_inforec(tmp_ctx, trps, route);
1343 tr_err("trps_route_to_upd: could not create route info record for realm %.*s in comm %.*s.",
1344 trp_route_get_realm(route)->len, trp_route_get_realm(route)->buf,
1345 trp_route_get_comm(route)->len, trp_route_get_comm(route)->buf);
1346 upd=NULL; /* it's till in tmp_ctx, so it will be freed */
1349 trp_upd_add_inforec(upd, rec);
1352 talloc_steal(mem_ctx, upd);
1355 talloc_free(tmp_ctx);
1359 /* select the correct route to comm/realm to be announced to peer */
1360 static TRP_ROUTE *trps_select_realm_update(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer_gssname)
1364 /* Take the currently selected route unless it is through the peer we're sending the update to.
1365 * I.e., enforce the split horizon rule. */
1366 route=trp_rtable_get_selected_entry(trps->rtable, comm, realm);
1368 /* No selected route, this should only happen if the only route has been retracted,
1369 * in which case we do not want to advertise it. */
1372 tr_debug("trps_select_realm_update: %s vs %s", peer_gssname->buf,
1373 trp_route_get_peer(route)->buf);
1374 if (0==tr_name_cmp(peer_gssname, trp_route_get_peer(route))) {
1375 tr_debug("trps_select_realm_update: matched, finding alternate route");
1376 /* the selected entry goes through the peer we're reporting to, choose an alternate */
1377 route=trps_find_best_route(trps, comm, realm, peer_gssname);
1378 if ((route==NULL) || (!trp_metric_is_finite(trp_route_get_metric(route))))
1379 return NULL; /* don't advertise a nonexistent or retracted route */
1384 /* Add TRP_UPD msgs to the updates GPtrArray. Caller needs to arrange for these to be freed. */
1385 static TRP_RC trps_select_route_updates_for_peer(TALLOC_CTX *mem_ctx,
1387 TRPS_INSTANCE *trps,
1388 TR_NAME *peer_gssname,
1392 TR_NAME **comm=trp_rtable_get_comms(trps->rtable, &n_comm);
1393 TR_NAME **realm=NULL;
1396 TRP_ROUTE *best=NULL;
1402 for (ii=0; ii<n_comm; ii++) {
1403 realm=trp_rtable_get_comm_realms(trps->rtable, comm[ii], &n_realm);
1404 for (jj=0; jj<n_realm; jj++) {
1405 best=trps_select_realm_update(trps, comm[ii], realm[jj], peer_gssname);
1406 /* If we found a route, add it to the list. If triggered!=0, then only
1407 * add triggered routes. */
1408 if ((best!=NULL) && ((!triggered) || trp_route_is_triggered(best))) {
1409 upd=trps_route_to_upd(mem_ctx, trps, best);
1411 tr_err("trps_select_route_updates_for_peer: unable to create update message.");
1414 g_ptr_array_add(updates, upd);
1430 static TRP_INFOREC *trps_memb_to_inforec(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TR_COMM_MEMB *memb)
1432 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1433 TRP_INFOREC *rec=NULL;
1439 comm=tr_comm_memb_get_comm(memb);
1440 rec=trp_inforec_new(tmp_ctx, TRP_INFOREC_TYPE_COMMUNITY);
1444 if (TRP_SUCCESS!=trp_inforec_set_comm_type(rec, tr_comm_get_type(comm))) {
1449 if (TRP_SUCCESS!=trp_inforec_set_role(rec, tr_comm_memb_get_role(memb))) {
1454 if ((NULL!=tr_comm_get_apcs(comm)) &&
1455 ( (TRP_SUCCESS!=trp_inforec_set_apcs(rec,
1456 tr_apc_dup(rec, tr_comm_get_apcs(comm)))) ||
1457 (NULL==trp_inforec_get_apcs(rec)))) {
1462 if ((NULL!=tr_comm_get_owner_realm(comm)) &&
1463 ( (TRP_SUCCESS!=trp_inforec_set_owner_realm(rec, tr_dup_name(tr_comm_get_owner_realm(comm)))) ||
1464 (NULL==trp_inforec_get_owner_realm(rec)))) {
1469 if ((NULL!=tr_comm_get_owner_contact(comm)) &&
1470 ( (TRP_SUCCESS!=trp_inforec_set_owner_contact(rec, tr_dup_name(tr_comm_get_owner_contact(comm)))) ||
1471 (NULL==trp_inforec_get_owner_contact(rec)))) {
1476 if ((NULL!=tr_comm_memb_get_provenance(memb)) &&
1477 (TRP_SUCCESS!=trp_inforec_set_provenance(rec, tr_comm_memb_get_provenance(memb)))) {
1482 if (TRP_SUCCESS!=trp_inforec_set_interval(rec, trps_get_update_interval(trps))) {
1488 talloc_steal(mem_ctx, rec);
1491 talloc_free(tmp_ctx);
1495 /* construct an update with all the inforecs for comm/realm/role to be sent to peer */
1496 static TRP_UPD *trps_comm_update(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TR_NAME *peer_gssname, TR_COMM *comm, TR_REALM *realm)
1498 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1499 TRP_UPD *upd=trp_upd_new(tmp_ctx);
1500 TRP_INFOREC *rec=NULL;
1501 TR_COMM_ITER *iter=NULL;
1502 TR_COMM_MEMB *memb=NULL;
1507 trp_upd_set_comm(upd, tr_comm_dup_id(comm));
1508 trp_upd_set_realm(upd, tr_realm_dup_id(realm));
1509 /* leave peer empty */
1511 iter=tr_comm_iter_new(tmp_ctx);
1513 tr_err("trps_comm_update: unable to allocate iterator.");
1518 /* now add inforecs */
1519 switch (realm->role) {
1521 memb=tr_comm_table_find_idp_memb(trps->ctable,
1522 tr_realm_get_id(realm),
1523 tr_comm_get_id(comm));
1526 memb=tr_comm_table_find_rp_memb(trps->ctable,
1527 tr_realm_get_id(realm),
1528 tr_comm_get_id(comm));
1534 for (memb=tr_comm_memb_iter_first(iter, memb);
1536 memb=tr_comm_memb_iter_next(iter)) {
1537 rec=trps_memb_to_inforec(tmp_ctx, trps, memb);
1539 tr_err("trps_comm_update: unable to allocate inforec.");
1543 trp_upd_add_inforec(upd, rec);
1547 if (trp_upd_get_inforec(upd)==NULL)
1548 upd=NULL; /* no inforecs, no reason to send the update */
1550 talloc_steal(mem_ctx, upd); /* success! */
1553 talloc_free(tmp_ctx);
1557 /* Find all community updates to send to a peer and add these as TR_UPD records
1558 * to the updates GPtrArray. */
1559 static TRP_RC trps_select_comm_updates_for_peer(TALLOC_CTX *mem_ctx,
1561 TRPS_INSTANCE *trps,
1562 TR_NAME *peer_gssname,
1565 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1566 TR_COMM_ITER *comm_iter=NULL;
1568 TR_COMM_ITER *realm_iter=NULL;
1569 TR_REALM *realm=NULL;
1571 TRP_RC rc=TRP_ERROR;
1573 /* currently do not send any communities on triggered updates */
1579 comm_iter=tr_comm_iter_new(tmp_ctx);
1580 realm_iter=tr_comm_iter_new(tmp_ctx);
1581 if ((comm_iter==NULL) || (realm_iter==NULL)) {
1582 tr_err("trps_select_comm_updates_for_peer: unable to allocate iterator.");
1587 /* do every community */
1588 for (comm=tr_comm_table_iter_first(comm_iter, trps->ctable);
1590 comm=tr_comm_table_iter_next(comm_iter)) {
1591 /* do every realm in this community */
1592 tr_debug("trps_select_comm_updates_for_peer: looking through community %.*s",
1593 tr_comm_get_id(comm)->len,
1594 tr_comm_get_id(comm)->buf);
1595 for (realm=tr_realm_iter_first(realm_iter, trps->ctable, tr_comm_get_id(comm));
1597 realm=tr_realm_iter_next(realm_iter)) {
1598 /* get the update for this comm/realm */
1599 tr_debug("trps_select_comm_updates_for_peer: adding realm %.*s",
1600 tr_realm_get_id(realm)->len,
1601 tr_realm_get_id(realm)->buf);
1602 upd=trps_comm_update(mem_ctx, trps, peer_gssname, comm, realm);
1604 g_ptr_array_add(updates, upd);
1609 talloc_free(tmp_ctx);
1614 /* helper for trps_update_one_peer. Frees the TRP_UPD pointed to by a GPtrArray element */
1615 static void trps_trp_upd_destroy(gpointer data)
1617 trp_upd_free((TRP_UPD *)data);
1620 /* all routes/communities to a single peer, unless comm/realm are specified (both or neither must be NULL) */
1621 static TRP_RC trps_update_one_peer(TRPS_INSTANCE *trps,
1623 TRP_UPDATE_TYPE update_type,
1627 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1628 TR_MSG msg; /* not a pointer! */
1630 TRP_ROUTE *route=NULL;
1633 TRP_RC rc=TRP_ERROR;
1634 TR_NAME *peer_label=trp_peer_get_label(peer);
1635 GPtrArray *updates=g_ptr_array_new_with_free_func(trps_trp_upd_destroy);
1637 if (updates==NULL) {
1638 tr_err("trps_update_one_peer: unable to allocate updates array.");
1643 switch (update_type) {
1644 case TRP_UPDATE_TRIGGERED:
1645 tr_debug("trps_update_one_peer: preparing triggered update for %.*s",
1646 peer_label->len, peer_label->buf);
1648 case TRP_UPDATE_SCHEDULED:
1649 tr_debug("trps_update_one_peer: preparing scheduled update for %.*s",
1650 peer_label->len, peer_label->buf);
1652 case TRP_UPDATE_REQUESTED:
1653 tr_debug("trps_update_one_peer: preparing requested update for %.*s",
1654 peer_label->len, peer_label->buf);
1657 tr_err("trps_update_one_peer: invalid update type requested.");
1662 /* First, gather route updates. */
1663 tr_debug("trps_update_one_peer: selecting route updates for %.*s.", peer_label->len, peer_label->buf);
1664 if ((comm==NULL) && (realm==NULL)) {
1666 rc=trps_select_route_updates_for_peer(tmp_ctx,
1670 update_type==TRP_UPDATE_TRIGGERED);
1671 } else if ((comm!=NULL) && (realm!=NULL)) {
1672 /* a single community/realm was requested */
1673 route=trps_select_realm_update(trps, comm, realm, peer_label);
1675 /* we have no actual update to send back, MUST send a retraction */
1676 tr_debug("trps_update_one_peer: community/realm without route requested, sending mandatory retraction.");
1677 route=trp_route_new(tmp_ctx);
1678 trp_route_set_comm(route, tr_dup_name(comm));
1679 trp_route_set_realm(route, tr_dup_name(realm));
1680 trp_route_set_peer(route, tr_new_name(""));
1681 trp_route_set_metric(route, TRP_METRIC_INFINITY);
1682 trp_route_set_trust_router(route, tr_new_name(""));
1683 trp_route_set_next_hop(route, tr_new_name(""));
1685 upd=trps_route_to_upd(tmp_ctx, trps, route);
1687 tr_err("trps_update_one_peer: unable to allocate route update.");
1691 g_ptr_array_add(updates, upd);
1693 tr_err("trps_update_one_peer: error: only comm or realm was specified. Need both or neither.");
1698 /* Second, gather community updates */
1699 tr_debug("trps_update_one_peer: selecting community updates for %.*s.", peer_label->len, peer_label->buf);
1700 rc=trps_select_comm_updates_for_peer(tmp_ctx, updates, trps, peer_label, update_type==TRP_UPDATE_TRIGGERED);
1702 /* see if we have anything to send */
1703 if (updates->len<=0)
1704 tr_debug("trps_update_one_peer: no updates for %.*s", peer_label->len, peer_label->buf);
1706 tr_debug("trps_update_one_peer: sending %d update messages.", updates->len);
1707 for (ii=0; ii<updates->len; ii++) {
1708 upd=(TRP_UPD *)g_ptr_array_index(updates, ii);
1709 /* now encode the update message */
1710 tr_msg_set_trp_upd(&msg, upd);
1711 encoded=tr_msg_encode(&msg);
1712 if (encoded==NULL) {
1713 tr_err("trps_update_one_peer: error encoding update.");
1718 tr_debug("trps_update_one_peer: adding message to queue.");
1719 if (trps_send_msg(trps, peer, encoded) != TRP_SUCCESS)
1720 tr_err("trps_update_one_peer: error queueing update.");
1722 tr_debug("trps_update_one_peer: update queued successfully.");
1724 tr_msg_free_encoded(encoded);
1733 g_ptr_array_free(updates, TRUE); /* frees any TRP_UPD records */
1734 talloc_free(tmp_ctx);
1738 /* all routes/communities to all peers */
1739 TRP_RC trps_update(TRPS_INSTANCE *trps, TRP_UPDATE_TYPE update_type)
1741 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1742 TRP_PTABLE_ITER *iter=trp_ptable_iter_new(tmp_ctx);
1743 TRP_PEER *peer=NULL;
1744 TRP_RC rc=TRP_SUCCESS;
1746 if (trps->ptable==NULL)
1747 return TRP_SUCCESS; /* no peers, nothing to do */
1750 tr_err("trps_update: failed to allocate peer table iterator.");
1751 talloc_free(tmp_ctx);
1755 for (peer=trp_ptable_iter_first(iter, trps->ptable);
1756 (peer!=NULL) && (rc==TRP_SUCCESS);
1757 peer=trp_ptable_iter_next(iter))
1759 if (!trps_peer_connected(trps, peer)) {
1760 TR_NAME *peer_label=trp_peer_get_label(peer);
1761 tr_debug("trps_update: no TRP connection to %.*s, skipping.",
1762 peer_label->len, peer_label->buf);
1765 rc=trps_update_one_peer(trps, peer, update_type, NULL, NULL);
1768 tr_debug("trps_update: rc=%u after attempting update.", rc);
1769 trp_ptable_iter_free(iter);
1770 trp_rtable_clear_triggered(trps->rtable); /* don't re-send triggered updates */
1771 talloc_free(tmp_ctx);
1775 TRP_RC trps_add_route(TRPS_INSTANCE *trps, TRP_ROUTE *route)
1777 trp_rtable_add(trps->rtable, route); /* should return status */
1781 /* steals the peer object */
1782 TRP_RC trps_add_peer(TRPS_INSTANCE *trps, TRP_PEER *peer)
1784 if (trps->ptable==NULL) {
1785 trps->ptable=trp_ptable_new(trps);
1786 if (trps->ptable==NULL)
1789 return trp_ptable_add(trps->ptable, peer);
1792 TRP_PEER *trps_get_peer_by_gssname(TRPS_INSTANCE *trps, TR_NAME *gssname)
1794 if (trps->ptable==NULL)
1797 return trp_ptable_find_gss_name(trps->ptable, gssname);
1800 TRP_PEER *trps_get_peer_by_servicename(TRPS_INSTANCE *trps, TR_NAME *servicename)
1802 if (trps->ptable==NULL)
1805 return trp_ptable_find_servicename(trps->ptable, servicename);
1808 int trps_peer_connected(TRPS_INSTANCE *trps, TRP_PEER *peer)
1810 TRPC_INSTANCE *trpc=trps_find_trpc(trps, peer);
1814 if (trpc_get_status(trpc)==TRP_CONNECTION_UP)
1821 static TRP_RC trps_handle_request(TRPS_INSTANCE *trps, TRP_REQ *req)
1824 TR_NAME *realm=NULL;
1826 tr_debug("trps_handle_request: handling TRP request.");
1828 if (trps_validate_request(trps, req) != TRP_SUCCESS) {
1829 tr_notice("trps_handle_request: received invalid TRP request.");
1833 if (!trp_req_is_wildcard(req)) {
1834 comm=trp_req_get_comm(req);
1835 realm=trp_req_get_realm(req);
1836 tr_debug("trps_handle_request: route for %.*s/%.*s requested.",
1837 comm->len, comm->buf, realm->len, realm->buf);
1839 tr_debug("trps_handle_request: all routes requested.");
1840 /* leave comm/realm NULL */
1842 return trps_update_one_peer(trps,
1843 trps_get_peer_by_gssname(trps, trp_req_get_peer(req)),
1844 TRP_UPDATE_REQUESTED,
1850 TRP_RC trps_handle_tr_msg(TRPS_INSTANCE *trps, TR_MSG *tr_msg)
1852 TRP_RC rc=TRP_ERROR;
1854 switch (tr_msg_get_msg_type(tr_msg)) {
1856 rc=trps_handle_update(trps, tr_msg_get_trp_upd(tr_msg));
1857 if (rc==TRP_SUCCESS) {
1858 rc=trps_update_active_routes(trps);
1859 trps_update(trps, TRP_UPDATE_TRIGGERED); /* send any triggered routes */
1864 rc=trps_handle_request(trps, tr_msg_get_trp_req(tr_msg));
1868 /* unknown error or one we don't care about (e.g., TID messages) */
1873 /* send wildcard route request to a peer */
1874 TRP_RC trps_wildcard_route_req(TRPS_INSTANCE *trps, TR_NAME *peer_servicename)
1876 TALLOC_CTX *tmp_ctx=talloc_new(NULL);
1877 TRP_PEER *peer=trps_get_peer_by_servicename(trps, peer_servicename);
1878 TR_MSG msg; /* not a pointer */
1879 TRP_REQ *req=trp_req_new(tmp_ctx);
1881 TRP_RC rc=TRP_ERROR;
1884 tr_err("trps_wildcard_route_req: unknown peer (%.*s).", peer_servicename->len, peer_servicename->buf);
1888 if ((req==NULL) || (trp_req_make_wildcard(req)!=TRP_SUCCESS)) {
1889 tr_err("trps_wildcard_route_req: unable to create wildcard TRP request.");
1894 tr_msg_set_trp_req(&msg, req);
1895 encoded=tr_msg_encode(&msg);
1896 if (encoded==NULL) {
1897 tr_err("trps_wildcard_route_req: error encoding wildcard TRP request.");
1902 tr_debug("trps_wildcard_route_req: adding message to queue.");
1903 if (trps_send_msg(trps, peer, encoded) != TRP_SUCCESS) {
1904 tr_err("trps_wildcard_route_req: error queueing request.");
1907 tr_debug("trps_wildcard_route_req: request queued successfully.");
1913 tr_msg_free_encoded(encoded);
1917 talloc_free(tmp_ctx);