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
  80   return 0;
  81 }
  82
  83 static void *trp_inforec_route_new(TALLOC_CTX *mem_ctx)
  84 {
  85   TRP_INFOREC_ROUTE *new_rec=talloc(mem_ctx, TRP_INFOREC_ROUTE);
  86
  87   if (new_rec != NULL) {
  88     new_rec->comm=NULL;
  89     new_rec->realm=NULL;
  90     new_rec->trust_router=NULL;
  91     new_rec->metric=TRP_METRIC_INFINITY;
  92     new_rec->interval=0;
  93     talloc_set_destructor((void *)new_rec, trp_inforec_route_destructor);
  94   }
  95   return new_rec;
  96 }
  97
  98 TRP_INFOREC *trp_inforec_get_next(TRP_INFOREC *rec)
  99 {
 100   if (rec!=NULL)
 101     return rec->next;
 102   else
 103     return NULL;
 104 }
 105
 106 void trp_inforec_set_next(TRP_INFOREC *rec, TRP_INFOREC *next_rec)
 107 {
 108   if (rec !=NULL)
 109     rec->next=next_rec;
 110 }
 111
 112 TRP_INFOREC_TYPE trp_inforec_get_type(TRP_INFOREC *rec)
 113 {
 114   if (rec)
 115     return rec->type;
 116   else
 117     return TRP_INFOREC_TYPE_UNKNOWN;
 118 }
 119
 120 void trp_inforec_set_type(TRP_INFOREC *rec, TRP_INFOREC_TYPE type)
 121 {
 122   if (rec!=NULL)
 123     rec->type=type;
 124 }
 125
 126 TR_NAME *trp_inforec_get_comm(TRP_INFOREC *rec)
 127 {
 128   switch (rec->type) {
 129   case TRP_INFOREC_TYPE_ROUTE:
 130     if (rec->data.route!=NULL)
 131       return rec->data.route->comm;
 132     break;
 133   default:
 134     break;
 135   }
 136   return NULL;
 137 }
 138
 139 TRP_RC trp_inforec_set_comm(TRP_INFOREC *rec, TR_NAME *comm)
 140 {
 141   switch (rec->type) {
 142   case TRP_INFOREC_TYPE_ROUTE:
 143     if (rec->data.route!=NULL) {
 144       rec->data.route->comm=comm;
 145       return TRP_SUCCESS;
 146     }
 147     break;
 148   default:
 149     break;
 150   }
 151   return TRP_ERROR;
 152 }
 153
 154 TR_NAME *trp_inforec_get_realm(TRP_INFOREC *rec)
 155 {
 156   switch (rec->type) {
 157   case TRP_INFOREC_TYPE_ROUTE:
 158     if (rec->data.route!=NULL)
 159       return rec->data.route->realm;
 160     break;
 161   default:
 162     break;
 163   }
 164   return NULL;
 165 }
 166
 167 TRP_RC trp_inforec_set_realm(TRP_INFOREC *rec, TR_NAME *realm)
 168 {
 169   switch (rec->type) {
 170   case TRP_INFOREC_TYPE_ROUTE:
 171     if (rec->data.route!=NULL) {
 172       rec->data.route->realm=realm;
 173       return TRP_SUCCESS;
 174     }
 175     break;
 176   default:
 177     break;
 178   }
 179   return TRP_ERROR;
 180 }
 181
 182 TR_NAME *trp_inforec_get_trust_router(TRP_INFOREC *rec)
 183 {
 184   switch (rec->type) {
 185   case TRP_INFOREC_TYPE_ROUTE:
 186     if (rec->data.route!=NULL)
 187       return rec->data.route->trust_router;
 188     break;
 189   default:
 190     break;
 191   }
 192   return NULL;
 193 }
 194
 195 TRP_RC trp_inforec_set_trust_router(TRP_INFOREC *rec, TR_NAME *trust_router)
 196 {
 197   switch (rec->type) {
 198   case TRP_INFOREC_TYPE_ROUTE:
 199     if (rec->data.route!=NULL) {
 200       rec->data.route->trust_router=trust_router;
 201       return TRP_SUCCESS;
 202     }
 203     break;
 204   default:
 205     break;
 206   }
 207   return TRP_ERROR;
 208 }
 209
 210 unsigned int trp_inforec_get_metric(TRP_INFOREC *rec)
 211 {
 212   switch (rec->type) {
 213   case TRP_INFOREC_TYPE_ROUTE:
 214     if (rec->data.route!=NULL)
 215       return rec->data.route->metric;
 216     break;
 217   default:
 218     break;
 219   }
 220   return TRP_METRIC_INVALID;
 221 }
 222
 223 TRP_RC trp_inforec_set_metric(TRP_INFOREC *rec, unsigned int metric)
 224 {
 225   switch (rec->type) {
 226   case TRP_INFOREC_TYPE_ROUTE:
 227     if (rec->data.route!=NULL) {
 228       rec->data.route->metric=metric;
 229       return TRP_SUCCESS;
 230     }
 231     break;
 232   default:
 233     break;
 234   }
 235   return TRP_ERROR;
 236 }
 237
 238 unsigned int trp_inforec_get_interval(TRP_INFOREC *rec)
 239 {
 240   switch (rec->type) {
 241   case TRP_INFOREC_TYPE_ROUTE:
 242     if (rec->data.route!=NULL)
 243       return rec->data.route->interval;
 244     break;
 245   default:
 246     break;
 247   }
 248   return TRP_INTERVAL_INVALID;
 249 }
 250
 251 TRP_RC trp_inforec_set_interval(TRP_INFOREC *rec, unsigned int interval)
 252 {
 253   switch (rec->type) {
 254   case TRP_INFOREC_TYPE_ROUTE:
 255     if (rec->data.route!=NULL) {
 256       rec->data.route->interval=interval;
 257       return TRP_SUCCESS;
 258   default:
 259     break;
 260     }
 261     break;
 262   }
 263   return TRP_ERROR;
 264 }
 265
 266 /* for internal use only; must set rec->type before calling this */
 267 static TRP_RC trp_inforec_set_data(TRP_INFOREC *rec, void *data)
 268 {
 269   if (data==NULL)
 270     return TRP_ERROR;
 271
 272   switch (rec->type) {
 273   case TRP_INFOREC_TYPE_ROUTE:
 274     rec->data.route=talloc_get_type(data, TRP_INFOREC_ROUTE);
 275     break;
 276   default:
 277     return TRP_BADTYPE;
 278   }
 279   return TRP_SUCCESS;
 280 }
 281
 282 /* generic record type */
 283 TRP_INFOREC *trp_inforec_new(TALLOC_CTX *mem_ctx, TRP_INFOREC_TYPE type)
 284 {
 285   TRP_INFOREC *new_rec=talloc(mem_ctx, TRP_INFOREC);
 286   struct trp_inforec_type_entry *dtype=get_trp_inforec_type_entry(type);
 287
 288   if ((new_rec != NULL) && (dtype->type != TRP_INFOREC_TYPE_UNKNOWN)) {
 289     trp_inforec_set_type(new_rec, type);
 290     trp_inforec_set_next(new_rec, NULL);
 291     if (dtype->allocate!=NULL) {
 292       if (TRP_SUCCESS!=trp_inforec_set_data(new_rec, dtype->allocate(new_rec))) {
 293         talloc_free(new_rec);
 294         new_rec=NULL;
 295       }
 296     }
 297   }
 298   return new_rec;
 299 }
 300
 301 void trp_inforec_free(TRP_INFOREC *rec)
 302 {
 303   if (rec!=NULL)
 304     talloc_free(rec);
 305 }
 306
 307 TRP_UPD *trp_upd_new(TALLOC_CTX *mem_ctx)
 308 {
 309   TRP_UPD *new_body=talloc(mem_ctx, TRP_UPD);
 310
 311   if (new_body!=NULL) {
 312     new_body->records=NULL;
 313   }
 314   return new_body;
 315 }
 316
 317 void trp_upd_free(TRP_UPD *update)
 318 {
 319   if (update!=NULL)
 320     talloc_free(update);
 321 }
 322
 323 TRP_INFOREC *trp_upd_get_inforec(TRP_UPD *upd)
 324 {
 325   if (upd!=NULL)
 326     return upd->records;
 327   else
 328     return NULL;
 329 }
 330
 331 void trp_upd_set_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
 332 {
 333   if (upd!=NULL)
 334     upd->records=rec;
 335 }
 336
 337 /* pretty print */
 338 static void trp_inforec_route_print(TRP_INFOREC_DATA data)
 339 {
 340   if (data.route!=NULL) {
 341     printf("     community=%.*s\n     realm=%.*s\n     trust_router=%.*s\n     metric=%d\n     interval=%d]\n",
 342            data.route->comm->len, data.route->comm->buf,
 343            data.route->realm->len, data.route->realm->buf,
 344            data.route->trust_router->len, data.route->trust_router->buf,
 345            data.route->metric, data.route->interval);
 346   }
 347 }
 348