Merge pull request #86 from painless-security/jennifer/aaa_server_port
[trust_router.git] / trp / trp_ptable.c
index 4a276d5..fa0ca84 100644 (file)
+/*
+ * 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 <time.h>
 #include <talloc.h>
 
-#include <trust_router/tr_name.h>
+#include <tr_name_internal.h>
 #include <trp_internal.h>
+#include <tr_gss_names.h>
 #include <trp_ptable.h>
 #include <tr_debug.h>
+#include <trp_peer.h>
 
-TRP_PEER *trp_peer_new(TALLOC_CTX *memctx)
-{
-  TRP_PEER *peer=talloc(memctx, TRP_PEER);
-  if (peer!=NULL) {
-    peer->next=NULL;
-    peer->server=NULL;
-    peer->port=0;
-    peer->linkcost=TRP_METRIC_INFINITY;
-    peer->last_conn_attempt=(struct timespec){0,0};
-  }
-  return peer;
-}
-
-void trp_peer_free(TRP_PEER *peer)
-{
-  talloc_free(peer);
-}
-
-static TRP_PEER *trp_peer_tail(TRP_PEER *peer)
-{
-  while (peer->next!=NULL) {
-    peer=peer->next;
-  }
-  return peer;
-}
-
-char *trp_peer_get_server(TRP_PEER *peer)
-{
-  return peer->server;
-}
-
-/* copies input; on error, peer->gssname will be null */
-void trp_peer_set_server(TRP_PEER *peer, char *server)
-{
-  peer->server=talloc_strdup(peer, server); /* will be null on error */
-}
-
-/* get the peer name based on the server name; caller is responsible for freeing the TR_NAME */
-TR_NAME *trp_peer_get_gssname(TRP_PEER *peer)
-{
-  TR_NAME *gssname=NULL;
-  char *s=NULL;
-
-  if (0<asprintf(&s, "trustrouter@%s", peer->server)) {
-    if (tr_new_name(s)!=NULL) {
-      gssname=tr_new_name(s);
-    }
-    free(s);
-  }
-  return gssname;
-}
-
-unsigned int trp_peer_get_port(TRP_PEER *peer)
-{
-  return peer->port;
-}
-
-void trp_peer_set_port(TRP_PEER *peer, unsigned int port)
-{
-  peer->port=port;
-}
-
-unsigned int trp_peer_get_linkcost(TRP_PEER *peer)
-{
-  if (peer!=NULL)
-    return peer->linkcost;
-  else
-    return 1;
-}
-
-void trp_peer_set_linkcost(TRP_PEER *peer, unsigned int linkcost)
-{
-  if ((linkcost>TRP_METRIC_INFINITY) && (linkcost!=TRP_METRIC_INVALID)) {
-    /* This indicates a programming error, but probably means an already infinite metric
-     * was (incorrectly) incremented. Issue a warning and proceed with an infinite metric. */
-    tr_warning("trp_peer_set_linkcost: link cost > infinity encountered, setting to infinity");
-    linkcost=TRP_METRIC_INFINITY;
-  }
-  peer->linkcost=linkcost;
-}
-
-struct timespec *trp_peer_get_last_conn_attempt(TRP_PEER *peer)
-{
-  return &(peer->last_conn_attempt);
-}
-
-void trp_peer_set_last_conn_attempt(TRP_PEER *peer, struct timespec *time)
-{
-  peer->last_conn_attempt=*time;
-}
 
 TRP_PTABLE *trp_ptable_new(TALLOC_CTX *memctx)
 {
@@ -113,12 +59,12 @@ void trp_ptable_free(TRP_PTABLE *ptbl)
 
 TRP_RC trp_ptable_add(TRP_PTABLE *ptbl, TRP_PEER *newpeer)
 {
-  if (ptbl->head==NULL) {
+  if (ptbl->head==NULL)
     ptbl->head=newpeer;
-  } else {
+  else
     trp_peer_tail(ptbl->head)->next=newpeer;
-    talloc_steal(ptbl, newpeer);
-  }
+
+  talloc_steal(ptbl, newpeer);
   return TRP_SUCCESS;
 }
 
@@ -147,42 +93,20 @@ TRP_RC trp_ptable_remove(TRP_PTABLE *ptbl, TRP_PEER *peer)
   return TRP_ERROR;
 }
 
-TRP_PEER *trp_ptable_find(TRP_PTABLE *ptbl, TR_NAME *gssname)
+TRP_PEER *trp_ptable_find_gss_name(TRP_PTABLE *ptbl, TR_NAME *gssname)
 {
   TRP_PEER *cur=ptbl->head;
-  while ((cur!=NULL) && (0 != tr_name_cmp(trp_peer_get_gssname(cur), gssname)))
+  while ((cur!=NULL) && (!tr_gss_names_matches(trp_peer_get_gss_names(cur), gssname)))
     cur=cur->next;
   return cur;
 }
 
-char *trp_peer_to_str(TALLOC_CTX *memctx, TRP_PEER *peer, const char *sep)
-{
-  if (sep==NULL)
-    sep=", ";
-  return talloc_asprintf(memctx,
-                         "%s:%u%s0x%04X",
-                         peer->server, peer->port, sep,
-                         peer->linkcost);
-}
-
-/* this is horribly inefficient but should be ok for small peer tables */
-char *trp_ptable_to_str(TALLOC_CTX *memctx, TRP_PTABLE *ptbl, const char *sep, const char *lineterm)
+TRP_PEER *trp_ptable_find_servicename(TRP_PTABLE *ptbl, TR_NAME *servicename)
 {
-  TALLOC_CTX *tmpctx=talloc_new(NULL);
-  TRP_PEER *peer=NULL;
-  char *result=talloc_strdup(tmpctx, "");
-
-  if (lineterm==NULL)
-    lineterm="\n";
-
-  /* this leaves intermediate result strings in the tmpctx context, we'll free these when
-   * we're done */
-  for (peer=ptbl->head; peer!=NULL; peer=peer->next)
-    result=talloc_asprintf(tmpctx, "%s%s%s", result, lineterm, trp_peer_to_str(tmpctx, peer, sep));
-
-  talloc_steal(memctx, result); /* hand result over to caller */
-  talloc_free(tmpctx); /* free detritus */
-  return result;
+  TRP_PEER *cur=ptbl->head;
+  while ((cur!=NULL) && (0 != tr_name_cmp(trp_peer_get_servicename(cur), servicename)))
+    cur=cur->next;
+  return cur;
 }
 
 TRP_PTABLE_ITER *trp_ptable_iter_new(TALLOC_CTX *mem_ctx)
@@ -194,13 +118,17 @@ TRP_PTABLE_ITER *trp_ptable_iter_new(TALLOC_CTX *mem_ctx)
 
 TRP_PEER *trp_ptable_iter_first(TRP_PTABLE_ITER *iter, TRP_PTABLE *ptbl)
 {
-  *iter=ptbl->head;
+  if (ptbl==NULL)
+    *iter=NULL;
+  else
+    *iter=ptbl->head;
   return *iter;
 }
 
 TRP_PEER *trp_ptable_iter_next(TRP_PTABLE_ITER *iter)
 {
-  *iter=(*iter)->next;
+  if (*iter!=NULL)
+    *iter=(*iter)->next;
   return *iter;
 }