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->servicename!=NULL)
  13     tr_free_name(peer->servicename);
  14   if (peer->gssname!=NULL)
  15     tr_free_name(peer->gssname);
  16   return 0;
  17 }
  18 TRP_PEER *trp_peer_new(TALLOC_CTX *memctx)
  19 {
  20   TRP_PEER *peer=talloc(memctx, TRP_PEER);
  21   if (peer!=NULL) {
  22     peer->next=NULL;
  23     peer->server=NULL;
  24     peer->servicename=NULL;
  25     peer->gssname=NULL;
  26     peer->port=0;
  27     peer->linkcost=TRP_LINKCOST_DEFAULT;
  28     peer->last_conn_attempt=(struct timespec){0,0};
  29     peer->outgoing_status=PEER_DISCONNECTED;
  30     peer->incoming_status=PEER_DISCONNECTED;
  31     peer->conn_status_cb=NULL;
  32     peer->conn_status_cookie=NULL;
  33     talloc_set_destructor((void *)peer, trp_peer_destructor);
  34   }
  35   return peer;
  36 }
  37
  38 void trp_peer_free(TRP_PEER *peer)
  39 {
  40   talloc_free(peer);
  41 }
  42
  43 static TRP_PEER *trp_peer_tail(TRP_PEER *peer)
  44 {
  45   while (peer->next!=NULL) {
  46     peer=peer->next;
  47   }
  48   return peer;
  49 }
  50
  51 char *trp_peer_get_server(TRP_PEER *peer)
  52 {
  53   return peer->server;
  54 }
  55
  56 static void trp_peer_set_servicename(TRP_PEER *peer, const char *server)
  57 {
  58   char *name=NULL;
  59   if (peer->servicename !=NULL)
  60     tr_free_name(peer->servicename);
  61
  62   if (server!=NULL)
  63     name=talloc_asprintf(NULL, "trustrouter/%s", server);
  64
  65   if (name!=NULL) {
  66     peer->servicename=tr_new_name(name);
  67     talloc_free(name);
  68   } else {
  69     peer->servicename=NULL;
  70   }
  71 }
  72
  73 /* copies input; on error, peer->gssname will be null */
  74 void trp_peer_set_server(TRP_PEER *peer, char *server)
  75 {
  76   peer->server=talloc_strdup(peer, server); /* will be null on error */
  77   trp_peer_set_servicename(peer, server);
  78 }
  79
  80 void trp_peer_set_gssname(TRP_PEER *peer, TR_NAME *gssname)
  81 {
  82   peer->gssname=gssname;
  83 }
  84
  85 /* get the peer gssname, caller must not free the result */
  86 TR_NAME *trp_peer_get_gssname(TRP_PEER *peer)
  87 {
  88   return peer->gssname;
  89 }
  90
  91 /* get a copy of the peer gssname, caller must free via tr_free_name() */
  92 TR_NAME *trp_peer_dup_gssname(TRP_PEER *peer)
  93 {
  94   return tr_dup_name(peer->gssname);
  95 }
  96
  97 /* get the service name (i.e., gssname we see when we connect to this peer) */
  98 TR_NAME *trp_peer_get_servicename(TRP_PEER *peer)
  99 {
 100   return peer->servicename;
 101 }
 102
 103 /* get a copy of the servicename, caller must free via tr_free_name */
 104 TR_NAME *trp_peer_dup_servicename(TRP_PEER *peer)
 105 {
 106   return tr_dup_name(peer->servicename);
 107 }
 108
 109 unsigned int trp_peer_get_port(TRP_PEER *peer)
 110 {
 111   return peer->port;
 112 }
 113
 114 void trp_peer_set_port(TRP_PEER *peer, unsigned int port)
 115 {
 116   peer->port=port;
 117 }
 118
 119 unsigned int trp_peer_get_linkcost(TRP_PEER *peer)
 120 {
 121   if (peer!=NULL)
 122     return peer->linkcost;
 123   else
 124     return 1;
 125 }
 126
 127 void trp_peer_set_linkcost(TRP_PEER *peer, unsigned int linkcost)
 128 {
 129   if ((linkcost>TRP_METRIC_INFINITY) && (linkcost!=TRP_METRIC_INVALID)) {
 130     /* This indicates a programming error, but probably means an already infinite metric
 131      * was (incorrectly) incremented. Issue a warning and proceed with an infinite metric. */
 132     tr_warning("trp_peer_set_linkcost: link cost > infinity encountered, setting to infinity");
 133     linkcost=TRP_METRIC_INFINITY;
 134   }
 135   peer->linkcost=linkcost;
 136 }
 137
 138 void trp_peer_set_conn_status_cb(TRP_PEER *peer, void (*cb)(TRP_PEER *, void *), void *cookie)
 139 {
 140   peer->conn_status_cb=cb;
 141   peer->conn_status_cookie=cookie;
 142 }
 143
 144 struct timespec *trp_peer_get_last_conn_attempt(TRP_PEER *peer)
 145 {
 146   return &(peer->last_conn_attempt);
 147 }
 148
 149 void trp_peer_set_last_conn_attempt(TRP_PEER *peer, struct timespec *time)
 150 {
 151   peer->last_conn_attempt=*time;
 152 }
 153
 154 TRP_PTABLE *trp_ptable_new(TALLOC_CTX *memctx)
 155 {
 156   TRP_PTABLE *ptbl=talloc(memctx, TRP_PTABLE);
 157   if (ptbl!=NULL) {
 158     ptbl->head=NULL;
 159   }
 160   return ptbl;
 161 }
 162
 163 void trp_peer_set_outgoing_status(TRP_PEER *peer, TRP_PEER_CONN_STATUS status)
 164 {
 165   int was_connected=trp_peer_is_connected(peer);
 166   peer->outgoing_status=status;
 167   tr_debug("trp_peer_set_outgoing_status: %s: status=%d peer connected was %d now %d.",
 168            trp_peer_get_gssname(peer)->buf, status, was_connected, trp_peer_is_connected(peer));
 169   if ((trp_peer_is_connected(peer) != was_connected) && (peer->conn_status_cb!=NULL))
 170     peer->conn_status_cb(peer, peer->conn_status_cookie);
 171 }
 172
 173 TRP_PEER_CONN_STATUS trp_peer_get_outgoing_status(TRP_PEER *peer)
 174 {
 175   return peer->outgoing_status;
 176 }
 177
 178 void trp_peer_set_incoming_status(TRP_PEER *peer, TRP_PEER_CONN_STATUS status)
 179 {
 180   int was_connected=trp_peer_is_connected(peer);
 181   peer->incoming_status=status;
 182   tr_debug("trp_peer_set_incoming_status: %s: status=%d peer connected was %d now %d.",
 183            trp_peer_get_gssname(peer)->buf, status, was_connected, trp_peer_is_connected(peer));
 184   if ((trp_peer_is_connected(peer) != was_connected) && (peer->conn_status_cb!=NULL))
 185     peer->conn_status_cb(peer, peer->conn_status_cookie);
 186 }
 187
 188 TRP_PEER_CONN_STATUS trp_peer_get_incoming_status(TRP_PEER *peer)
 189 {
 190   return peer->incoming_status;
 191 }
 192
 193 int trp_peer_is_connected(TRP_PEER *peer)
 194 {
 195   return (peer->outgoing_status==PEER_CONNECTED) && (peer->incoming_status==PEER_CONNECTED);
 196 }
 197
 198 void trp_ptable_free(TRP_PTABLE *ptbl)
 199 {
 200   talloc_free(ptbl);
 201 }
 202
 203 TRP_RC trp_ptable_add(TRP_PTABLE *ptbl, TRP_PEER *newpeer)
 204 {
 205   if (ptbl->head==NULL) {
 206     ptbl->head=newpeer;
 207   } else {
 208     trp_peer_tail(ptbl->head)->next=newpeer;
 209     talloc_steal(ptbl, newpeer);
 210   }
 211   return TRP_SUCCESS;
 212 }
 213
 214 /* peer pointer is invalid after successful removal. Does nothing and returns
 215  * TRP_ERROR if peer is not in the list. */
 216 TRP_RC trp_ptable_remove(TRP_PTABLE *ptbl, TRP_PEER *peer)
 217 {
 218   TRP_PEER *cur=NULL;
 219   TRP_PEER *last=NULL;
 220   if (ptbl->head!=NULL) {
 221     if (ptbl->head==peer) {
 222       /* special case for removing head of list */
 223       cur=ptbl->head;
 224       ptbl->head=ptbl->head->next; /* advance the head */
 225       trp_peer_free(cur);
 226     }
 227     for (cur=ptbl->head->next; cur!=NULL; last=cur,cur=cur->next) {
 228       if (cur==peer) {
 229         if (last!=NULL)
 230           last->next=cur->next;
 231         trp_peer_free(cur);
 232         return TRP_SUCCESS;
 233       }
 234     }
 235   }
 236   return TRP_ERROR;
 237 }
 238
 239 TRP_PEER *trp_ptable_find_gssname(TRP_PTABLE *ptbl, TR_NAME *gssname)
 240 {
 241   TRP_PEER *cur=ptbl->head;
 242   while ((cur!=NULL) && (0 != tr_name_cmp(trp_peer_get_gssname(cur), gssname)))
 243     cur=cur->next;
 244   return cur;
 245 }
 246
 247 TRP_PEER *trp_ptable_find_servicename(TRP_PTABLE *ptbl, TR_NAME *servicename)
 248 {
 249   TRP_PEER *cur=ptbl->head;
 250   while ((cur!=NULL) && (0 != tr_name_cmp(trp_peer_get_servicename(cur), servicename)))
 251     cur=cur->next;
 252   return cur;
 253 }
 254
 255 char *trp_peer_to_str(TALLOC_CTX *memctx, TRP_PEER *peer, const char *sep)
 256 {
 257   if (sep==NULL)
 258     sep=", ";
 259   return talloc_asprintf(memctx,
 260                          "%s:%u%s0x%04X",
 261                          peer->server, peer->port, sep,
 262                          peer->linkcost);
 263 }
 264
 265 /* this is horribly inefficient but should be ok for small peer tables */
 266 char *trp_ptable_to_str(TALLOC_CTX *memctx, TRP_PTABLE *ptbl, const char *sep, const char *lineterm)
 267 {
 268   TALLOC_CTX *tmpctx=talloc_new(NULL);
 269   TRP_PEER *peer=NULL;
 270   char *result=talloc_strdup(tmpctx, "");
 271
 272   if (lineterm==NULL)
 273     lineterm="\n";
 274
 275   /* this leaves intermediate result strings in the tmpctx context, we'll free these when
 276    * we're done */
 277   for (peer=ptbl->head; peer!=NULL; peer=peer->next)
 278     result=talloc_asprintf(tmpctx, "%s%s%s", result, lineterm, trp_peer_to_str(tmpctx, peer, sep));
 279
 280   talloc_steal(memctx, result); /* hand result over to caller */
 281   talloc_free(tmpctx); /* free detritus */
 282   return result;
 283 }
 284
 285 TRP_PTABLE_ITER *trp_ptable_iter_new(TALLOC_CTX *mem_ctx)
 286 {
 287   TRP_PTABLE_ITER *iter=talloc(mem_ctx, TRP_PTABLE_ITER);
 288   *iter=NULL;
 289   return iter;
 290 }
 291
 292 TRP_PEER *trp_ptable_iter_first(TRP_PTABLE_ITER *iter, TRP_PTABLE *ptbl)
 293 {
 294   *iter=ptbl->head;
 295   return *iter;
 296 }
 297
 298 TRP_PEER *trp_ptable_iter_next(TRP_PTABLE_ITER *iter)
 299 {
 300   *iter=(*iter)->next;
 301   return *iter;
 302 }
 303
 304 void trp_ptable_iter_free(TRP_PTABLE_ITER *iter)
 305 {
 306   talloc_free(iter);
 307 }
 308