Bug fixes to make TID forwarding work between trust routers.
[trust_router.git] / trp / trps.c
index 0b1f25d..1c6cde5 100644 (file)
@@ -8,6 +8,7 @@
 #include <tr_rp.h>
 #include <trust_router/tr_name.h>
 #include <trp_internal.h>
+#include <tr_gss.h>
 #include <trp_ptable.h>
 #include <trp_rtable.h>
 #include <tr_debug.h>
@@ -32,6 +33,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 +42,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 */
@@ -128,15 +123,22 @@ void trps_set_sweep_interval(TRPS_INSTANCE *trps, unsigned int interval)
   trps->sweep_interval.tv_usec=0;
 }
 
+void trps_set_ptable(TRPS_INSTANCE *trps, TRP_PTABLE *ptable)
+{
+  if (trps->ptable!=NULL)
+    trp_ptable_free(trps->ptable);
+  trps->ptable=ptable;
+}
+
 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;
     }
   }
@@ -242,6 +244,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 +277,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 +291,39 @@ static TRP_RC trps_read_message(TRPS_INSTANCE *trps, TRP_CONNECTION *conn, TR_MS
     return TRP_ERROR;
   }
 
-  tr_debug("trps_read_message(): Request 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);
   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 */
     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;
@@ -514,7 +530,7 @@ static struct timespec *trps_compute_expiry(TRPS_INSTANCE *trps, unsigned int in
   return ts;
 }
 
-static TRP_RC trps_accept_update(TRPS_INSTANCE *trps, TRP_INFOREC *rec)
+static TRP_RC trps_accept_update(TRPS_INSTANCE *trps, TRP_UPD *upd, TRP_INFOREC *rec)
 {
   TRP_ROUTE *entry=NULL;
 
@@ -529,12 +545,13 @@ static TRP_RC trps_accept_update(TRPS_INSTANCE *trps, TRP_INFOREC *rec)
       return TRP_NOMEM;
     }
 
-    trp_route_set_apc(entry, tr_dup_name(trp_inforec_get_comm(rec)));
-    trp_route_set_realm(entry, tr_dup_name(trp_inforec_get_realm(rec)));
-    trp_route_set_peer(entry, tr_dup_name(trp_inforec_get_next_hop(rec)));
-    trp_route_set_trust_router(entry, tr_dup_name(trp_inforec_get_trust_router(rec)));
-    trp_route_set_next_hop(entry, tr_dup_name(trp_inforec_get_next_hop(rec)));
-    if ((trp_route_get_apc(entry)==NULL)
+    trp_route_set_comm(entry, trp_inforec_dup_comm(rec));
+    trp_route_set_realm(entry, trp_inforec_dup_realm(rec));
+    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)
        ||(trp_route_get_trust_router(entry)==NULL)
@@ -600,13 +617,13 @@ static TRP_RC trps_handle_update(TRPS_INSTANCE *trps, TRP_UPD *upd)
     route=trps_get_route(trps,
                          trp_inforec_get_comm(rec),
                          trp_inforec_get_realm(rec),
-                         trp_inforec_get_next_hop(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, rec);
+        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),
@@ -619,7 +636,7 @@ static TRP_RC trps_handle_update(TRPS_INSTANCE *trps, TRP_UPD *upd)
       /* 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, rec);
+        trps_accept_update(trps, upd, rec);
     }
   }
   return TRP_SUCCESS;
@@ -684,23 +701,23 @@ static TRP_ROUTE *trps_find_best_route(TRPS_INSTANCE *trps,
  * to avoid flapping between routers or routes. */
 TRP_RC trps_update_active_routes(TRPS_INSTANCE *trps)
 {
-  size_t n_apc=0, ii=0;
-  TR_NAME **apc=trp_rtable_get_apcs(trps->rtable, &n_apc);
+  size_t n_comm=0, ii=0;
+  TR_NAME **comm=trp_rtable_get_comms(trps->rtable, &n_comm);
   size_t n_realm=0, jj=0;
   TR_NAME **realm=NULL;
   TRP_ROUTE *best_route=NULL, *cur_route=NULL;
   unsigned int best_metric=0, cur_metric=0;
 
-  for (ii=0; ii<n_apc; ii++) {
-    realm=trp_rtable_get_apc_realms(trps->rtable, apc[ii], &n_realm);
+  for (ii=0; ii<n_comm; ii++) {
+    realm=trp_rtable_get_comm_realms(trps->rtable, comm[ii], &n_realm);
     for (jj=0; jj<n_realm; jj++) {
-      best_route=trps_find_best_route(trps, apc[ii], realm[jj], NULL);
+      best_route=trps_find_best_route(trps, comm[ii], realm[jj], NULL);
       if (best_route==NULL)
         best_metric=TRP_METRIC_INFINITY;
       else
         best_metric=trp_route_get_metric(best_route);
 
-      cur_route=trps_get_selected_route(trps, apc[ii], realm[jj]);
+      cur_route=trps_get_selected_route(trps, comm[ii], realm[jj]);
       if (cur_route!=NULL) {
         cur_metric=trp_route_get_metric(cur_route);
         if ((best_metric < cur_metric) && (trp_metric_is_finite(best_metric))) {
@@ -717,9 +734,9 @@ TRP_RC trps_update_active_routes(TRPS_INSTANCE *trps)
       talloc_free(realm);
     realm=NULL; n_realm=0;
   }
-  if (apc!=NULL)
-    talloc_free(apc);
-  apc=NULL; n_apc=0;
+  if (comm!=NULL)
+    talloc_free(comm);
+  comm=NULL; n_comm=0;
 
   return TRP_SUCCESS;
 }
@@ -809,8 +826,8 @@ static TRP_ROUTE **trps_select_updates_for_peer(TALLOC_CTX *memctx,
                                                  int triggered,
                                                  size_t *n_update)
 {
-  size_t n_apc=0;
-  TR_NAME **apc=trp_rtable_get_apcs(trps->rtable, &n_apc);
+  size_t n_comm=0;
+  TR_NAME **comm=trp_rtable_get_comms(trps->rtable, &n_comm);
   TR_NAME **realm=NULL;
   size_t n_realm=0;
   size_t ii=0, jj=0;
@@ -824,15 +841,15 @@ static TRP_ROUTE **trps_select_updates_for_peer(TALLOC_CTX *memctx,
    * unlikely to be significant in the near future. */
   result=talloc_array(memctx, TRP_ROUTE *, trp_rtable_size(trps->rtable));
   if (result==NULL) {
-    talloc_free(apc);
+    talloc_free(comm);
     *n_update=0;
     return NULL;
   }
   
-  for (ii=0; ii<n_apc; ii++) {
-    realm=trp_rtable_get_apc_realms(trps->rtable, apc[ii], &n_realm);
+  for (ii=0; ii<n_comm; ii++) {
+    realm=trp_rtable_get_comm_realms(trps->rtable, comm[ii], &n_realm);
     for (jj=0; jj<n_realm; jj++) {
-      best=trps_select_realm_update(trps, apc[ii], realm[jj], peer_gssname);
+      best=trps_select_realm_update(trps, comm[ii], realm[jj], peer_gssname);
       /* If we found a route, add it to the list. If triggered!=0, then only
        * add triggered routes. */
       if ((best!=NULL) && ((!triggered) || trp_route_is_triggered(best)))
@@ -843,8 +860,8 @@ static TRP_ROUTE **trps_select_updates_for_peer(TALLOC_CTX *memctx,
     realm=NULL;
     n_realm=0;
   }
-  if (apc!=NULL)
-    talloc_free(apc);
+  if (comm!=NULL)
+    talloc_free(comm);
 
   *n_update=n_used;
   return result;
@@ -876,12 +893,12 @@ static TRP_INFOREC *trps_route_to_inforec(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *tr
       linkcost=0;
     else {
       linkcost=trp_peer_get_linkcost(trps_get_peer_by_gssname(trps,
-                                                              trp_route_get_next_hop(route)));
+                                                              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_comm(rec, trp_route_dup_apc(route)) != TRP_SUCCESS)
+    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,
@@ -898,7 +915,7 @@ static TRP_INFOREC *trps_route_to_inforec(TALLOC_CTX *mem_ctx, TRPS_INSTANCE *tr
 
 /* all routes 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)
@@ -911,25 +928,20 @@ static TRP_RC trps_update_one_peer(TRPS_INSTANCE *trps,
   size_t n_updates=0, 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);
 
-  if (!trps_peer_connected(trps, peer)) {
-    tr_debug("trps_update_one_peer: no TRP connection to %.*s, skipping.",
-             peer_gssname->len, peer_gssname->buf);
-    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);
+             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);
+             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);
+             peer_label->len, peer_label->buf);
   }
 
   /* do not fill in peer, recipient does that */
@@ -937,7 +949,7 @@ static TRP_RC trps_update_one_peer(TRPS_INSTANCE *trps,
     /* do all realms */
     update_list=trps_select_updates_for_peer(tmp_ctx,
                                              trps,
-                                             peer_gssname,
+                                             peer_label,
                                              update_type==TRP_UPDATE_TRIGGERED,
                                             &n_updates);
   } else if ((comm!=NULL) && (realm!=NULL)) {
@@ -948,12 +960,12 @@ static TRP_RC trps_update_one_peer(TRPS_INSTANCE *trps,
       rc=TRP_NOMEM;
       goto cleanup;
     }
-    *update_list=trps_select_realm_update(trps, comm, realm, peer_gssname);
+    *update_list=trps_select_realm_update(trps, comm, realm, peer_label);
     if (*update_list==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_apc(*update_list, tr_dup_name(comm));
+      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);
@@ -1016,6 +1028,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);
@@ -1026,7 +1041,13 @@ TRP_RC trps_update(TRPS_INSTANCE *trps, TRP_UPDATE_TYPE update_type)
        peer!=NULL && rc==TRP_SUCCESS;
        peer=trp_ptable_iter_next(iter))
   {
-    rc=trps_update_one_peer(trps, trp_peer_get_gssname(peer), update_type, NULL, NULL);
+    if (!trps_peer_connected(trps, peer)) {
+      TR_NAME *peer_label=trp_peer_get_label(peer);
+      tr_debug("trps_update: no TRP connection to %.*s, skipping.",
+               peer_label->len, peer_label->buf);
+      continue;
+    }
+    rc=trps_update_one_peer(trps, peer, update_type, NULL, NULL);
   }
   
   trp_ptable_iter_free(iter);
@@ -1044,16 +1065,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);
 }
 
@@ -1089,8 +1121,13 @@ static TRP_RC trps_handle_request(TRPS_INSTANCE *trps, TRP_REQ *req)
              comm->len, comm->buf, realm->len, realm->buf);
   } else {
     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,
+                              comm,
+                              realm);
 }