2 * Copyright (c) 2016, JANET(UK)
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
16 * 3. Neither the name of JANET(UK) nor the names of its contributors
17 * may be used to endorse or promote products derived from this software
18 * without specific prior written permission.
20 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
21 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
22 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
23 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
24 * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
25 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
26 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
27 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
29 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
30 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
31 * OF THE POSSIBILITY OF SUCH DAMAGE.
38 #include <trust_router/tr_name.h>
39 #include <trp_internal.h>
45 /* static prototypes */
46 static TRP_INFOREC_DATA *trp_inforec_route_new(TALLOC_CTX *mem_ctx);
47 static void trp_inforec_route_print(TRP_INFOREC_DATA *);
48 static TRP_INFOREC_DATA *trp_inforec_comm_new(TALLOC_CTX *mem_ctx);
49 static void trp_inforec_comm_print(TRP_INFOREC_DATA *);
52 struct trp_inforec_type_entry {
54 TRP_INFOREC_TYPE type;
55 TRP_INFOREC_DATA *(*allocate)(TALLOC_CTX *);
56 void (*print)(TRP_INFOREC_DATA *);
58 static struct trp_inforec_type_entry trp_inforec_type_table[] = {
59 { "route", TRP_INFOREC_TYPE_ROUTE, trp_inforec_route_new, trp_inforec_route_print },
60 { "comm", TRP_INFOREC_TYPE_COMMUNITY, trp_inforec_comm_new, trp_inforec_comm_print },
61 { NULL, TRP_INFOREC_TYPE_UNKNOWN, NULL, NULL } /* must be the last entry */
65 /* look up an entry in the trp_inforec_type_table */
66 static struct trp_inforec_type_entry *get_trp_inforec_type_entry(TRP_INFOREC_TYPE msgtype)
68 struct trp_inforec_type_entry *entry=trp_inforec_type_table;
70 while ((entry->type != TRP_INFOREC_TYPE_UNKNOWN)
71 && (entry->type != msgtype)) {
77 /* translate strings to codes */
78 TRP_INFOREC_TYPE trp_inforec_type_from_string(const char *s)
80 struct trp_inforec_type_entry *entry=trp_inforec_type_table;
82 while ((entry->type != TRP_INFOREC_TYPE_UNKNOWN)
83 && (strcmp(s, entry->name)!=0)) {
88 /* translate codes to strings (do not need to be freed)
89 * Returns NULL on an unknown code */
90 const char *trp_inforec_type_to_string(TRP_INFOREC_TYPE msgtype)
92 struct trp_inforec_type_entry *entry=get_trp_inforec_type_entry(msgtype);
96 /* called by talloc when destroying an update message body */
97 static int trp_inforec_route_destructor(void *object)
99 TRP_INFOREC_ROUTE *body=talloc_get_type_abort(object, TRP_INFOREC_ROUTE);
101 /* clean up TR_NAME data, which are not managed by talloc */
102 if (body->trust_router != NULL)
103 tr_free_name(body->trust_router);
104 if (body->next_hop != NULL)
105 tr_free_name(body->next_hop);
109 static TRP_INFOREC_DATA *trp_inforec_route_new(TALLOC_CTX *mem_ctx)
111 TRP_INFOREC_DATA *new_data=talloc(mem_ctx, TRP_INFOREC_DATA);
112 TRP_INFOREC_ROUTE *new_rec=NULL;
117 new_rec=talloc(new_data, TRP_INFOREC_ROUTE);
118 if (new_rec == NULL) {
119 talloc_free(new_data);
122 new_rec->trust_router=NULL;
123 new_rec->next_hop=NULL;
124 new_rec->next_hop_port=0;
125 new_rec->metric=TRP_METRIC_INFINITY;
127 talloc_set_destructor((void *)new_rec, trp_inforec_route_destructor);
128 new_data->route=new_rec;
135 static int trp_inforec_comm_destructor(void *obj)
137 TRP_INFOREC_COMM *rec=talloc_get_type_abort(obj, TRP_INFOREC_COMM);
138 if (rec->owner_realm!=NULL)
139 tr_free_name(rec->owner_realm);
140 if (rec->owner_contact!=NULL)
141 tr_free_name(rec->owner_contact);
142 if (rec->provenance!=NULL)
143 json_decref(rec->provenance);
147 static TRP_INFOREC_DATA *trp_inforec_comm_new(TALLOC_CTX *mem_ctx)
149 TRP_INFOREC_DATA *new_data=talloc(mem_ctx, TRP_INFOREC_DATA);
150 TRP_INFOREC_COMM *new_rec=NULL;
155 new_rec=talloc(new_data, TRP_INFOREC_COMM);
157 talloc_free(new_data);
160 new_rec->comm_type=TR_COMM_UNKNOWN;
161 new_rec->role=TR_ROLE_UNKNOWN;
163 new_rec->owner_realm=NULL;
164 new_rec->owner_contact=NULL;
165 new_rec->provenance=NULL;
166 talloc_set_destructor((void *)new_rec, trp_inforec_comm_destructor);
167 new_data->comm=new_rec;
173 TRP_INFOREC *trp_inforec_get_next(TRP_INFOREC *rec)
181 static TRP_INFOREC *trp_inforec_get_tail(TRP_INFOREC *rec)
183 while ((rec->next)!=NULL)
184 rec=trp_inforec_get_next(rec);
188 void trp_inforec_set_next(TRP_INFOREC *rec, TRP_INFOREC *next_rec)
194 TRP_INFOREC_TYPE trp_inforec_get_type(TRP_INFOREC *rec)
199 return TRP_INFOREC_TYPE_UNKNOWN;
202 void trp_inforec_set_type(TRP_INFOREC *rec, TRP_INFOREC_TYPE type)
208 TR_NAME *trp_inforec_get_trust_router(TRP_INFOREC *rec)
211 case TRP_INFOREC_TYPE_ROUTE:
212 if (rec->data->route!=NULL)
213 return rec->data->route->trust_router;
221 TR_NAME *trp_inforec_dup_trust_router(TRP_INFOREC *rec)
223 return tr_dup_name(trp_inforec_get_trust_router(rec));
226 TRP_RC trp_inforec_set_trust_router(TRP_INFOREC *rec, TR_NAME *trust_router)
229 case TRP_INFOREC_TYPE_ROUTE:
230 if (rec->data->route!=NULL) {
231 rec->data->route->trust_router=trust_router;
241 /* TODO: need to return hostname/port --jlr */
242 TR_NAME *trp_inforec_get_next_hop(TRP_INFOREC *rec)
245 case TRP_INFOREC_TYPE_ROUTE:
246 if (rec->data->route!=NULL)
247 return rec->data->route->next_hop;
255 TR_NAME *trp_inforec_dup_next_hop(TRP_INFOREC *rec)
257 return tr_dup_name(trp_inforec_get_next_hop(rec));
260 TRP_RC trp_inforec_set_next_hop(TRP_INFOREC *rec, TR_NAME *next_hop)
263 case TRP_INFOREC_TYPE_ROUTE:
264 if (rec->data->route!=NULL) {
265 rec->data->route->next_hop=next_hop;
275 unsigned int trp_inforec_get_metric(TRP_INFOREC *rec)
278 case TRP_INFOREC_TYPE_ROUTE:
279 if (rec->data->route!=NULL)
280 return rec->data->route->metric;
285 return TRP_METRIC_INVALID;
288 TRP_RC trp_inforec_set_metric(TRP_INFOREC *rec, unsigned int metric)
291 case TRP_INFOREC_TYPE_ROUTE:
292 if (rec->data->route!=NULL) {
293 rec->data->route->metric=metric;
303 unsigned int trp_inforec_get_interval(TRP_INFOREC *rec)
306 case TRP_INFOREC_TYPE_ROUTE:
307 if (rec->data->route!=NULL)
308 return rec->data->route->interval;
310 case TRP_INFOREC_TYPE_COMMUNITY:
311 if (rec->data->comm!=NULL)
312 return rec->data->comm->interval;
317 return TRP_INTERVAL_INVALID;
320 TRP_RC trp_inforec_set_interval(TRP_INFOREC *rec, unsigned int interval)
323 case TRP_INFOREC_TYPE_ROUTE:
324 if (rec->data->route!=NULL) {
325 rec->data->route->interval=interval;
329 case TRP_INFOREC_TYPE_COMMUNITY:
330 if (rec->data->comm!=NULL) {
331 rec->data->comm->interval=interval;
340 TR_COMM_TYPE trp_inforec_get_comm_type(TRP_INFOREC *rec)
343 case TRP_INFOREC_TYPE_COMMUNITY:
344 if (rec->data->comm!=NULL)
345 return rec->data->comm->comm_type;
350 return TR_COMM_UNKNOWN;
353 TRP_RC trp_inforec_set_comm_type(TRP_INFOREC *rec, TR_COMM_TYPE type)
356 case TRP_INFOREC_TYPE_COMMUNITY:
357 if (rec->data->comm!=NULL) {
358 rec->data->comm->comm_type=type;
368 TR_REALM_ROLE trp_inforec_get_role(TRP_INFOREC *rec)
371 case TRP_INFOREC_TYPE_COMMUNITY:
372 if (rec->data->comm!=NULL)
373 return rec->data->comm->role;
378 return TR_ROLE_UNKNOWN;
381 TRP_RC trp_inforec_set_role(TRP_INFOREC *rec, TR_REALM_ROLE role)
384 case TRP_INFOREC_TYPE_COMMUNITY:
385 if (rec->data->comm!=NULL) {
386 rec->data->comm->role=role;
396 TR_APC *trp_inforec_get_apcs(TRP_INFOREC *rec)
399 case TRP_INFOREC_TYPE_COMMUNITY:
400 if (rec->data->comm!=NULL)
401 return rec->data->comm->apcs;
409 TRP_RC trp_inforec_set_apcs(TRP_INFOREC *rec, TR_APC *apcs)
412 case TRP_INFOREC_TYPE_COMMUNITY:
413 if (rec->data->comm!=NULL) {
414 rec->data->comm->apcs=apcs;
415 talloc_steal(rec, apcs);
425 TR_NAME *trp_inforec_get_owner_realm(TRP_INFOREC *rec)
428 case TRP_INFOREC_TYPE_COMMUNITY:
429 if (rec->data->comm!=NULL)
430 return rec->data->comm->owner_realm;
438 TRP_RC trp_inforec_set_owner_realm(TRP_INFOREC *rec, TR_NAME *name)
441 case TRP_INFOREC_TYPE_COMMUNITY:
442 if (rec->data->comm!=NULL) {
443 rec->data->comm->owner_realm=name;
453 TR_NAME *trp_inforec_get_owner_contact(TRP_INFOREC *rec)
456 case TRP_INFOREC_TYPE_COMMUNITY:
457 if (rec->data->comm!=NULL)
458 return rec->data->comm->owner_contact;
466 TRP_RC trp_inforec_set_owner_contact(TRP_INFOREC *rec, TR_NAME *name)
469 case TRP_INFOREC_TYPE_COMMUNITY:
470 if (rec->data->comm!=NULL) {
471 rec->data->comm->owner_contact=name;
481 /* caller needs to incref the output if they're going to hang on to it */
482 json_t *trp_inforec_get_provenance(TRP_INFOREC *rec)
485 case TRP_INFOREC_TYPE_COMMUNITY:
486 if (rec->data->comm!=NULL)
487 return rec->data->comm->provenance;
495 /* increments the reference count */
496 TRP_RC trp_inforec_set_provenance(TRP_INFOREC *rec, json_t *prov)
499 case TRP_INFOREC_TYPE_COMMUNITY:
500 if (rec->data->comm!=NULL) {
501 if (rec->data->comm->provenance!=NULL)
502 json_decref(rec->data->comm->provenance);
503 rec->data->comm->provenance=prov;
514 /* generic record type */
515 TRP_INFOREC *trp_inforec_new(TALLOC_CTX *mem_ctx, TRP_INFOREC_TYPE type)
517 TRP_INFOREC *new_rec=talloc(mem_ctx, TRP_INFOREC);
518 TRP_INFOREC_DATA *data=NULL;
519 struct trp_inforec_type_entry *dtype=get_trp_inforec_type_entry(type);
521 if ((new_rec != NULL) && (dtype->type != TRP_INFOREC_TYPE_UNKNOWN)) {
522 trp_inforec_set_type(new_rec, type);
523 trp_inforec_set_next(new_rec, NULL);
524 if (dtype->allocate!=NULL) {
525 data=dtype->allocate(new_rec);
529 talloc_free(new_rec);
537 void trp_inforec_free(TRP_INFOREC *rec)
543 static int trp_upd_destructor(void *object)
545 TRP_UPD *upd=talloc_get_type_abort(object, TRP_UPD);
546 if (upd->realm!=NULL)
547 tr_free_name(upd->realm);
549 tr_free_name(upd->comm);
551 tr_free_name(upd->peer);
555 TRP_UPD *trp_upd_new(TALLOC_CTX *mem_ctx)
557 TRP_UPD *new_body=talloc(mem_ctx, TRP_UPD);
559 if (new_body!=NULL) {
560 new_body->realm=NULL;
562 new_body->records=NULL;
564 talloc_set_destructor((void *)new_body, trp_upd_destructor);
569 void trp_upd_free(TRP_UPD *update)
575 TRP_INFOREC *trp_upd_get_inforec(TRP_UPD *upd)
583 void trp_upd_set_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
589 void trp_upd_add_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
591 tr_debug("trp_upd_add_inforec: adding record.");
592 if (upd->records==NULL)
595 trp_inforec_set_next(trp_inforec_get_tail(upd->records), rec);
596 talloc_steal(upd, rec);
599 TR_NAME *trp_upd_get_realm(TRP_UPD *upd)
604 TR_NAME *trp_upd_dup_realm(TRP_UPD *upd)
606 return tr_dup_name(upd->realm);
609 void trp_upd_set_realm(TRP_UPD *upd, TR_NAME *realm)
611 if (upd->realm!=NULL)
612 tr_free_name(upd->realm);
616 TR_NAME *trp_upd_get_comm(TRP_UPD *upd)
621 TR_NAME *trp_upd_dup_comm(TRP_UPD *upd)
623 return tr_dup_name(upd->comm);
626 void trp_upd_set_comm(TRP_UPD *upd, TR_NAME *comm)
629 tr_free_name(upd->comm);
633 TR_NAME *trp_upd_get_peer(TRP_UPD *upd)
638 TR_NAME *trp_upd_dup_peer(TRP_UPD *upd)
640 return tr_dup_name(upd->peer);
643 void trp_upd_set_peer(TRP_UPD *upd, TR_NAME *peer)
648 void trp_upd_set_next_hop(TRP_UPD *upd, const char *hostname, unsigned int port)
650 TRP_INFOREC *rec=NULL;
653 for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
654 if (trp_inforec_set_next_hop(rec, cpy=tr_new_name(hostname)) != TRP_SUCCESS) {
655 tr_err("trp_upd_set_peer: error setting peer.");
662 static void trp_inforec_route_print(TRP_INFOREC_DATA *data)
664 if (data->route!=NULL) {
665 printf(" trust_router=%.*s\n metric=%d\n interval=%d]\n",
666 data->route->trust_router->len, data->route->trust_router->buf,
667 data->route->metric, data->route->interval);
671 static void trp_inforec_comm_print(TRP_INFOREC_DATA *data)
673 if (data->comm!=NULL) {
674 printf(" type=%s\n role=%s\n owner=%.*s\n contact=%.*s]\n",
675 tr_comm_type_to_str(data->comm->comm_type),
676 tr_realm_role_to_str(data->comm->role),
677 data->comm->owner_realm->len, data->comm->owner_realm->buf,
678 data->comm->owner_contact->len, data->comm->owner_contact->buf);
679 /* TODO: print apcs */