X-Git-Url: http://www.project-moonshot.org/gitweb/?a=blobdiff_plain;f=trp%2Ftrps.c;h=b737aba7ab2279fc6efed2016c6fa43cd5d183f4;hb=e404e4c6cbddaca8fcb9a563ce20fb19d6a722a3;hp=1997c54bc266500214498d8a836bec849206058b;hpb=ee6787f44e1c2507add20b083a2d652b34632f32;p=trust_router.git diff --git a/trp/trps.c b/trp/trps.c index 1997c54..b737aba 100644 --- a/trp/trps.c +++ b/trp/trps.c @@ -1,17 +1,60 @@ +/* + * Copyright (c) 2016, JANET(UK) + * All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * 3. Neither the name of JANET(UK) nor the names of its contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS + * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE + * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, + * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES + * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) + * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, + * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED + * OF THE POSSIBILITY OF SUCH DAMAGE. + * + */ + #include #include #include #include #include +#include +#include +#include // for nfds_t #include +#include +#include #include -#include +#include +#include #include +#include +#include #include #include #include - +#include +#include static int trps_destructor(void *object) { @@ -32,6 +75,7 @@ TRPS_INSTANCE *trps_new (TALLOC_CTX *mem_ctx) trps->trpc=NULL; trps->update_interval=(struct timeval){0,0}; trps->sweep_interval=(struct timeval){0,0}; + trps->ptable=NULL; trps->mq=tr_mq_new(trps); if (trps->mq==NULL) { @@ -40,13 +84,6 @@ TRPS_INSTANCE *trps_new (TALLOC_CTX *mem_ctx) return NULL; } - trps->ptable=trp_ptable_new(trps); - if (trps->ptable==NULL) { - /* failed to allocate ptable */ - talloc_free(trps); - return NULL; - } - trps->rtable=NULL; if (trps_init_rtable(trps) != TRP_SUCCESS) { /* failed to allocate rtable */ @@ -87,7 +124,7 @@ void trps_free (TRPS_INSTANCE *trps) TR_MQ_MSG *trps_mq_pop(TRPS_INSTANCE *trps) { - return tr_mq_pop(trps->mq); + return tr_mq_pop(trps->mq, 0); } void trps_mq_add(TRPS_INSTANCE *trps, TR_MQ_MSG *msg) @@ -128,15 +165,56 @@ void trps_set_sweep_interval(TRPS_INSTANCE *trps, unsigned int interval) trps->sweep_interval.tv_usec=0; } +void trps_set_ctable(TRPS_INSTANCE *trps, TR_COMM_TABLE *comm) +{ + trps->ctable=comm; +} + +void trps_set_ptable(TRPS_INSTANCE *trps, TRP_PTABLE *ptable) +{ + if (trps->ptable!=NULL) + trp_ptable_free(trps->ptable); + trps->ptable=ptable; +} + +void trps_set_peer_status_callback(TRPS_INSTANCE *trps, void (*cb)(TRP_PEER *, void *), void *cookie) +{ + TRP_PTABLE_ITER *iter=NULL; + TRP_PEER *peer=NULL; + if (trps->ptable==NULL) + return; + + iter=trp_ptable_iter_new(NULL); + for (peer=trp_ptable_iter_first(iter, trps->ptable); peer!=NULL; peer=trp_ptable_iter_next(iter)) + trp_peer_set_conn_status_cb(peer, cb, cookie); + trp_ptable_iter_free(iter); +} + +/* Get the label peers will know us by - needs to match trp_peer_get_label() output. + * There is no get, only dup, because we don't store the label except when requested. */ +TR_NAME *trps_dup_label(TRPS_INSTANCE *trps) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TR_NAME *label=NULL; + char *s=talloc_asprintf(tmp_ctx, "%s:%u", trps->hostname, trps->port); + if (s==NULL) + goto cleanup; + label=tr_new_name(s); + +cleanup: + talloc_free(tmp_ctx); + return label; +} + TRPC_INSTANCE *trps_find_trpc(TRPS_INSTANCE *trps, TRP_PEER *peer) { TRPC_INSTANCE *cur=NULL; TR_NAME *name=NULL; - TR_NAME *peer_gssname=trp_peer_get_gssname(peer); + TR_NAME *peer_servicename=trp_peer_get_servicename(peer); for (cur=trps->trpc; cur!=NULL; cur=trpc_get_next(cur)) { name=trpc_get_gssname(cur); - if ((name!=NULL) && (0==tr_name_cmp(peer_gssname, name))) { + if ((name!=NULL) && (0==tr_name_cmp(peer_servicename, name))) { break; } } @@ -187,8 +265,11 @@ TRP_RC trps_send_msg(TRPS_INSTANCE *trps, TRP_PEER *peer, const char *msg) /* get the connection for this peer */ trpc=trps_find_trpc(trps, peer); - /* instead, let's let that happen and then clear the queue when an attempt to - * connect fails */ + /* The peer connection (trpc) usually exists even if the connection is down. + * We will queue messages even if the connection is down. To prevent this from + * endlessly increasing the size of the queue, the trpc handler needs to clear + * its queue periodically, even if it is unable to send the messages + */ if (trpc==NULL) { tr_warning("trps_send_msg: skipping message queued for missing TRP client entry."); } else { @@ -202,38 +283,6 @@ TRP_RC trps_send_msg(TRPS_INSTANCE *trps, TRP_PEER *peer, const char *msg) return rc; } -static int trps_listen (TRPS_INSTANCE *trps, int port) -{ - int rc = 0; - int conn = -1; - int optval = 1; - - union { - struct sockaddr_storage storage; - struct sockaddr_in in4; - } addr; - - struct sockaddr_in *saddr = (struct sockaddr_in *) &addr.in4; - - saddr->sin_port = htons (port); - saddr->sin_family = AF_INET; - saddr->sin_addr.s_addr = INADDR_ANY; - - if (0 > (conn = socket (AF_INET, SOCK_STREAM, 0))) - return conn; - - setsockopt(conn, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval)); - - if (0 > (rc = bind (conn, (struct sockaddr *) saddr, sizeof(struct sockaddr_in)))) - return rc; - - if (0 > (rc = listen(conn, 512))) - return rc; - - tr_debug("trps_listen: TRP Server listening on port %d", port); - return conn; -} - /* get the currently selected route if available */ TRP_ROUTE *trps_get_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer) { @@ -242,6 +291,7 @@ TRP_ROUTE *trps_get_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR TRP_ROUTE *trps_get_selected_route(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm) { + tr_debug("trps_get_selected_route: entered. trps=%p, comm=%p, realm=%p", trps, comm, realm); return trp_rtable_get_selected_entry(trps->rtable, comm, realm); } @@ -274,7 +324,8 @@ static TRP_RC trps_read_message(TRPS_INSTANCE *trps, TRP_CONNECTION *conn, TR_MS int err=0; char *buf=NULL; size_t buflen = 0; - TR_NAME *peer=NULL; + TRP_PEER *peer=NULL; /* entry in the peer table */ + TR_NAME *conn_peer=NULL; /* name from the TRP_CONN, which comes from the gss context */ tr_debug("trps_read_message: started"); if (err = gsscon_read_encrypted_token(trp_connection_get_fd(conn), @@ -287,27 +338,41 @@ static TRP_RC trps_read_message(TRPS_INSTANCE *trps, TRP_CONNECTION *conn, TR_MS return TRP_ERROR; } - tr_debug("trps_read_message(): message received, %u bytes.", (unsigned) buflen); - tr_debug("trps_read_message(): %.*s", buflen, buf); + tr_debug("trps_read_message: message received, %u bytes.", (unsigned) buflen); + tr_debug("trps_read_message: %.*s", buflen, buf); - *msg=tr_msg_decode(buf, buflen); + *msg= tr_msg_decode(NULL, buf, buflen); free(buf); if (*msg==NULL) return TRP_NOPARSE; - peer=trp_connection_get_peer(conn); + conn_peer=trp_connection_get_peer(conn); + if (conn_peer==NULL) { + tr_err("trps_read_message: connection has no peer name"); + return TRP_ERROR; + } + + peer=trps_get_peer_by_gssname(trps, conn_peer); + if (peer==NULL) { + tr_err("trps_read_message: could not find peer with gssname=%s", trp_connection_get_gssname(conn)); + return TRP_ERROR; + } + /* verify we received a message we support, otherwise drop it now */ switch (tr_msg_get_msg_type(*msg)) { case TRP_UPDATE: - trp_upd_set_peer(tr_msg_get_trp_upd(*msg), tr_dup_name(peer)); + trp_upd_set_peer(tr_msg_get_trp_upd(*msg), tr_dup_name(conn_peer)); + 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 */ + /* update provenance if necessary */ + trp_upd_add_to_provenance(tr_msg_get_trp_upd(*msg), trp_peer_get_label(peer)); break; case TRP_REQUEST: - trp_req_set_peer(tr_msg_get_trp_req(*msg), tr_dup_name(peer)); + trp_req_set_peer(tr_msg_get_trp_req(*msg), tr_dup_name(conn_peer)); break; default: - tr_debug("trps_read_message: received unsupported message from %.*s", peer->len, peer->buf); + tr_debug("trps_read_message: received unsupported message from %.*s", conn_peer->len, conn_peer->buf); tr_msg_free_decoded(*msg); *msg=NULL; return TRP_UNSUPPORTED; @@ -321,32 +386,36 @@ int trps_get_listener(TRPS_INSTANCE *trps, TRP_AUTH_FUNC auth_handler, const char *hostname, unsigned int port, - void *cookie) + void *cookie, + int *fd_out, + size_t max_fd) { - int listen = -1; + nfds_t n_fd=0; + nfds_t ii=0; - if (0 > (listen = trps_listen(trps, port))) { - char errbuf[256]; - if (0 == strerror_r(errno, errbuf, 256)) { - tr_debug("trps_get_listener: Error opening port %d: %s.", port, errbuf); - } else { - tr_debug("trps_get_listener: Unknown error openining port %d.", port); - } - } + n_fd = tr_sock_listen_all(port, fd_out, max_fd); - if (listen > 0) { + if (n_fd == 0) + tr_err("trps_get_listener: Error opening port %d."); + else { /* opening port succeeded */ - tr_debug("trps_get_listener: Opened port %d.", port); + tr_info("trps_get_listener: Opened port %d.", port); - /* make this socket non-blocking */ - if (0 != fcntl(listen, F_SETFL, O_NONBLOCK)) { - tr_debug("trps_get_listener: Error setting O_NONBLOCK."); - close(listen); - listen=-1; + /* make the sockets non-blocking */ + for (ii=0; ii 0) { + if (n_fd > 0) { /* store the caller's request handler & cookie */ trps->msg_handler = msg_handler; trps->auth_handler = auth_handler; @@ -355,7 +424,7 @@ int trps_get_listener(TRPS_INSTANCE *trps, trps->cookie = cookie; } - return listen; + return (int) n_fd; } TRP_RC trps_authorize_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *conn) @@ -395,6 +464,7 @@ void trps_handle_connection(TRPS_INSTANCE *trps, TRP_CONNECTION *conn) tr_debug("trps_handle_connection: connection closed."); } +/* TODO: check realm/comm, now part of the update instead of inforec */ static TRP_RC trps_validate_update(TRPS_INSTANCE *trps, TRP_UPD *upd) { if (upd==NULL) { @@ -402,6 +472,16 @@ static TRP_RC trps_validate_update(TRPS_INSTANCE *trps, TRP_UPD *upd) return TRP_BADARG; } + if (trp_upd_get_realm(upd)==NULL) { + tr_notice("trps_validate_update: received TRP update without realm."); + return TRP_ERROR; + } + + if (trp_upd_get_comm(upd)==NULL) { + tr_notice("trps_validate_update: received TRP update without community."); + return TRP_ERROR; + } + if (trp_upd_get_inforec(upd)==NULL) { tr_notice("trps_validate_update: received TRP update with no info records."); return TRP_ERROR; @@ -411,6 +491,7 @@ static TRP_RC trps_validate_update(TRPS_INSTANCE *trps, TRP_UPD *upd) tr_notice("trps_validate_update: received TRP update without origin peer information."); return TRP_ERROR; } + return TRP_SUCCESS; } @@ -420,9 +501,7 @@ static TRP_RC trps_validate_inforec(TRPS_INSTANCE *trps, TRP_INFOREC *rec) { switch(trp_inforec_get_type(rec)) { case TRP_INFOREC_TYPE_ROUTE: - if ((trp_inforec_get_comm(rec)==NULL) - || (trp_inforec_get_realm(rec)==NULL) - || (trp_inforec_get_trust_router(rec)==NULL) + if ((trp_inforec_get_trust_router(rec)==NULL) || (trp_inforec_get_next_hop(rec)==NULL)) { tr_debug("trps_validate_inforec: missing record info."); return TRP_ERROR; @@ -441,6 +520,10 @@ static TRP_RC trps_validate_inforec(TRPS_INSTANCE *trps, TRP_INFOREC *rec) } break; + case TRP_INFOREC_TYPE_COMMUNITY: + /* TODO: validate community updates */ + break; + default: tr_notice("trps_validate_inforec: unsupported record type."); return TRP_UNSUPPORTED; @@ -463,7 +546,7 @@ static unsigned int trps_advertised_metric(TRPS_INSTANCE *trps, TR_NAME *comm, T return trp_route_get_metric(entry) + trps_cost(trps, peer); } -static int trps_check_feasibility(TRPS_INSTANCE *trps, TRP_INFOREC *rec) +static int trps_check_feasibility(TRPS_INSTANCE *trps, TR_NAME *realm, TR_NAME *comm, TRP_INFOREC *rec) { unsigned int rec_metric=trp_inforec_get_metric(rec); unsigned int new_metric=0; @@ -479,9 +562,7 @@ static int trps_check_feasibility(TRPS_INSTANCE *trps, TRP_INFOREC *rec) return 1; /* updates from our current next hop are always feasible*/ - next_hop=trps_get_next_hop(trps, - trp_inforec_get_comm(rec), - trp_inforec_get_realm(rec));; + next_hop=trps_get_next_hop(trps, comm, realm); if ((next_hop!=NULL) && (0==tr_name_cmp(next_hop,trp_inforec_get_next_hop(rec)))) { return 1; @@ -490,10 +571,7 @@ static int trps_check_feasibility(TRPS_INSTANCE *trps, TRP_INFOREC *rec) /* compare the existing metric we advertise to what we would advertise * if we accept this update */ - current_metric=trps_advertised_metric(trps, - trp_inforec_get_comm(rec), - trp_inforec_get_realm(rec), - trp_inforec_get_next_hop(rec)); + current_metric=trps_advertised_metric(trps, comm, realm, trp_inforec_get_next_hop(rec)); new_metric=rec_metric + trps_cost(trps, trp_inforec_get_next_hop(rec)); if (new_metric <= current_metric) return 1; @@ -505,11 +583,12 @@ static int trps_check_feasibility(TRPS_INSTANCE *trps, TRP_INFOREC *rec) static struct timespec *trps_compute_expiry(TRPS_INSTANCE *trps, unsigned int interval, struct timespec *ts) { const unsigned int small_factor=3; /* how many intervals we wait before expiring */ - if (0!=clock_gettime(CLOCK_REALTIME, ts)) { + if (0!=clock_gettime(TRP_CLOCK, ts)) { tr_err("trps_compute_expiry: could not read realtime clock."); ts->tv_sec=0; ts->tv_nsec=0; } + tr_debug("trps_compute_expiry: tv_sec=%u, interval=%u, small_factor*interval=%u", ts->tv_sec, interval, small_factor*interval); ts->tv_sec += small_factor*interval; return ts; } @@ -519,8 +598,8 @@ static TRP_RC trps_accept_update(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC TRP_ROUTE *entry=NULL; entry=trp_rtable_get_entry(trps->rtable, - trp_inforec_get_comm(rec), - trp_inforec_get_realm(rec), + trp_upd_get_comm(upd), + trp_upd_get_realm(upd), trp_inforec_get_next_hop(rec)); if (entry==NULL) { entry=trp_route_new(NULL); @@ -529,11 +608,12 @@ static TRP_RC trps_accept_update(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC return TRP_NOMEM; } - trp_route_set_comm(entry, trp_inforec_dup_comm(rec)); - trp_route_set_realm(entry, trp_inforec_dup_realm(rec)); + trp_route_set_comm(entry, trp_upd_dup_comm(upd)); + trp_route_set_realm(entry, trp_upd_dup_realm(upd)); trp_route_set_peer(entry, trp_upd_dup_peer(upd)); trp_route_set_trust_router(entry, trp_inforec_dup_trust_router(rec)); trp_route_set_next_hop(entry, trp_inforec_dup_next_hop(rec)); + /* TODO: pass next hop port (now defaults to TID_PORT) --jlr */ if ((trp_route_get_comm(entry)==NULL) ||(trp_route_get_realm(entry)==NULL) ||(trp_route_get_peer(entry)==NULL) @@ -572,11 +652,342 @@ static TRP_RC trps_accept_update(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC return TRP_SUCCESS; } -static TRP_RC trps_handle_update(TRPS_INSTANCE *trps, TRP_UPD *upd) + +static TRP_RC trps_handle_inforec_route(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC *rec) { + TRP_ROUTE *route=NULL; unsigned int feas=0; + + /* determine feasibility */ + feas=trps_check_feasibility(trps, trp_upd_get_realm(upd), trp_upd_get_comm(upd), rec); + tr_debug("trps_handle_update: record feasibility=%d", feas); + + /* do we have an existing route? */ + route=trps_get_route(trps, + trp_upd_get_comm(upd), + trp_upd_get_realm(upd), + trp_upd_get_peer(upd)); + if (route!=NULL) { + /* there was a route table entry already */ + tr_debug("trps_handle_updates: route entry already exists."); + if (feas) { + /* Update is feasible. Accept it. */ + trps_accept_update(trps, upd, rec); + } else { + /* Update is infeasible. Ignore it unless the trust router has changed. */ + if (0!=tr_name_cmp(trp_route_get_trust_router(route), + trp_inforec_get_trust_router(rec))) { + /* the trust router associated with the route has changed, treat update as a retraction */ + trps_retract_route(trps, route); + } + } + } else { + /* No existing route table entry. Ignore it unless it is feasible and not a retraction. */ + tr_debug("trps_handle_update: no route entry exists yet."); + if (feas && trp_metric_is_finite(trp_inforec_get_metric(rec))) + trps_accept_update(trps, upd, rec); + } + + return TRP_SUCCESS; +} + +static int trps_name_in_provenance(TR_NAME *name, json_t *prov) +{ + size_t ii=0; + TR_NAME *this_name=NULL; + const char *s=NULL; + + if (prov==NULL) + return 0; /* no provenance list, so it has no names in it */ + + /* now check to see if name is in the provenance */ + for (ii=0; iiexpiration_interval=trp_inforec_get_exp_interval(rec); + talloc_steal(mem_ctx, comm); + +cleanup: + talloc_free(tmp_ctx); + return comm; +} + +static TR_RP_REALM *trps_create_new_rp_realm(TALLOC_CTX *mem_ctx, TR_NAME *realm_id, TRP_INFOREC *rec) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TR_RP_REALM *rp=tr_rp_realm_new(tmp_ctx); + + if (rp==NULL) { + tr_debug("trps_create_new_rp_realm: unable to allocate new realm."); + goto cleanup; + } + /* fill in the realm */ + tr_rp_realm_set_id(rp, tr_dup_name(realm_id)); + if (tr_rp_realm_get_id(rp)==NULL) { + tr_debug("trps_create_new_rp_realm: unable to allocate realm name."); + rp=NULL; + goto cleanup; + } + talloc_steal(mem_ctx, rp); + +cleanup: + talloc_free(tmp_ctx); + return rp; +} + +static TR_IDP_REALM *trps_create_new_idp_realm(TALLOC_CTX *mem_ctx, TR_NAME *realm_id, TRP_INFOREC *rec) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TR_IDP_REALM *idp=tr_idp_realm_new(tmp_ctx); + + if (idp==NULL) { + tr_debug("trps_create_new_idp_realm: unable to allocate new realm."); + goto cleanup; + } + /* fill in the realm */ + tr_idp_realm_set_id(idp, tr_dup_name(realm_id)); + if (tr_idp_realm_get_id(idp)==NULL) { + tr_debug("trps_create_new_idp_realm: unable to allocate realm name."); + idp=NULL; + goto cleanup; + } + if (trp_inforec_get_apcs(rec)!=NULL) { + tr_idp_realm_set_apcs(idp, tr_apc_dup(tmp_ctx, trp_inforec_get_apcs(rec))); + if (tr_idp_realm_get_apcs(idp)==NULL) { + tr_debug("trps_create_new_idp_realm: unable to allocate APC list."); + idp=NULL; + goto cleanup; + } + } + idp->origin=TR_REALM_DISCOVERED; + + talloc_steal(mem_ctx, idp); + +cleanup: + talloc_free(tmp_ctx); + return idp; +} + +static TRP_RC trps_handle_inforec_comm(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC *rec) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TR_NAME *comm_id=trp_upd_get_comm(upd); + TR_NAME *realm_id=trp_upd_get_realm(upd); + TR_NAME *origin_id=NULL; + TR_NAME *our_peer_label=NULL; + TR_COMM *comm=NULL; + TR_RP_REALM *rp_realm=NULL; + TR_IDP_REALM *idp_realm=NULL; + struct timespec expiry={0,0}; + TRP_RC rc=TRP_ERROR; + + if ((comm_id==NULL) || (realm_id==NULL)) + goto cleanup; + + origin_id=trp_inforec_dup_origin(rec); + if (origin_id==NULL) + goto cleanup; + + /* see whether we want to add this */ + our_peer_label=trps_dup_label(trps); + if (our_peer_label==NULL) { + tr_debug("trps_handle_inforec_comm: unable to allocate peer label."); + goto cleanup; + } + + if (trps_name_in_provenance(our_peer_label, trp_inforec_get_provenance(rec))) + tr_debug("trps_handle_inforec_comm: rejecting community inforec to avoid provenance loop."); + else { + /* no loop occurring, accept the update */ + comm=tr_comm_table_find_comm(trps->ctable, comm_id); + if (comm==NULL) { + tr_debug("trps_handle_inforec_comm: unknown community %.*s in inforec, creating it.", + comm_id->len, comm_id->buf); + comm=trps_create_new_comm(tmp_ctx, comm_id, rec); + if (comm==NULL) { + tr_debug("trps_handle_inforec_comm: unable to create new community."); + goto cleanup; + } + tr_comm_table_add_comm(trps->ctable, comm); + } + /* TODO: see if other comm data match the new inforec and update or complain */ + + trps_compute_expiry(trps, trp_inforec_get_interval(rec), &expiry); + if ((expiry.tv_sec==0)&&(expiry.tv_nsec==0)) + goto cleanup; + + switch (trp_inforec_get_role(rec)) { + case TR_ROLE_RP: + rp_realm=tr_rp_realm_lookup(trps->ctable->rp_realms, realm_id); + if (rp_realm==NULL) { + tr_debug("trps_handle_inforec_comm: unknown RP realm %.*s in inforec, creating it.", + realm_id->len, realm_id->buf); + rp_realm=trps_create_new_rp_realm(tmp_ctx, realm_id, rec); + if (rp_realm==NULL) { + tr_debug("trps_handle_inforec_comm: unable to create new RP realm."); + /* we may leave an unused community in the table, but it will only last until + * the next table sweep if it does not get any realms before that happens */ + goto cleanup; + } + tr_comm_table_add_rp_realm(trps->ctable, rp_realm); + } + /* TODO: if realm existed, see if data match the new inforec and update or complain */ + tr_comm_add_rp_realm(trps->ctable, comm, rp_realm, trp_inforec_get_interval(rec), trp_inforec_get_provenance(rec), &expiry); + tr_debug("trps_handle_inforec_comm: added RP realm %.*s to comm %.*s (origin %.*s).", + realm_id->len, realm_id->buf, + comm_id->len, comm_id->buf, + origin_id->len, origin_id->buf); + break; + case TR_ROLE_IDP: + idp_realm=tr_idp_realm_lookup(trps->ctable->idp_realms, realm_id); + if (idp_realm==NULL) { + tr_debug("trps_handle_inforec_comm: unknown IDP realm %.*s in inforec, creating it.", + realm_id->len, realm_id->buf); + idp_realm=trps_create_new_idp_realm(tmp_ctx, realm_id, rec); + if (idp_realm==NULL) { + tr_debug("trps_handle_inforec_comm: unable to create new IDP realm."); + /* we may leave an unused community in the table, but it will only last until + * the next table sweep if it does not get any realms before that happens */ + goto cleanup; + } + tr_comm_table_add_idp_realm(trps->ctable, idp_realm); + } + /* TODO: if realm existed, see if data match the new inforec and update or complain */ + tr_comm_add_idp_realm(trps->ctable, comm, idp_realm, trp_inforec_get_interval(rec), trp_inforec_get_provenance(rec), &expiry); + tr_debug("trps_handle_inforec_comm: added IDP realm %.*s to comm %.*s (origin %.*s).", + realm_id->len, realm_id->buf, + comm_id->len, comm_id->buf, + origin_id->len, origin_id->buf); + break; + default: + tr_debug("trps_handle_inforec_comm: unable to add realm."); + goto cleanup; + } + } + + rc=TRP_SUCCESS; + +cleanup: + if (our_peer_label!=NULL) + tr_free_name(our_peer_label); + if (origin_id!=NULL) + tr_free_name(origin_id); + talloc_free(tmp_ctx); + return rc; +} + +/** + * Apply applicable TRP_INBOUND filters to an inforec. Rejects everything if peer has no filters. + * + * @param trps Active TRPS instance + * @param upd TRP_UPD that contains the inforec to filter + * @param rec Inforec to filter + * @return 1 if accepted by the filter, 0 otherwise + */ +static int trps_filter_inbound_inforec(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC *rec) +{ + TRP_PEER *peer=NULL; + TR_NAME *peer_name=NULL; + TR_FILTER_ACTION action=TR_FILTER_ACTION_REJECT; + TR_FILTER_TARGET *target=NULL; + int retval=0; + + /* Look up the peer. For inbound messages, the peer is identified by its GSS name */ + peer_name=trp_upd_get_peer(upd); + peer=trps_get_peer_by_gssname(trps, peer_name); + if (peer==NULL) { + tr_err("trps_filter_inbound_inforec: received inforec from unknown peer (%.*s), rejecting.", + peer_name->len, + peer_name->buf); + return 0; + } + + /* tr_filter_apply() and tr_filter_set_get() handle null filter sets/filters by rejecting */ + target= tr_filter_target_trp_inforec(NULL, upd, rec); + if (target==NULL) { + /* TODO: signal that filtering failed. Until then, just filter everything and give an error message. */ + tr_crit("trps_filter_inbound_inforec: Unable to allocate filter target, cannot apply filter!"); + } + if ((target==NULL) + || (TR_FILTER_NO_MATCH==tr_filter_apply(target, + tr_filter_set_get(peer->filters, TR_FILTER_TYPE_TRP_INBOUND), + NULL, + &action)) + || (action!=TR_FILTER_ACTION_ACCEPT)) { + /* either the filter did not match or it matched a reject rule or allocating the target failed */ + retval=0; + } else + retval=1; + if (target!=NULL) + tr_filter_target_free(target); + + /* filter matched an accept rule */ + return retval; +} + + +static TRP_RC trps_handle_update(TRPS_INSTANCE *trps, TRP_UPD *upd) +{ TRP_INFOREC *rec=NULL; - TRP_ROUTE *route=NULL; if (trps_validate_update(trps, upd) != TRP_SUCCESS) { tr_notice("trps_handle_update: received invalid TRP update."); @@ -586,40 +997,32 @@ static TRP_RC trps_handle_update(TRPS_INSTANCE *trps, TRP_UPD *upd) for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) { /* validate/sanity check the record update */ if (trps_validate_inforec(trps, rec) != TRP_SUCCESS) { - tr_notice("trps_handle_update: invalid record in TRP update, discarding entire update."); + tr_notice("trps_handle_update: invalid inforec in TRP update, discarding entire update."); return TRP_ERROR; } } for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) { - /* determine feasibility */ - feas=trps_check_feasibility(trps, rec); - tr_debug("trps_handle_update: record feasibility=%d", feas); - - /* do we have an existing route? */ - route=trps_get_route(trps, - trp_inforec_get_comm(rec), - trp_inforec_get_realm(rec), - trp_upd_get_peer(upd)); - if (route!=NULL) { - /* there was a route table entry already */ - tr_debug("trps_handle_updates: route entry already exists."); - if (feas) { - /* Update is feasible. Accept it. */ - trps_accept_update(trps, upd, rec); - } else { - /* Update is infeasible. Ignore it unless the trust router has changed. */ - if (0!=tr_name_cmp(trp_route_get_trust_router(route), - trp_inforec_get_trust_router(rec))) { - /* the trust router associated with the route has changed, treat update as a retraction */ - trps_retract_route(trps, route); - } - } - } else { - /* No existing route table entry. Ignore it unless it is feasible and not a retraction. */ - tr_debug("trps_handle_update: no route entry exists yet."); - if (feas && trp_metric_is_finite(trp_inforec_get_metric(rec))) - trps_accept_update(trps, upd, rec); + if (!trps_filter_inbound_inforec(trps, upd, rec)) { + tr_debug("trps_handle_update: inforec rejected by filter."); + continue; /* just go on to the next record */ + } + + switch (trp_inforec_get_type(rec)) { + case TRP_INFOREC_TYPE_ROUTE: + tr_debug("trps_handle_update: handling route inforec."); + if (TRP_SUCCESS!=trps_handle_inforec_route(trps, upd, rec)) + tr_notice("trps_handle_update: error handling route inforec."); + break; + case TRP_INFOREC_TYPE_COMMUNITY: + tr_debug("trps_handle_update: handling community inforec."); + if (TRP_SUCCESS!=trps_handle_inforec_comm(trps, upd, rec)) + tr_notice("trps_handle_update: error handling community inforec."); + + break; + default: + tr_notice("trps_handle_update: unsupported inforec in TRP update."); + break; } } return TRP_SUCCESS; @@ -727,9 +1130,7 @@ TRP_RC trps_update_active_routes(TRPS_INSTANCE *trps) /* true if curtime >= expiry */ static int trps_expired(struct timespec *expiry, struct timespec *curtime) { - return ((curtime->tv_sec > expiry->tv_sec) - || ((curtime->tv_sec == expiry->tv_sec) - &&(curtime->tv_nsec > expiry->tv_nsec))); + return (tr_cmp_timespec(curtime, expiry) >= 0); } /* Sweep for expired routes. For each expired route, if its metric is infinite, the route is flushed. @@ -742,14 +1143,14 @@ TRP_RC trps_sweep_routes(TRPS_INSTANCE *trps) size_t ii=0; /* use a single time for the entire sweep */ - if (0!=clock_gettime(CLOCK_REALTIME, &sweep_time)) { + if (0!=clock_gettime(TRP_CLOCK, &sweep_time)) { tr_err("trps_sweep_routes: could not read realtime clock."); sweep_time.tv_sec=0; sweep_time.tv_nsec=0; return TRP_ERROR; } - entry=trp_rtable_get_entries(trps->rtable, &n_entry); /* must talloc_free *entry */ + entry= trp_rtable_get_entries(NULL, trps->rtable, &n_entry); /* must talloc_free *entry */ /* loop over the entries */ for (ii=0; iictable); + memb!=NULL; + memb=tr_comm_memb_iter_all_next(iter)) { + if (tr_comm_memb_get_origin(memb)==NULL) + continue; /* do not expire local entries */ + + if (tr_comm_memb_is_expired(memb, &sweep_time)) { + if (tr_comm_memb_get_times_expired(memb)>0) { + /* Already expired once; flush. */ + tr_debug("trps_sweep_ctable: flushing expired community membership (%.*s in %.*s, origin %.*s, expired %s).", + tr_comm_memb_get_realm_id(memb)->len, tr_comm_memb_get_realm_id(memb)->buf, + tr_comm_get_id(tr_comm_memb_get_comm(memb))->len, tr_comm_get_id(tr_comm_memb_get_comm(memb))->buf, + tr_comm_memb_get_origin(memb)->len, tr_comm_memb_get_origin(memb)->buf, + timespec_to_str(tr_comm_memb_get_expiry(memb))); + tr_comm_table_remove_memb(trps->ctable, memb); + tr_comm_memb_free(memb); + } else { + /* This is the first expiration. Note this and reset the expiry time. */ + tr_comm_memb_expire(memb); + trps_compute_expiry(trps, tr_comm_memb_get_interval(memb), tr_comm_memb_get_expiry(memb)); + tr_debug("trps_sweep_ctable: community membership expired at %s, resetting expiry to %s (%.*s in %.*s, origin %.*s).", + timespec_to_str(&sweep_time), + timespec_to_str(tr_comm_memb_get_expiry(memb)), + tr_comm_memb_get_realm_id(memb)->len, tr_comm_memb_get_realm_id(memb)->buf, + tr_comm_get_id(tr_comm_memb_get_comm(memb))->len, tr_comm_get_id(tr_comm_memb_get_comm(memb))->buf, + tr_comm_memb_get_origin(memb)->len, tr_comm_memb_get_origin(memb)->buf); + } + } + } + + /* get rid of any unreferenced realms, etc */ + tr_comm_table_sweep(trps->ctable); + +cleanup: + talloc_free(tmp_ctx); + return rc; +} + +/* add metrics */ +static unsigned int trps_metric_add(unsigned int m1, unsigned int m2) +{ + if (trp_metric_is_invalid(m1) || trp_metric_is_invalid(m2)) + return TRP_METRIC_INVALID; + + if (trp_metric_is_infinite(m1) || trp_metric_is_infinite(m2)) + return TRP_METRIC_INFINITY; + + if (trp_metric_is_finite(m1+m2)) + return m1+m2; + else + return TRP_METRIC_INFINITY; +} + +/* convert an rentry into a new trp update info record */ +static TRP_INFOREC *trps_route_to_inforec(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TRP_ROUTE *route) +{ + TRP_INFOREC *rec=trp_inforec_new(mem_ctx, TRP_INFOREC_TYPE_ROUTE); + unsigned int linkcost=0; + + if (rec!=NULL) { + if (trp_route_is_local(route)) + linkcost=0; + else { + linkcost=trp_peer_get_linkcost(trps_get_peer_by_gssname(trps, + trp_route_get_peer(route))); + } + + /* Note that we leave the next hop empty since the recipient fills that in. + * This is where we add the link cost (currently always 1) to the next peer. */ + if ((trp_inforec_set_trust_router(rec, trp_route_dup_trust_router(route)) != TRP_SUCCESS) + ||(trp_inforec_set_metric(rec, + trps_metric_add(trp_route_get_metric(route), + linkcost)) != TRP_SUCCESS) + ||(trp_inforec_set_interval(rec, trps_get_update_interval(trps)) != TRP_SUCCESS)) { + tr_err("trps_route_to_inforec: error creating route update."); + talloc_free(rec); + rec=NULL; + } + } + return rec; +} + +static TRP_UPD *trps_route_to_upd(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TRP_ROUTE *route) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TRP_UPD *upd=trp_upd_new(tmp_ctx); + TRP_INFOREC *rec=NULL; + + if (upd==NULL) { + tr_err("trps_route_to_upd: could not create update message."); + goto cleanup; + } + trp_upd_set_realm(upd, trp_route_dup_realm(route)); + if (trp_upd_get_realm(upd)==NULL) { + tr_err("trps_route_to_upd: could not copy realm."); + upd=NULL; /* it's still in tmp_ctx, so it will be freed */ + goto cleanup; + } + trp_upd_set_comm(upd, trp_route_dup_comm(route)); + if (trp_upd_get_comm(upd)==NULL) { + tr_err("trps_route_to_upd: could not copy comm."); + upd=NULL; /* it's still in tmp_ctx, so it will be freed */ + goto cleanup; + } + rec=trps_route_to_inforec(tmp_ctx, trps, route); + if (rec==NULL) { + tr_err("trps_route_to_upd: could not create route info record for realm %.*s in comm %.*s.", + trp_route_get_realm(route)->len, trp_route_get_realm(route)->buf, + trp_route_get_comm(route)->len, trp_route_get_comm(route)->buf); + upd=NULL; /* it's till in tmp_ctx, so it will be freed */ + goto cleanup; + } + trp_upd_add_inforec(upd, rec); + + /* sucess */ + talloc_steal(mem_ctx, upd); + +cleanup: + talloc_free(tmp_ctx); + return upd; +} + /* select the correct route to comm/realm to be announced to peer */ static TRP_ROUTE *trps_select_realm_update(TRPS_INSTANCE *trps, TR_NAME *comm, TR_NAME *realm, TR_NAME *peer_gssname) { @@ -800,14 +1348,12 @@ static TRP_ROUTE *trps_select_realm_update(TRPS_INSTANCE *trps, TR_NAME *comm, T return route; } -/* returns an array of pointers to updates (*not* an array of updates). Returns number of entries - * via n_update parameter. (The allocated space will generally be larger than required, see note in - * the code.) If triggered is set, sends only triggered updates. */ -static TRP_ROUTE **trps_select_updates_for_peer(TALLOC_CTX *memctx, +/* Add TRP_UPD msgs to the updates GPtrArray. Caller needs to arrange for these to be freed. */ +static TRP_RC trps_select_route_updates_for_peer(TALLOC_CTX *mem_ctx, + GPtrArray *updates, TRPS_INSTANCE *trps, TR_NAME *peer_gssname, - int triggered, - size_t *n_update) + int triggered) { size_t n_comm=0; TR_NAME **comm=trp_rtable_get_comms(trps->rtable, &n_comm); @@ -815,195 +1361,407 @@ static TRP_ROUTE **trps_select_updates_for_peer(TALLOC_CTX *memctx, size_t n_realm=0; size_t ii=0, jj=0; TRP_ROUTE *best=NULL; - TRP_ROUTE **result=NULL; - size_t n_used=0; - - /* Need to allocate space for the results. For simplicity, we just allocate a block - * with space for every route table entry to be returned. This is guaranteed to be large - * enough. If the routing table gets very large, this may be wasteful, but that seems - * unlikely to be significant in the near future. */ - result=talloc_array(memctx, TRP_ROUTE *, trp_rtable_size(trps->rtable)); - if (result==NULL) { - talloc_free(comm); - *n_update=0; - return NULL; - } - + TRP_UPD *upd=NULL; + + if (updates==NULL) + return TRP_BADARG; + for (ii=0; iirtable, comm[ii], &n_realm); for (jj=0; jjrole) { + case TR_ROLE_IDP: + memb=tr_comm_table_find_idp_memb(trps->ctable, + tr_realm_get_id(realm), + tr_comm_get_id(comm)); + break; + case TR_ROLE_RP: + memb=tr_comm_table_find_rp_memb(trps->ctable, + tr_realm_get_id(realm), + tr_comm_get_id(comm)); + break; + default: + break; + } + if (memb!=NULL) { + for (memb=tr_comm_memb_iter_first(iter, memb); + memb!=NULL; + memb=tr_comm_memb_iter_next(iter)) { + rec=trps_memb_to_inforec(tmp_ctx, trps, memb); + if (rec==NULL) { + tr_err("trps_comm_update: unable to allocate inforec."); + upd=NULL; + goto cleanup; + } + trp_upd_add_inforec(upd, rec); + } + } + + if (trp_upd_get_inforec(upd)==NULL) + upd=NULL; /* no inforecs, no reason to send the update */ else - return TRP_METRIC_INFINITY; + talloc_steal(mem_ctx, upd); /* success! */ + +cleanup: + talloc_free(tmp_ctx); + return upd; } -/* convert an rentry into a new trp update info record */ -static TRP_INFOREC *trps_route_to_inforec(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *trps, TRP_ROUTE *route) +/* Find all community updates to send to a peer and add these as TR_UPD records + * to the updates GPtrArray. */ +static TRP_RC trps_select_comm_updates_for_peer(TALLOC_CTX *mem_ctx, + GPtrArray *updates, + TRPS_INSTANCE *trps, + TR_NAME *peer_gssname, + int triggered) { - TRP_INFOREC *rec=trp_inforec_new(mem_ctx, TRP_INFOREC_TYPE_ROUTE); - unsigned int linkcost=0; + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TR_COMM_ITER *comm_iter=NULL; + TR_COMM *comm=NULL; + TR_COMM_ITER *realm_iter=NULL; + TR_REALM *realm=NULL; + TRP_UPD *upd=NULL; + TRP_RC rc=TRP_ERROR; - if (rec!=NULL) { - if (trp_route_is_local(route)) - linkcost=0; - else { - linkcost=trp_peer_get_linkcost(trps_get_peer_by_gssname(trps, - trp_route_get_peer(route))); + /* currently do not send any communities on triggered updates */ + if (triggered) { + rc=TRP_SUCCESS; + goto cleanup; + } + + comm_iter=tr_comm_iter_new(tmp_ctx); + realm_iter=tr_comm_iter_new(tmp_ctx); + if ((comm_iter==NULL) || (realm_iter==NULL)) { + tr_err("trps_select_comm_updates_for_peer: unable to allocate iterator."); + rc=TRP_NOMEM; + goto cleanup; + } + + /* do every community */ + for (comm=tr_comm_table_iter_first(comm_iter, trps->ctable); + comm!=NULL; + comm=tr_comm_table_iter_next(comm_iter)) { + /* do every realm in this community */ + tr_debug("trps_select_comm_updates_for_peer: looking through community %.*s", + tr_comm_get_id(comm)->len, + tr_comm_get_id(comm)->buf); + for (realm=tr_realm_iter_first(realm_iter, trps->ctable, tr_comm_get_id(comm)); + realm!=NULL; + realm=tr_realm_iter_next(realm_iter)) { + /* get the update for this comm/realm */ + tr_debug("trps_select_comm_updates_for_peer: adding realm %.*s", + tr_realm_get_id(realm)->len, + tr_realm_get_id(realm)->buf); + upd=trps_comm_update(mem_ctx, trps, peer_gssname, comm, realm); + if (upd!=NULL) + g_ptr_array_add(updates, upd); } + } - /* Note that we leave the next hop empty since the recipient fills that in. - * This is where we add the link cost (currently always 1) to the next peer. */ - if ((trp_inforec_set_comm(rec, trp_route_dup_comm(route)) != TRP_SUCCESS) - ||(trp_inforec_set_realm(rec, trp_route_dup_realm(route)) != TRP_SUCCESS) - ||(trp_inforec_set_trust_router(rec, trp_route_dup_trust_router(route)) != TRP_SUCCESS) - ||(trp_inforec_set_metric(rec, - trps_metric_add(trp_route_get_metric(route), - linkcost)) != TRP_SUCCESS) - ||(trp_inforec_set_interval(rec, trps_get_update_interval(trps)) != TRP_SUCCESS)) { - tr_err("trps_route_to_inforec: error creating route update."); - talloc_free(rec); - rec=NULL; +cleanup: + talloc_free(tmp_ctx); + return rc; +} + +/** + * Filter the inforecs in a single update + * + * @param filt The filter to apply + * @param upd The update to filter + */ +static void trps_filter_one_outbound_update(TR_FILTER *filt, TRP_UPD *upd) +{ + TRP_INFOREC *this=NULL, *next=NULL; + TR_FILTER_ACTION action=TR_FILTER_ACTION_REJECT; + TR_FILTER_TARGET *target=NULL; + + for(this=trp_upd_get_inforec(upd); this!=NULL; this=next) { + next=this->next; + target= tr_filter_target_trp_inforec(NULL, upd, this); + if (target==NULL) { + /* TODO: signal that filtering failed. Until then, just filter everything and give an error message. */ + tr_crit("trps_filter_one_outbound_update: Unable to allocate filter target, cannot apply filter!"); + } + if ((target==NULL) + || (TR_FILTER_NO_MATCH==tr_filter_apply(target, filt, NULL, &action)) + || (action!=TR_FILTER_ACTION_ACCEPT)) { + /* Either no filter matched or one matched and rejected this record. + * Also filter out record if we were unable to allocate a target. */ + trp_upd_remove_inforec(upd, this); /* "this" is now invalid */ } + if (target!=NULL) + tr_filter_target_free(target); + } +} + +/** + * May shuffle the update list. + * + * @param filters The filter set for the relevant TRP peer + * @param updates GPtrArray of updates to filter + */ +static void trps_filter_outbound_updates(TR_FILTER_SET *filters, GPtrArray *updates) +{ + TRP_UPD *upd=NULL; + guint ii=0; + + /* Walk backward through the array so we can remove elements. Careful about loop + * termination - remember that ii is unsigned. */ + for (ii=updates->len; ii>0; ii--) { + upd=g_ptr_array_index(updates, ii-1); + trps_filter_one_outbound_update(tr_filter_set_get(filters, TR_FILTER_TYPE_TRP_OUTBOUND), upd); + /* see if we removed all the records from this update */ + if (trp_upd_num_inforecs(upd)==0) + g_ptr_array_remove_index_fast(updates, ii-1); /* does not preserve order at index ii or higher */ } - return rec; } -/* all routes to a single peer, unless comm/realm are specified (both or neither must be NULL) */ +/* helper for trps_update_one_peer. Frees the TRP_UPD pointed to by a GPtrArray element */ +static void trps_trp_upd_destroy(gpointer data) +{ + trp_upd_free((TRP_UPD *)data); +} + +/* all routes/communities to a single peer, unless comm/realm are specified (both or neither must be NULL) */ static TRP_RC trps_update_one_peer(TRPS_INSTANCE *trps, - TR_NAME *peer_gssname, + TRP_PEER *peer, TRP_UPDATE_TYPE update_type, - TR_NAME *comm, - TR_NAME *realm) + TR_NAME *realm, + TR_NAME *comm) { TALLOC_CTX *tmp_ctx=talloc_new(NULL); TR_MSG msg; /* not a pointer! */ TRP_UPD *upd=NULL; - TRP_ROUTE **update_list=NULL; - TRP_INFOREC *rec=NULL; - size_t n_updates=0, ii=0; + TRP_ROUTE *route=NULL; + size_t ii=0; char *encoded=NULL; TRP_RC rc=TRP_ERROR; - TRP_PEER *peer=trps_get_peer_by_gssname(trps, peer_gssname); + TR_NAME *peer_label=trp_peer_get_label(peer); + GPtrArray *updates=g_ptr_array_new_with_free_func(trps_trp_upd_destroy); + + if (updates==NULL) { + tr_err("trps_update_one_peer: unable to allocate updates array."); + rc=TRP_NOMEM; + goto cleanup; + } switch (update_type) { case TRP_UPDATE_TRIGGERED: - tr_debug("trps_update_one_peer: preparing triggered route update for %.*s", - peer_gssname->len, peer_gssname->buf); + tr_debug("trps_update_one_peer: preparing triggered update for %.*s", + peer_label->len, peer_label->buf); break; case TRP_UPDATE_SCHEDULED: - tr_debug("trps_update_one_peer: preparing scheduled route update for %.*s", - peer_gssname->len, peer_gssname->buf); + tr_debug("trps_update_one_peer: preparing scheduled update for %.*s", + peer_label->len, peer_label->buf); break; case TRP_UPDATE_REQUESTED: - tr_debug("trps_update_one_peer: preparing requested route update for %.*s", - peer_gssname->len, peer_gssname->buf); + tr_debug("trps_update_one_peer: preparing requested update for %.*s", + peer_label->len, peer_label->buf); + break; + default: + tr_err("trps_update_one_peer: invalid update type requested."); + rc=TRP_BADARG; + goto cleanup; } - /* do not fill in peer, recipient does that */ + /* First, gather route updates. */ + tr_debug("trps_update_one_peer: selecting route updates for %.*s.", peer_label->len, peer_label->buf); if ((comm==NULL) && (realm==NULL)) { /* do all realms */ - update_list=trps_select_updates_for_peer(tmp_ctx, - trps, - peer_gssname, - update_type==TRP_UPDATE_TRIGGERED, - &n_updates); + rc=trps_select_route_updates_for_peer(tmp_ctx, + updates, + trps, + peer_label, + update_type==TRP_UPDATE_TRIGGERED); } else if ((comm!=NULL) && (realm!=NULL)) { /* a single community/realm was requested */ - update_list=talloc(tmp_ctx, TRP_ROUTE *); - if (update_list==NULL) { - tr_err("trps_update_one_peer: could not allocate update_list."); - rc=TRP_NOMEM; - goto cleanup; - } - *update_list=trps_select_realm_update(trps, comm, realm, peer_gssname); - if (*update_list==NULL) { + route=trps_select_realm_update(trps, comm, realm, peer_label); + if (route==NULL) { /* we have no actual update to send back, MUST send a retraction */ tr_debug("trps_update_one_peer: community/realm without route requested, sending mandatory retraction."); - *update_list=trp_route_new(update_list); - trp_route_set_comm(*update_list, tr_dup_name(comm)); - trp_route_set_realm(*update_list, tr_dup_name(realm)); - trp_route_set_peer(*update_list, tr_new_name("")); - trp_route_set_metric(*update_list, TRP_METRIC_INFINITY); - trp_route_set_trust_router(*update_list, tr_new_name("")); - trp_route_set_next_hop(*update_list, tr_new_name("")); + route=trp_route_new(tmp_ctx); + trp_route_set_comm(route, tr_dup_name(comm)); + trp_route_set_realm(route, tr_dup_name(realm)); + trp_route_set_peer(route, tr_new_name("")); + trp_route_set_metric(route, TRP_METRIC_INFINITY); + trp_route_set_trust_router(route, tr_new_name("")); + trp_route_set_next_hop(route, tr_new_name("")); + } + upd=trps_route_to_upd(tmp_ctx, trps, route); + if (upd==NULL) { + tr_err("trps_update_one_peer: unable to allocate route update."); + rc=TRP_NOMEM; + goto cleanup; } - n_updates=1; + g_ptr_array_add(updates, upd); } else { - tr_err("trps_update_one_peer: error: only comm or realm was specified."); + tr_err("trps_update_one_peer: error: only comm or realm was specified. Need both or neither."); rc=TRP_ERROR; goto cleanup; } - if ((n_updates>0) && (update_list!=NULL)) { - tr_debug("trps_update_one_peer: sending %u update records.", (unsigned int)n_updates); - upd=trp_upd_new(tmp_ctx); - for (ii=0; iilen, peer_label->buf); + rc=trps_select_comm_updates_for_peer(tmp_ctx, updates, trps, peer_label, update_type==TRP_UPDATE_TRIGGERED); - tr_debug("trps_update_one_peer: adding message to queue."); - if (trps_send_msg(trps, peer, encoded) != TRP_SUCCESS) - tr_err("trps_update_one_peer: error queueing update."); - else - tr_debug("trps_update_one_peer: update queued successfully."); + /* see if we have anything to send */ + if (updates->len<=0) + tr_debug("trps_update_one_peer: no updates for %.*s", peer_label->len, peer_label->buf); + else { + /* Apply outbound TRP filters for this peer */ + trps_filter_outbound_updates(peer->filters, updates); - tr_msg_free_encoded(encoded); - encoded=NULL; - trp_upd_free(upd); - upd=NULL; + if (updates->len<=0) + tr_debug("trps_update_one_peer: no updates for %.*s after filtering.", peer_label->len, peer_label->buf); + else { + tr_debug("trps_update_one_peer: sending %d update messages.", updates->len); + for (ii=0; iilen; ii++) { + upd = (TRP_UPD *) g_ptr_array_index(updates, ii); + /* now encode the update message */ + tr_msg_set_trp_upd(&msg, upd); + encoded = tr_msg_encode(NULL, &msg); + if (encoded == NULL) { + tr_err("trps_update_one_peer: error encoding update."); + rc = TRP_ERROR; + goto cleanup; + } + + tr_debug("trps_update_one_peer: adding message to queue."); + if (trps_send_msg(trps, peer, encoded) != TRP_SUCCESS) + tr_err("trps_update_one_peer: error queueing update."); + else + tr_debug("trps_update_one_peer: update queued successfully."); + + tr_msg_free_encoded(encoded); + encoded = NULL; + } + } } + rc=TRP_SUCCESS; + cleanup: + if (updates!=NULL) + g_ptr_array_free(updates, TRUE); /* frees any TRP_UPD records */ talloc_free(tmp_ctx); return rc; } -/* all routes to all peers */ +/* all routes/communities to all peers */ TRP_RC trps_update(TRPS_INSTANCE *trps, TRP_UPDATE_TYPE update_type) { TALLOC_CTX *tmp_ctx=talloc_new(NULL); @@ -1011,6 +1769,9 @@ TRP_RC trps_update(TRPS_INSTANCE *trps, TRP_UPDATE_TYPE update_type) TRP_PEER *peer=NULL; TRP_RC rc=TRP_SUCCESS; + if (trps->ptable==NULL) + return TRP_SUCCESS; /* no peers, nothing to do */ + if (iter==NULL) { tr_err("trps_update: failed to allocate peer table iterator."); talloc_free(tmp_ctx); @@ -1018,18 +1779,19 @@ TRP_RC trps_update(TRPS_INSTANCE *trps, TRP_UPDATE_TYPE update_type) } for (peer=trp_ptable_iter_first(iter, trps->ptable); - peer!=NULL && rc==TRP_SUCCESS; + (peer!=NULL) && (rc==TRP_SUCCESS); peer=trp_ptable_iter_next(iter)) { if (!trps_peer_connected(trps, peer)) { - TR_NAME *peer_gssname=trp_peer_get_gssname(peer); + TR_NAME *peer_label=trp_peer_get_label(peer); tr_debug("trps_update: no TRP connection to %.*s, skipping.", - peer_gssname->len, peer_gssname->buf); + peer_label->len, peer_label->buf); continue; } - rc=trps_update_one_peer(trps, trp_peer_get_gssname(peer), update_type, NULL, NULL); + rc=trps_update_one_peer(trps, peer, update_type, NULL, NULL); } - + + tr_debug("trps_update: rc=%u after attempting update.", rc); trp_ptable_iter_free(iter); trp_rtable_clear_triggered(trps->rtable); /* don't re-send triggered updates */ talloc_free(tmp_ctx); @@ -1045,16 +1807,27 @@ TRP_RC trps_add_route(TRPS_INSTANCE *trps, TRP_ROUTE *route) /* steals the peer object */ TRP_RC trps_add_peer(TRPS_INSTANCE *trps, TRP_PEER *peer) { + if (trps->ptable==NULL) { + trps->ptable=trp_ptable_new(trps); + if (trps->ptable==NULL) + return TRP_NOMEM; + } return trp_ptable_add(trps->ptable, peer); } TRP_PEER *trps_get_peer_by_gssname(TRPS_INSTANCE *trps, TR_NAME *gssname) { - return trp_ptable_find_gssname(trps->ptable, gssname); + if (trps->ptable==NULL) + return NULL; + + return trp_ptable_find_gss_name(trps->ptable, gssname); } TRP_PEER *trps_get_peer_by_servicename(TRPS_INSTANCE *trps, TR_NAME *servicename) { + if (trps->ptable==NULL) + return NULL; + return trp_ptable_find_servicename(trps->ptable, servicename); } @@ -1092,7 +1865,11 @@ static TRP_RC trps_handle_request(TRPS_INSTANCE *trps, TRP_REQ *req) tr_debug("trps_handle_request: all routes requested."); /* leave comm/realm NULL */ } - return trps_update_one_peer(trps, trp_req_get_peer(req), TRP_UPDATE_REQUESTED, comm, realm); + return trps_update_one_peer(trps, + trps_get_peer_by_gssname(trps, trp_req_get_peer(req)), + TRP_UPDATE_REQUESTED, + realm, + comm); } @@ -1141,7 +1918,7 @@ TRP_RC trps_wildcard_route_req(TRPS_INSTANCE *trps, TR_NAME *peer_servicename) } tr_msg_set_trp_req(&msg, req); - encoded=tr_msg_encode(&msg); + encoded= tr_msg_encode(NULL, &msg); if (encoded==NULL) { tr_err("trps_wildcard_route_req: error encoding wildcard TRP request."); rc=TRP_ERROR;