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.
39 #include <tr_name_internal.h>
40 #include <trp_internal.h>
46 /* static prototypes */
47 static TRP_INFOREC_DATA *trp_inforec_route_new(TALLOC_CTX *mem_ctx);
48 static void trp_inforec_route_print(TRP_INFOREC_DATA *);
49 static TRP_INFOREC_DATA *trp_inforec_comm_new(TALLOC_CTX *mem_ctx);
50 static void trp_inforec_comm_print(TRP_INFOREC_DATA *);
53 struct trp_inforec_type_entry {
55 TRP_INFOREC_TYPE type;
56 TRP_INFOREC_DATA *(*allocate)(TALLOC_CTX *);
57 void (*print)(TRP_INFOREC_DATA *);
59 static struct trp_inforec_type_entry trp_inforec_type_table[] = {
60 { "route", TRP_INFOREC_TYPE_ROUTE, trp_inforec_route_new, trp_inforec_route_print },
61 { "comm", TRP_INFOREC_TYPE_COMMUNITY, trp_inforec_comm_new, trp_inforec_comm_print },
62 { NULL, TRP_INFOREC_TYPE_UNKNOWN, NULL, NULL } /* must be the last entry */
66 /* look up an entry in the trp_inforec_type_table */
67 static struct trp_inforec_type_entry *get_trp_inforec_type_entry(TRP_INFOREC_TYPE msgtype)
69 struct trp_inforec_type_entry *entry=trp_inforec_type_table;
71 while ((entry->type != TRP_INFOREC_TYPE_UNKNOWN)
72 && (entry->type != msgtype)) {
78 /* translate strings to codes */
79 TRP_INFOREC_TYPE trp_inforec_type_from_string(const char *s)
81 struct trp_inforec_type_entry *entry=trp_inforec_type_table;
83 while ((entry->type != TRP_INFOREC_TYPE_UNKNOWN)
84 && (strcasecmp(s, entry->name)!=0)) {
89 /* translate codes to strings (do not need to be freed)
90 * Returns NULL on an unknown code */
91 const char *trp_inforec_type_to_string(TRP_INFOREC_TYPE msgtype)
93 struct trp_inforec_type_entry *entry=get_trp_inforec_type_entry(msgtype);
97 /* called by talloc when destroying an update message body */
98 static int trp_inforec_route_destructor(void *object)
100 TRP_INFOREC_ROUTE *body=talloc_get_type_abort(object, TRP_INFOREC_ROUTE);
102 /* clean up TR_NAME data, which are not managed by talloc */
103 if (body->trust_router != NULL)
104 tr_free_name(body->trust_router);
105 if (body->next_hop != NULL)
106 tr_free_name(body->next_hop);
110 static TRP_INFOREC_DATA *trp_inforec_route_new(TALLOC_CTX *mem_ctx)
112 TRP_INFOREC_DATA *new_data=talloc(mem_ctx, TRP_INFOREC_DATA);
113 TRP_INFOREC_ROUTE *new_rec=NULL;
118 new_rec=talloc(new_data, TRP_INFOREC_ROUTE);
119 if (new_rec == NULL) {
120 talloc_free(new_data);
123 new_rec->trust_router=NULL;
124 new_rec->next_hop=NULL;
125 new_rec->next_hop_port=0;
126 new_rec->metric=TRP_METRIC_INFINITY;
128 talloc_set_destructor((void *)new_rec, trp_inforec_route_destructor);
129 new_data->route=new_rec;
136 static int trp_inforec_comm_destructor(void *obj)
138 TRP_INFOREC_COMM *rec=talloc_get_type_abort(obj, TRP_INFOREC_COMM);
139 if (rec->owner_realm!=NULL)
140 tr_free_name(rec->owner_realm);
141 if (rec->owner_contact!=NULL)
142 tr_free_name(rec->owner_contact);
143 if (rec->provenance!=NULL)
144 json_decref(rec->provenance);
148 static TRP_INFOREC_DATA *trp_inforec_comm_new(TALLOC_CTX *mem_ctx)
150 TRP_INFOREC_DATA *new_data=talloc(mem_ctx, TRP_INFOREC_DATA);
151 TRP_INFOREC_COMM *new_rec=NULL;
156 new_rec=talloc(new_data, TRP_INFOREC_COMM);
158 talloc_free(new_data);
161 new_rec->comm_type=TR_COMM_UNKNOWN;
162 new_rec->role=TR_ROLE_UNKNOWN;
164 new_rec->owner_realm=NULL;
165 new_rec->owner_contact=NULL;
166 new_rec->expiration_interval=0;
167 new_rec->provenance=NULL;
169 talloc_set_destructor((void *)new_rec, trp_inforec_comm_destructor);
170 new_data->comm=new_rec;
176 TRP_INFOREC *trp_inforec_get_next(TRP_INFOREC *rec)
184 static TRP_INFOREC *trp_inforec_get_tail(TRP_INFOREC *rec)
186 while ((rec->next)!=NULL)
187 rec=trp_inforec_get_next(rec);
191 void trp_inforec_set_next(TRP_INFOREC *rec, TRP_INFOREC *next_rec)
197 TRP_INFOREC_TYPE trp_inforec_get_type(TRP_INFOREC *rec)
202 return TRP_INFOREC_TYPE_UNKNOWN;
205 void trp_inforec_set_type(TRP_INFOREC *rec, TRP_INFOREC_TYPE type)
211 TR_NAME *trp_inforec_get_trust_router(TRP_INFOREC *rec)
214 case TRP_INFOREC_TYPE_ROUTE:
215 if (rec->data->route!=NULL)
216 return rec->data->route->trust_router;
224 TR_NAME *trp_inforec_dup_trust_router(TRP_INFOREC *rec)
226 return tr_dup_name(trp_inforec_get_trust_router(rec));
229 TRP_RC trp_inforec_set_trust_router(TRP_INFOREC *rec, TR_NAME *trust_router)
232 case TRP_INFOREC_TYPE_ROUTE:
233 if (rec->data->route!=NULL) {
234 rec->data->route->trust_router=trust_router;
244 /* TODO: need to return hostname/port --jlr */
245 TR_NAME *trp_inforec_get_next_hop(TRP_INFOREC *rec)
248 case TRP_INFOREC_TYPE_ROUTE:
249 if (rec->data->route!=NULL)
250 return rec->data->route->next_hop;
258 TR_NAME *trp_inforec_dup_next_hop(TRP_INFOREC *rec)
260 return tr_dup_name(trp_inforec_get_next_hop(rec));
263 TRP_RC trp_inforec_set_next_hop(TRP_INFOREC *rec, TR_NAME *next_hop)
266 case TRP_INFOREC_TYPE_ROUTE:
267 if (rec->data->route==NULL)
269 rec->data->route->next_hop=next_hop;
271 case TRP_INFOREC_TYPE_COMMUNITY:
272 /* next hop not used for community records */
280 unsigned int trp_inforec_get_metric(TRP_INFOREC *rec)
283 case TRP_INFOREC_TYPE_ROUTE:
284 if (rec->data->route!=NULL)
285 return rec->data->route->metric;
290 return TRP_METRIC_INVALID;
293 TRP_RC trp_inforec_set_metric(TRP_INFOREC *rec, unsigned int metric)
296 case TRP_INFOREC_TYPE_ROUTE:
297 if (rec->data->route!=NULL) {
298 rec->data->route->metric=metric;
308 unsigned int trp_inforec_get_interval(TRP_INFOREC *rec)
311 case TRP_INFOREC_TYPE_ROUTE:
312 if (rec->data->route!=NULL)
313 return rec->data->route->interval;
315 case TRP_INFOREC_TYPE_COMMUNITY:
316 if (rec->data->comm!=NULL)
317 return rec->data->comm->interval;
322 return TRP_INTERVAL_INVALID;
325 TRP_RC trp_inforec_set_interval(TRP_INFOREC *rec, unsigned int interval)
328 case TRP_INFOREC_TYPE_ROUTE:
329 if (rec->data->route!=NULL) {
330 rec->data->route->interval=interval;
334 case TRP_INFOREC_TYPE_COMMUNITY:
335 if (rec->data->comm!=NULL) {
336 rec->data->comm->interval=interval;
345 time_t trp_inforec_get_exp_interval(TRP_INFOREC *rec)
348 case TRP_INFOREC_TYPE_COMMUNITY:
349 if (rec->data->comm!=NULL)
350 return rec->data->comm->expiration_interval;
358 TRP_RC trp_inforec_set_exp_interval(TRP_INFOREC *rec, time_t expint)
361 case TRP_INFOREC_TYPE_COMMUNITY:
362 if (rec->data->comm!=NULL) {
363 rec->data->comm->expiration_interval=expint;
373 TR_COMM_TYPE trp_inforec_get_comm_type(TRP_INFOREC *rec)
376 case TRP_INFOREC_TYPE_COMMUNITY:
377 if (rec->data->comm!=NULL)
378 return rec->data->comm->comm_type;
383 return TR_COMM_UNKNOWN;
386 TRP_RC trp_inforec_set_comm_type(TRP_INFOREC *rec, TR_COMM_TYPE type)
389 case TRP_INFOREC_TYPE_COMMUNITY:
390 if (rec->data->comm!=NULL) {
391 rec->data->comm->comm_type=type;
401 TR_REALM_ROLE trp_inforec_get_role(TRP_INFOREC *rec)
404 case TRP_INFOREC_TYPE_COMMUNITY:
405 if (rec->data->comm!=NULL)
406 return rec->data->comm->role;
411 return TR_ROLE_UNKNOWN;
414 TRP_RC trp_inforec_set_role(TRP_INFOREC *rec, TR_REALM_ROLE role)
417 case TRP_INFOREC_TYPE_COMMUNITY:
418 if (rec->data->comm!=NULL) {
419 rec->data->comm->role=role;
429 TR_APC *trp_inforec_get_apcs(TRP_INFOREC *rec)
432 case TRP_INFOREC_TYPE_COMMUNITY:
433 if (rec->data->comm!=NULL)
434 return rec->data->comm->apcs;
442 TRP_RC trp_inforec_set_apcs(TRP_INFOREC *rec, TR_APC *apcs)
445 case TRP_INFOREC_TYPE_COMMUNITY:
446 if (rec->data->comm!=NULL) {
447 rec->data->comm->apcs=apcs;
448 talloc_steal(rec, apcs);
459 TR_NAME *trp_inforec_get_owner_realm(TRP_INFOREC *rec)
462 case TRP_INFOREC_TYPE_COMMUNITY:
463 if (rec->data->comm!=NULL)
464 return rec->data->comm->owner_realm;
472 TRP_RC trp_inforec_set_owner_realm(TRP_INFOREC *rec, TR_NAME *name)
475 case TRP_INFOREC_TYPE_COMMUNITY:
476 if (rec->data->comm!=NULL) {
477 rec->data->comm->owner_realm=name;
487 TR_NAME *trp_inforec_get_owner_contact(TRP_INFOREC *rec)
490 case TRP_INFOREC_TYPE_COMMUNITY:
491 if (rec->data->comm!=NULL)
492 return rec->data->comm->owner_contact;
500 TRP_RC trp_inforec_set_owner_contact(TRP_INFOREC *rec, TR_NAME *name)
503 case TRP_INFOREC_TYPE_COMMUNITY:
504 if (rec->data->comm!=NULL) {
505 rec->data->comm->owner_contact=name;
515 /* caller needs to incref the output if they're going to hang on to it */
516 json_t *trp_inforec_get_provenance(TRP_INFOREC *rec)
519 case TRP_INFOREC_TYPE_COMMUNITY:
520 if (rec->data->comm!=NULL)
521 return rec->data->comm->provenance;
529 /* increments the reference count */
530 TRP_RC trp_inforec_set_provenance(TRP_INFOREC *rec, json_t *prov)
533 case TRP_INFOREC_TYPE_COMMUNITY:
534 if (rec->data->comm!=NULL) {
535 if (rec->data->comm->provenance!=NULL)
536 json_decref(rec->data->comm->provenance);
537 rec->data->comm->provenance=prov;
548 static TRP_RC trp_inforec_add_to_provenance(TRP_INFOREC *rec, TR_NAME *name)
553 case TRP_INFOREC_TYPE_ROUTE:
554 /* no provenance list */
556 case TRP_INFOREC_TYPE_COMMUNITY:
557 jname=tr_name_to_json_string(name);
560 if (rec->data->comm->provenance==NULL) {
561 rec->data->comm->provenance=json_array();
562 if (rec->data->comm->provenance==NULL) {
567 if (0!=json_array_append_new(rec->data->comm->provenance, jname)) {
578 TR_NAME *trp_inforec_dup_origin(TRP_INFOREC *rec)
580 TR_NAME *origin=NULL;
581 json_t *prov=trp_inforec_get_provenance(rec);
587 s=json_string_value(json_array_get(prov, 0));
589 tr_debug("trp_inforec_dup_origin: empty origin in provenance list.");
592 origin=tr_new_name(s);
596 /* generic record type */
597 TRP_INFOREC *trp_inforec_new(TALLOC_CTX *mem_ctx, TRP_INFOREC_TYPE type)
599 TRP_INFOREC *new_rec=talloc(mem_ctx, TRP_INFOREC);
600 TRP_INFOREC_DATA *data=NULL;
601 struct trp_inforec_type_entry *dtype=get_trp_inforec_type_entry(type);
603 if ((new_rec != NULL) && (dtype->type != TRP_INFOREC_TYPE_UNKNOWN)) {
604 trp_inforec_set_type(new_rec, type);
605 trp_inforec_set_next(new_rec, NULL);
606 if (dtype->allocate!=NULL) {
607 data=dtype->allocate(new_rec);
611 talloc_free(new_rec);
619 void trp_inforec_free(TRP_INFOREC *rec)
625 static int trp_upd_destructor(void *object)
627 TRP_UPD *upd=talloc_get_type_abort(object, TRP_UPD);
628 if (upd->realm!=NULL)
629 tr_free_name(upd->realm);
631 tr_free_name(upd->comm);
633 tr_free_name(upd->peer);
637 TRP_UPD *trp_upd_new(TALLOC_CTX *mem_ctx)
639 TRP_UPD *new_body=talloc(mem_ctx, TRP_UPD);
641 if (new_body!=NULL) {
642 new_body->realm=NULL;
644 new_body->records=NULL;
646 talloc_set_destructor((void *)new_body, trp_upd_destructor);
651 void trp_upd_free(TRP_UPD *update)
657 TRP_INFOREC *trp_upd_get_inforec(TRP_UPD *upd)
665 void trp_upd_set_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
671 void trp_upd_add_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
673 tr_debug("trp_upd_add_inforec: adding record.");
674 if (upd->records==NULL)
677 trp_inforec_set_next(trp_inforec_get_tail(upd->records), rec);
678 talloc_steal(upd, rec);
682 * Removes and frees the selected inforec.
684 * @param upd Update to remove from
685 * @param rec Inforec to remove
687 void trp_upd_remove_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
689 TRP_INFOREC *this=upd->records;
691 /* special case for the first element */
693 upd->records=upd->records->next;
694 trp_inforec_free(this);
698 while (this->next!=NULL) {
699 if (this->next==rec) {
700 this->next=this->next->next; /* this->next is not null */
701 trp_inforec_free(rec);
707 size_t trp_upd_num_inforecs(TRP_UPD *upd)
710 TRP_INFOREC *this=upd->records;
712 while (this != NULL) {
720 TR_NAME *trp_upd_get_realm(TRP_UPD *upd)
725 TR_NAME *trp_upd_dup_realm(TRP_UPD *upd)
727 return tr_dup_name(upd->realm);
730 void trp_upd_set_realm(TRP_UPD *upd, TR_NAME *realm)
732 if (upd->realm!=NULL)
733 tr_free_name(upd->realm);
737 TR_NAME *trp_upd_get_comm(TRP_UPD *upd)
742 TR_NAME *trp_upd_dup_comm(TRP_UPD *upd)
744 return tr_dup_name(upd->comm);
747 void trp_upd_set_comm(TRP_UPD *upd, TR_NAME *comm)
750 tr_free_name(upd->comm);
754 TR_NAME *trp_upd_get_peer(TRP_UPD *upd)
759 TR_NAME *trp_upd_dup_peer(TRP_UPD *upd)
761 return tr_dup_name(upd->peer);
764 void trp_upd_set_peer(TRP_UPD *upd, TR_NAME *peer)
769 void trp_upd_set_next_hop(TRP_UPD *upd, const char *hostname, unsigned int port)
771 TRP_INFOREC *rec=NULL;
774 for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
775 if (trp_inforec_set_next_hop(rec, cpy=tr_new_name(hostname)) != TRP_SUCCESS) {
776 tr_err("trp_upd_set_next_hop: error setting next hop.");
782 void trp_upd_add_to_provenance(TRP_UPD *upd, TR_NAME *name)
784 TRP_INFOREC *rec=NULL;
786 /* add it to all inforecs */
787 for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
788 if (TRP_SUCCESS!=trp_inforec_add_to_provenance(rec, name))
789 tr_err("trp_upd_set_peer: error adding peer to provenance list.");
794 static void trp_inforec_route_print(TRP_INFOREC_DATA *data)
796 if (data->route!=NULL) {
797 tr_info(" trust_router=%.*s\n metric=%d\n interval=%d]\n",
798 data->route->trust_router->len, data->route->trust_router->buf,
799 data->route->metric, data->route->interval);
803 static void trp_inforec_comm_print(TRP_INFOREC_DATA *data)
805 if (data->comm!=NULL) {
806 tr_info(" type=%s\n role=%s\n owner=%.*s\n contact=%.*s]\n",
807 tr_comm_type_to_str(data->comm->comm_type),
808 tr_realm_role_to_str(data->comm->role),
809 data->comm->owner_realm->len, data->comm->owner_realm->buf,
810 data->comm->owner_contact->len, data->comm->owner_contact->buf);
811 /* TODO: print apcs */