trp/trp_upd.c

   1 #include <jansson.h>
   2 #include <talloc.h>
   3
   4 #include <trust_router/tr_name.h>
   5 #include <trp_internal.h>
   6 #include <tr_debug.h>
   7
   8
   9 /* static prototypes */
  10 static void *trp_inforec_route_new(TALLOC_CTX *mem_ctx);
  11 static void trp_inforec_route_print(TRP_INFOREC_DATA);
  12
  13
  14 struct trp_inforec_type_entry {
  15   const char *name;
  16   TRP_INFOREC_TYPE type;
  17   void *(*allocate)(TALLOC_CTX *);
  18   void (*print)(TRP_INFOREC_DATA);
  19 };
  20 static struct trp_inforec_type_entry trp_inforec_type_table[] = {
  21   { "route", TRP_INFOREC_TYPE_ROUTE, trp_inforec_route_new, trp_inforec_route_print },
  22   { "comm", TRP_INFOREC_TYPE_COMMUNITY, NULL, NULL },
  23   { NULL, TRP_INFOREC_TYPE_UNKNOWN, NULL, NULL } /* must be the last entry */
  24 };
  25
  26
  27 /* look up an entry in the trp_inforec_type_table */
  28 static struct trp_inforec_type_entry *get_trp_inforec_type_entry(TRP_INFOREC_TYPE msgtype)
  29 {
  30   struct trp_inforec_type_entry *entry=trp_inforec_type_table;
  31
  32   while ((entry->type != TRP_INFOREC_TYPE_UNKNOWN)
  33         && (entry->type != msgtype)) {
  34     entry ++;
  35   }
  36   return entry;
  37 }
  38
  39 /* translate strings to codes */
  40 TRP_INFOREC_TYPE trp_inforec_type_from_string(const char *s)
  41 {
  42   struct trp_inforec_type_entry *entry=trp_inforec_type_table;
  43
  44   while ((entry->type != TRP_INFOREC_TYPE_UNKNOWN)
  45         && (strcmp(s, entry->name)!=0)) {
  46     entry++;
  47   }
  48   return entry->type;
  49 }
  50 /* translate codes to strings (do not need to be freed)
  51  * Returns NULL on an unknown code */
  52 const char *trp_inforec_type_to_string(TRP_INFOREC_TYPE msgtype)
  53 {
  54   struct trp_inforec_type_entry *entry=get_trp_inforec_type_entry(msgtype);
  55   return entry->name;
  56 }
  57
  58 /* called by talloc when destroying an update message body */
  59 static int trp_inforec_route_destructor(void *object)
  60 {
  61   TRP_INFOREC_ROUTE *body=talloc_get_type_abort(object, TRP_INFOREC_ROUTE);
  62
  63   /* clean up TR_NAME data, which are not managed by talloc */
  64   if (body->comm != NULL) {
  65     tr_free_name(body->comm);
  66     body->comm=NULL;
  67     tr_debug("trp_inforec_route_destructor: freed community");
  68   }
  69   if (body->realm != NULL) {
  70     tr_free_name(body->realm);
  71     body->realm=NULL;
  72     tr_debug("trp_inforec_route_destructor: freed realm");
  73   }
  74   if (body->trust_router != NULL) {
  75     tr_free_name(body->trust_router);
  76     body->trust_router=NULL;
  77     tr_debug("trp_inforec_route_destructor: freed trust_router");
  78   }
  79   if (body->next_hop != NULL) {
  80     tr_free_name(body->next_hop);
  81     body->next_hop=NULL;
  82     tr_debug("trp_inforec_route_destructor: freed next_hop");
  83   }
  84
  85   return 0;
  86 }
  87
  88 static void *trp_inforec_route_new(TALLOC_CTX *mem_ctx)
  89 {
  90   TRP_INFOREC_ROUTE *new_rec=talloc(mem_ctx, TRP_INFOREC_ROUTE);
  91
  92   if (new_rec != NULL) {
  93     new_rec->comm=NULL;
  94     new_rec->realm=NULL;
  95     new_rec->trust_router=NULL;
  96     new_rec->next_hop=NULL;
  97     new_rec->metric=TRP_METRIC_INFINITY;
  98     new_rec->interval=0;
  99     talloc_set_destructor((void *)new_rec, trp_inforec_route_destructor);
 100   }
 101   return new_rec;
 102 }
 103
 104 TRP_INFOREC *trp_inforec_get_next(TRP_INFOREC *rec)
 105 {
 106   if (rec!=NULL)
 107     return rec->next;
 108   else
 109     return NULL;
 110 }
 111
 112 static TRP_INFOREC *trp_inforec_get_tail(TRP_INFOREC *rec)
 113 {
 114   while (rec!=NULL)
 115     rec=trp_inforec_get_next(rec);
 116   return rec;
 117 }
 118
 119 void trp_inforec_set_next(TRP_INFOREC *rec, TRP_INFOREC *next_rec)
 120 {
 121   if (rec !=NULL)
 122     rec->next=next_rec;
 123 }
 124
 125 TRP_INFOREC_TYPE trp_inforec_get_type(TRP_INFOREC *rec)
 126 {
 127   if (rec)
 128     return rec->type;
 129   else
 130     return TRP_INFOREC_TYPE_UNKNOWN;
 131 }
 132
 133 void trp_inforec_set_type(TRP_INFOREC *rec, TRP_INFOREC_TYPE type)
 134 {
 135   if (rec!=NULL)
 136     rec->type=type;
 137 }
 138
 139 TR_NAME *trp_inforec_get_comm(TRP_INFOREC *rec)
 140 {
 141   switch (rec->type) {
 142   case TRP_INFOREC_TYPE_ROUTE:
 143     if (rec->data.route!=NULL)
 144       return rec->data.route->comm;
 145     break;
 146   default:
 147     break;
 148   }
 149   return NULL;
 150 }
 151
 152 TRP_RC trp_inforec_set_comm(TRP_INFOREC *rec, TR_NAME *comm)
 153 {
 154   switch (rec->type) {
 155   case TRP_INFOREC_TYPE_ROUTE:
 156     if (rec->data.route!=NULL) {
 157       rec->data.route->comm=comm;
 158       return TRP_SUCCESS;
 159     }
 160     break;
 161   default:
 162     break;
 163   }
 164   return TRP_ERROR;
 165 }
 166
 167 TR_NAME *trp_inforec_get_realm(TRP_INFOREC *rec)
 168 {
 169   switch (rec->type) {
 170   case TRP_INFOREC_TYPE_ROUTE:
 171     if (rec->data.route!=NULL)
 172       return rec->data.route->realm;
 173     break;
 174   default:
 175     break;
 176   }
 177   return NULL;
 178 }
 179
 180 TRP_RC trp_inforec_set_realm(TRP_INFOREC *rec, TR_NAME *realm)
 181 {
 182   switch (rec->type) {
 183   case TRP_INFOREC_TYPE_ROUTE:
 184     if (rec->data.route!=NULL) {
 185       rec->data.route->realm=realm;
 186       return TRP_SUCCESS;
 187     }
 188     break;
 189   default:
 190     break;
 191   }
 192   return TRP_ERROR;
 193 }
 194
 195 TR_NAME *trp_inforec_get_trust_router(TRP_INFOREC *rec)
 196 {
 197   switch (rec->type) {
 198   case TRP_INFOREC_TYPE_ROUTE:
 199     if (rec->data.route!=NULL)
 200       return rec->data.route->trust_router;
 201     break;
 202   default:
 203     break;
 204   }
 205   return NULL;
 206 }
 207
 208 TRP_RC trp_inforec_set_trust_router(TRP_INFOREC *rec, TR_NAME *trust_router)
 209 {
 210   switch (rec->type) {
 211   case TRP_INFOREC_TYPE_ROUTE:
 212     if (rec->data.route!=NULL) {
 213       rec->data.route->trust_router=trust_router;
 214       return TRP_SUCCESS;
 215     }
 216     break;
 217   default:
 218     break;
 219   }
 220   return TRP_ERROR;
 221 }
 222
 223 TR_NAME *trp_inforec_get_next_hop(TRP_INFOREC *rec)
 224 {
 225   switch (rec->type) {
 226   case TRP_INFOREC_TYPE_ROUTE:
 227     if (rec->data.route!=NULL)
 228       return rec->data.route->next_hop;
 229     break;
 230   default:
 231     break;
 232   }
 233   return NULL;
 234 }
 235
 236 TRP_RC trp_inforec_set_next_hop(TRP_INFOREC *rec, TR_NAME *next_hop)
 237 {
 238   switch (rec->type) {
 239   case TRP_INFOREC_TYPE_ROUTE:
 240     if (rec->data.route!=NULL) {
 241       rec->data.route->next_hop=next_hop;
 242       return TRP_SUCCESS;
 243     }
 244     break;
 245   default:
 246     break;
 247   }
 248   return TRP_ERROR;
 249 }
 250
 251 unsigned int trp_inforec_get_metric(TRP_INFOREC *rec)
 252 {
 253   switch (rec->type) {
 254   case TRP_INFOREC_TYPE_ROUTE:
 255     if (rec->data.route!=NULL)
 256       return rec->data.route->metric;
 257     break;
 258   default:
 259     break;
 260   }
 261   return TRP_METRIC_INVALID;
 262 }
 263
 264 TRP_RC trp_inforec_set_metric(TRP_INFOREC *rec, unsigned int metric)
 265 {
 266   switch (rec->type) {
 267   case TRP_INFOREC_TYPE_ROUTE:
 268     if (rec->data.route!=NULL) {
 269       rec->data.route->metric=metric;
 270       return TRP_SUCCESS;
 271     }
 272     break;
 273   default:
 274     break;
 275   }
 276   return TRP_ERROR;
 277 }
 278
 279 unsigned int trp_inforec_get_interval(TRP_INFOREC *rec)
 280 {
 281   switch (rec->type) {
 282   case TRP_INFOREC_TYPE_ROUTE:
 283     if (rec->data.route!=NULL)
 284       return rec->data.route->interval;
 285     break;
 286   default:
 287     break;
 288   }
 289   return TRP_INTERVAL_INVALID;
 290 }
 291
 292 TRP_RC trp_inforec_set_interval(TRP_INFOREC *rec, unsigned int interval)
 293 {
 294   switch (rec->type) {
 295   case TRP_INFOREC_TYPE_ROUTE:
 296     if (rec->data.route!=NULL) {
 297       rec->data.route->interval=interval;
 298       return TRP_SUCCESS;
 299   default:
 300     break;
 301     }
 302     break;
 303   }
 304   return TRP_ERROR;
 305 }
 306
 307 /* for internal use only; must set rec->type before calling this */
 308 static TRP_RC trp_inforec_set_data(TRP_INFOREC *rec, void *data)
 309 {
 310   if (data==NULL)
 311     return TRP_ERROR;
 312
 313   switch (rec->type) {
 314   case TRP_INFOREC_TYPE_ROUTE:
 315     rec->data.route=talloc_get_type(data, TRP_INFOREC_ROUTE);
 316     break;
 317   default:
 318     return TRP_BADTYPE;
 319   }
 320   return TRP_SUCCESS;
 321 }
 322
 323 /* generic record type */
 324 TRP_INFOREC *trp_inforec_new(TALLOC_CTX *mem_ctx, TRP_INFOREC_TYPE type)
 325 {
 326   TRP_INFOREC *new_rec=talloc(mem_ctx, TRP_INFOREC);
 327   struct trp_inforec_type_entry *dtype=get_trp_inforec_type_entry(type);
 328
 329   if ((new_rec != NULL) && (dtype->type != TRP_INFOREC_TYPE_UNKNOWN)) {
 330     trp_inforec_set_type(new_rec, type);
 331     trp_inforec_set_next(new_rec, NULL);
 332     if (dtype->allocate!=NULL) {
 333       if (TRP_SUCCESS!=trp_inforec_set_data(new_rec, dtype->allocate(new_rec))) {
 334         talloc_free(new_rec);
 335         new_rec=NULL;
 336       }
 337     }
 338   }
 339   return new_rec;
 340 }
 341
 342 void trp_inforec_free(TRP_INFOREC *rec)
 343 {
 344   if (rec!=NULL)
 345     talloc_free(rec);
 346 }
 347
 348 static int trp_upd_destructor(void *object)
 349 {
 350   TRP_UPD *upd=talloc_get_type_abort(object, TRP_UPD);
 351   if (upd->peer!=NULL)
 352     tr_free_name(upd->peer);
 353   return 0;
 354 }
 355
 356 TRP_UPD *trp_upd_new(TALLOC_CTX *mem_ctx)
 357 {
 358   TRP_UPD *new_body=talloc(mem_ctx, TRP_UPD);
 359
 360   if (new_body!=NULL) {
 361     new_body->records=NULL;
 362     new_body->peer=NULL;
 363     talloc_set_destructor((void *)new_body, trp_upd_destructor);
 364   }
 365   return new_body;
 366 }
 367
 368 void trp_upd_free(TRP_UPD *update)
 369 {
 370   if (update!=NULL)
 371     talloc_free(update);
 372 }
 373
 374 TRP_INFOREC *trp_upd_get_inforec(TRP_UPD *upd)
 375 {
 376   if (upd!=NULL)
 377     return upd->records;
 378   else
 379     return NULL;
 380 }
 381
 382 void trp_upd_set_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
 383 {
 384   if (upd!=NULL)
 385     upd->records=rec;
 386 }
 387
 388 void trp_upd_add_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
 389 {
 390   if (upd->records==NULL)
 391     upd->records=rec;
 392   else
 393     trp_inforec_set_next(trp_inforec_get_tail(upd->records), rec);
 394   talloc_steal(upd, rec);
 395 }
 396
 397 TR_NAME *trp_upd_get_peer(TRP_UPD *upd)
 398 {
 399   return upd->peer;
 400 }
 401
 402 void trp_upd_set_peer(TRP_UPD *upd, TR_NAME *peer)
 403 {
 404   TRP_INFOREC *rec=NULL;
 405   TR_NAME *cpy=NULL;
 406
 407   upd->peer=peer;
 408   for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
 409     if (trp_inforec_set_next_hop(rec, cpy=tr_dup_name(peer)) != TRP_SUCCESS) {
 410       tr_err("trp_upd_set_peer: error setting peer.");
 411       tr_free_name(cpy);
 412     }
 413   }
 414 }
 415
 416 /* pretty print */
 417 static void trp_inforec_route_print(TRP_INFOREC_DATA data)
 418 {
 419   if (data.route!=NULL) {
 420     printf("     community=%.*s\n     realm=%.*s\n     trust_router=%.*s\n     metric=%d\n     interval=%d]\n",
 421            data.route->comm->len, data.route->comm->buf,
 422            data.route->realm->len, data.route->realm->buf,
 423            data.route->trust_router->len, data.route->trust_router->buf,
 424            data.route->metric, data.route->interval);
 425   }
 426 }
 427