trp/trp_ptable.c

   1 #include <time.h>
   2 #include <talloc.h>
   3
   4 #include <trust_router/tr_name.h>
   5 #include <trp_internal.h>
   6 #include <trp_ptable.h>
   7 #include <tr_debug.h>
   8
   9 static int trp_peer_destructor(void *object)
  10 {
  11   TRP_PEER *peer=talloc_get_type_abort(object, TRP_PEER);
  12   if (peer->gssname!=NULL)
  13     tr_free_name(peer->gssname);
  14   return 0;
  15 }
  16 TRP_PEER *trp_peer_new(TALLOC_CTX *memctx)
  17 {
  18   TRP_PEER *peer=talloc(memctx, TRP_PEER);
  19   if (peer!=NULL) {
  20     peer->next=NULL;
  21     peer->server=NULL;
  22     peer->gssname=NULL;
  23     peer->port=0;
  24     peer->linkcost=TRP_METRIC_INFINITY;
  25     peer->last_conn_attempt=(struct timespec){0,0};
  26     talloc_set_destructor((void *)peer, trp_peer_destructor);
  27   }
  28   return peer;
  29 }
  30
  31 void trp_peer_free(TRP_PEER *peer)
  32 {
  33   talloc_free(peer);
  34 }
  35
  36 static TRP_PEER *trp_peer_tail(TRP_PEER *peer)
  37 {
  38   while (peer->next!=NULL) {
  39     peer=peer->next;
  40   }
  41   return peer;
  42 }
  43
  44 char *trp_peer_get_server(TRP_PEER *peer)
  45 {
  46   return peer->server;
  47 }
  48
  49 /* copies input; on error, peer->gssname will be null */
  50 void trp_peer_set_server(TRP_PEER *peer, char *server)
  51 {
  52   peer->server=talloc_strdup(peer, server); /* will be null on error */
  53 }
  54
  55 void trp_peer_set_gssname(TRP_PEER *peer, TR_NAME *gssname)
  56 {
  57   peer->gssname=gssname;
  58 }
  59
  60 /* get the peer gssname, caller must not free the result */
  61 TR_NAME *trp_peer_get_gssname(TRP_PEER *peer)
  62 {
  63   return peer->gssname;
  64 }
  65
  66 /* get a copy of the peer gssname, caller must free via tr_free_name() */
  67 TR_NAME *trp_peer_dup_gssname(TRP_PEER *peer)
  68 {
  69   return tr_dup_name(peer->gssname);
  70 }
  71
  72 unsigned int trp_peer_get_port(TRP_PEER *peer)
  73 {
  74   return peer->port;
  75 }
  76
  77 void trp_peer_set_port(TRP_PEER *peer, unsigned int port)
  78 {
  79   peer->port=port;
  80 }
  81
  82 unsigned int trp_peer_get_linkcost(TRP_PEER *peer)
  83 {
  84   if (peer!=NULL)
  85     return peer->linkcost;
  86   else
  87     return 1;
  88 }
  89
  90 void trp_peer_set_linkcost(TRP_PEER *peer, unsigned int linkcost)
  91 {
  92   if ((linkcost>TRP_METRIC_INFINITY) && (linkcost!=TRP_METRIC_INVALID)) {
  93     /* This indicates a programming error, but probably means an already infinite metric
  94      * was (incorrectly) incremented. Issue a warning and proceed with an infinite metric. */
  95     tr_warning("trp_peer_set_linkcost: link cost > infinity encountered, setting to infinity");
  96     linkcost=TRP_METRIC_INFINITY;
  97   }
  98   peer->linkcost=linkcost;
  99 }
 100
 101 struct timespec *trp_peer_get_last_conn_attempt(TRP_PEER *peer)
 102 {
 103   return &(peer->last_conn_attempt);
 104 }
 105
 106 void trp_peer_set_last_conn_attempt(TRP_PEER *peer, struct timespec *time)
 107 {
 108   peer->last_conn_attempt=*time;
 109 }
 110
 111 TRP_PTABLE *trp_ptable_new(TALLOC_CTX *memctx)
 112 {
 113   TRP_PTABLE *ptbl=talloc(memctx, TRP_PTABLE);
 114   if (ptbl!=NULL) {
 115     ptbl->head=NULL;
 116   }
 117   return ptbl;
 118 }
 119
 120 void trp_ptable_free(TRP_PTABLE *ptbl)
 121 {
 122   talloc_free(ptbl);
 123 }
 124
 125 TRP_RC trp_ptable_add(TRP_PTABLE *ptbl, TRP_PEER *newpeer)
 126 {
 127   if (ptbl->head==NULL) {
 128     ptbl->head=newpeer;
 129   } else {
 130     trp_peer_tail(ptbl->head)->next=newpeer;
 131     talloc_steal(ptbl, newpeer);
 132   }
 133   return TRP_SUCCESS;
 134 }
 135
 136 /* peer pointer is invalid after successful removal. Does nothing and returns
 137  * TRP_ERROR if peer is not in the list. */
 138 TRP_RC trp_ptable_remove(TRP_PTABLE *ptbl, TRP_PEER *peer)
 139 {
 140   TRP_PEER *cur=NULL;
 141   TRP_PEER *last=NULL;
 142   if (ptbl->head!=NULL) {
 143     if (ptbl->head==peer) {
 144       /* special case for removing head of list */
 145       cur=ptbl->head;
 146       ptbl->head=ptbl->head->next; /* advance the head */
 147       trp_peer_free(cur);
 148     }
 149     for (cur=ptbl->head->next; cur!=NULL; last=cur,cur=cur->next) {
 150       if (cur==peer) {
 151         if (last!=NULL)
 152           last->next=cur->next;
 153         trp_peer_free(cur);
 154         return TRP_SUCCESS;
 155       }
 156     }
 157   }
 158   return TRP_ERROR;
 159 }
 160
 161 TRP_PEER *trp_ptable_find(TRP_PTABLE *ptbl, TR_NAME *gssname)
 162 {
 163   TRP_PEER *cur=ptbl->head;
 164   while ((cur!=NULL) && (0 != tr_name_cmp(trp_peer_get_gssname(cur), gssname)))
 165     cur=cur->next;
 166   return cur;
 167 }
 168
 169 char *trp_peer_to_str(TALLOC_CTX *memctx, TRP_PEER *peer, const char *sep)
 170 {
 171   if (sep==NULL)
 172     sep=", ";
 173   return talloc_asprintf(memctx,
 174                          "%s:%u%s0x%04X",
 175                          peer->server, peer->port, sep,
 176                          peer->linkcost);
 177 }
 178
 179 /* this is horribly inefficient but should be ok for small peer tables */
 180 char *trp_ptable_to_str(TALLOC_CTX *memctx, TRP_PTABLE *ptbl, const char *sep, const char *lineterm)
 181 {
 182   TALLOC_CTX *tmpctx=talloc_new(NULL);
 183   TRP_PEER *peer=NULL;
 184   char *result=talloc_strdup(tmpctx, "");
 185
 186   if (lineterm==NULL)
 187     lineterm="\n";
 188
 189   /* this leaves intermediate result strings in the tmpctx context, we'll free these when
 190    * we're done */
 191   for (peer=ptbl->head; peer!=NULL; peer=peer->next)
 192     result=talloc_asprintf(tmpctx, "%s%s%s", result, lineterm, trp_peer_to_str(tmpctx, peer, sep));
 193
 194   talloc_steal(memctx, result); /* hand result over to caller */
 195   talloc_free(tmpctx); /* free detritus */
 196   return result;
 197 }
 198
 199 TRP_PTABLE_ITER *trp_ptable_iter_new(TALLOC_CTX *mem_ctx)
 200 {
 201   TRP_PTABLE_ITER *iter=talloc(mem_ctx, TRP_PTABLE_ITER);
 202   *iter=NULL;
 203   return iter;
 204 }
 205
 206 TRP_PEER *trp_ptable_iter_first(TRP_PTABLE_ITER *iter, TRP_PTABLE *ptbl)
 207 {
 208   *iter=ptbl->head;
 209   return *iter;
 210 }
 211
 212 TRP_PEER *trp_ptable_iter_next(TRP_PTABLE_ITER *iter)
 213 {
 214   *iter=(*iter)->next;
 215   return *iter;
 216 }
 217
 218 void trp_ptable_iter_free(TRP_PTABLE_ITER *iter)
 219 {
 220   talloc_free(iter);
 221 }
 222