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));
264 * Set the next hop for the inforec
266 * Returns TRP_SUCCESS if it set the next hop value for the inforec.
267 * Returns TRP_UNSUPPORTED if the inforec does not have a next hop record but
268 * otherwise nothing went wrong.
269 * Returns TRP_ERROR or another error if there was a failure.
273 * @return TRP_SUCCESS if the value was set, TRP_UNSUPPORTED if the inforec does not support next hop, or an error code on failure
275 TRP_RC trp_inforec_set_next_hop(TRP_INFOREC *rec, TR_NAME *next_hop)
277 /* Any inforec types that support next_hop should set it here. */
279 case TRP_INFOREC_TYPE_ROUTE:
280 if (rec->data->route==NULL)
282 rec->data->route->next_hop=next_hop;
286 /* next hop not used for other records */
287 return TRP_UNSUPPORTED;
292 unsigned int trp_inforec_get_metric(TRP_INFOREC *rec)
295 case TRP_INFOREC_TYPE_ROUTE:
296 if (rec->data->route!=NULL)
297 return rec->data->route->metric;
302 return TRP_METRIC_INVALID;
305 TRP_RC trp_inforec_set_metric(TRP_INFOREC *rec, unsigned int metric)
308 case TRP_INFOREC_TYPE_ROUTE:
309 if (rec->data->route!=NULL) {
310 rec->data->route->metric=metric;
320 unsigned int trp_inforec_get_interval(TRP_INFOREC *rec)
323 case TRP_INFOREC_TYPE_ROUTE:
324 if (rec->data->route!=NULL)
325 return rec->data->route->interval;
327 case TRP_INFOREC_TYPE_COMMUNITY:
328 if (rec->data->comm!=NULL)
329 return rec->data->comm->interval;
334 return TRP_INTERVAL_INVALID;
337 TRP_RC trp_inforec_set_interval(TRP_INFOREC *rec, unsigned int interval)
340 case TRP_INFOREC_TYPE_ROUTE:
341 if (rec->data->route!=NULL) {
342 rec->data->route->interval=interval;
346 case TRP_INFOREC_TYPE_COMMUNITY:
347 if (rec->data->comm!=NULL) {
348 rec->data->comm->interval=interval;
357 time_t trp_inforec_get_exp_interval(TRP_INFOREC *rec)
360 case TRP_INFOREC_TYPE_COMMUNITY:
361 if (rec->data->comm!=NULL)
362 return rec->data->comm->expiration_interval;
370 TRP_RC trp_inforec_set_exp_interval(TRP_INFOREC *rec, time_t expint)
373 case TRP_INFOREC_TYPE_COMMUNITY:
374 if (rec->data->comm!=NULL) {
375 rec->data->comm->expiration_interval=expint;
385 TR_COMM_TYPE trp_inforec_get_comm_type(TRP_INFOREC *rec)
388 case TRP_INFOREC_TYPE_COMMUNITY:
389 if (rec->data->comm!=NULL)
390 return rec->data->comm->comm_type;
395 return TR_COMM_UNKNOWN;
398 TRP_RC trp_inforec_set_comm_type(TRP_INFOREC *rec, TR_COMM_TYPE type)
401 case TRP_INFOREC_TYPE_COMMUNITY:
402 if (rec->data->comm!=NULL) {
403 rec->data->comm->comm_type=type;
413 TR_REALM_ROLE trp_inforec_get_role(TRP_INFOREC *rec)
416 case TRP_INFOREC_TYPE_COMMUNITY:
417 if (rec->data->comm!=NULL)
418 return rec->data->comm->role;
423 return TR_ROLE_UNKNOWN;
426 TRP_RC trp_inforec_set_role(TRP_INFOREC *rec, TR_REALM_ROLE role)
429 case TRP_INFOREC_TYPE_COMMUNITY:
430 if (rec->data->comm!=NULL) {
431 rec->data->comm->role=role;
441 TR_APC *trp_inforec_get_apcs(TRP_INFOREC *rec)
444 case TRP_INFOREC_TYPE_COMMUNITY:
445 if (rec->data->comm!=NULL)
446 return rec->data->comm->apcs;
454 TRP_RC trp_inforec_set_apcs(TRP_INFOREC *rec, TR_APC *apcs)
457 case TRP_INFOREC_TYPE_COMMUNITY:
458 if (rec->data->comm!=NULL) {
459 rec->data->comm->apcs=apcs;
460 talloc_steal(rec, apcs);
471 TR_NAME *trp_inforec_get_owner_realm(TRP_INFOREC *rec)
474 case TRP_INFOREC_TYPE_COMMUNITY:
475 if (rec->data->comm!=NULL)
476 return rec->data->comm->owner_realm;
484 TRP_RC trp_inforec_set_owner_realm(TRP_INFOREC *rec, TR_NAME *name)
487 case TRP_INFOREC_TYPE_COMMUNITY:
488 if (rec->data->comm!=NULL) {
489 rec->data->comm->owner_realm=name;
499 TR_NAME *trp_inforec_get_owner_contact(TRP_INFOREC *rec)
502 case TRP_INFOREC_TYPE_COMMUNITY:
503 if (rec->data->comm!=NULL)
504 return rec->data->comm->owner_contact;
512 TRP_RC trp_inforec_set_owner_contact(TRP_INFOREC *rec, TR_NAME *name)
515 case TRP_INFOREC_TYPE_COMMUNITY:
516 if (rec->data->comm!=NULL) {
517 rec->data->comm->owner_contact=name;
527 /* caller needs to incref the output if they're going to hang on to it */
528 json_t *trp_inforec_get_provenance(TRP_INFOREC *rec)
531 case TRP_INFOREC_TYPE_COMMUNITY:
532 if (rec->data->comm!=NULL)
533 return rec->data->comm->provenance;
541 /* increments the reference count */
542 TRP_RC trp_inforec_set_provenance(TRP_INFOREC *rec, json_t *prov)
545 case TRP_INFOREC_TYPE_COMMUNITY:
546 if (rec->data->comm!=NULL) {
547 if (rec->data->comm->provenance!=NULL)
548 json_decref(rec->data->comm->provenance);
549 rec->data->comm->provenance=prov;
560 static TRP_RC trp_inforec_add_to_provenance(TRP_INFOREC *rec, TR_NAME *name)
565 case TRP_INFOREC_TYPE_ROUTE:
566 /* no provenance list */
568 case TRP_INFOREC_TYPE_COMMUNITY:
569 jname=tr_name_to_json_string(name);
572 if (rec->data->comm->provenance==NULL) {
573 rec->data->comm->provenance=json_array();
574 if (rec->data->comm->provenance==NULL) {
579 if (0!=json_array_append_new(rec->data->comm->provenance, jname)) {
590 TR_NAME *trp_inforec_dup_origin(TRP_INFOREC *rec)
592 TR_NAME *origin=NULL;
593 json_t *prov=trp_inforec_get_provenance(rec);
599 s=json_string_value(json_array_get(prov, 0));
601 tr_debug("trp_inforec_dup_origin: empty origin in provenance list.");
604 origin=tr_new_name(s);
608 /* generic record type */
609 TRP_INFOREC *trp_inforec_new(TALLOC_CTX *mem_ctx, TRP_INFOREC_TYPE type)
611 TRP_INFOREC *new_rec=talloc(mem_ctx, TRP_INFOREC);
612 TRP_INFOREC_DATA *data=NULL;
613 struct trp_inforec_type_entry *dtype=get_trp_inforec_type_entry(type);
615 if ((new_rec != NULL) && (dtype->type != TRP_INFOREC_TYPE_UNKNOWN)) {
616 trp_inforec_set_type(new_rec, type);
617 trp_inforec_set_next(new_rec, NULL);
618 if (dtype->allocate!=NULL) {
619 data=dtype->allocate(new_rec);
623 talloc_free(new_rec);
631 void trp_inforec_free(TRP_INFOREC *rec)
637 static int trp_upd_destructor(void *object)
639 TRP_UPD *upd=talloc_get_type_abort(object, TRP_UPD);
640 if (upd->realm!=NULL)
641 tr_free_name(upd->realm);
643 tr_free_name(upd->comm);
645 tr_free_name(upd->peer);
649 TRP_UPD *trp_upd_new(TALLOC_CTX *mem_ctx)
651 TRP_UPD *new_body=talloc(mem_ctx, TRP_UPD);
653 if (new_body!=NULL) {
654 new_body->realm=NULL;
656 new_body->records=NULL;
658 talloc_set_destructor((void *)new_body, trp_upd_destructor);
663 void trp_upd_free(TRP_UPD *update)
669 TRP_INFOREC *trp_upd_get_inforec(TRP_UPD *upd)
677 void trp_upd_set_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
683 void trp_upd_add_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
685 tr_debug("trp_upd_add_inforec: adding record.");
686 if (upd->records==NULL)
689 trp_inforec_set_next(trp_inforec_get_tail(upd->records), rec);
690 talloc_steal(upd, rec);
694 * Removes and frees the selected inforec.
696 * @param upd Update to remove from
697 * @param rec Inforec to remove
699 void trp_upd_remove_inforec(TRP_UPD *upd, TRP_INFOREC *rec)
701 TRP_INFOREC *this=upd->records;
703 /* special case for the first element */
705 upd->records=upd->records->next;
706 trp_inforec_free(this);
710 while (this->next!=NULL) {
711 if (this->next==rec) {
712 this->next=this->next->next; /* this->next is not null */
713 trp_inforec_free(rec);
719 size_t trp_upd_num_inforecs(TRP_UPD *upd)
722 TRP_INFOREC *this=upd->records;
724 while (this != NULL) {
732 TR_NAME *trp_upd_get_realm(TRP_UPD *upd)
737 TR_NAME *trp_upd_dup_realm(TRP_UPD *upd)
739 return tr_dup_name(upd->realm);
742 void trp_upd_set_realm(TRP_UPD *upd, TR_NAME *realm)
744 if (upd->realm!=NULL)
745 tr_free_name(upd->realm);
749 TR_NAME *trp_upd_get_comm(TRP_UPD *upd)
754 TR_NAME *trp_upd_dup_comm(TRP_UPD *upd)
756 return tr_dup_name(upd->comm);
759 void trp_upd_set_comm(TRP_UPD *upd, TR_NAME *comm)
762 tr_free_name(upd->comm);
766 TR_NAME *trp_upd_get_peer(TRP_UPD *upd)
771 TR_NAME *trp_upd_dup_peer(TRP_UPD *upd)
773 return tr_dup_name(upd->peer);
776 void trp_upd_set_peer(TRP_UPD *upd, TR_NAME *peer)
781 void trp_upd_set_next_hop(TRP_UPD *upd, const char *hostname, unsigned int port)
783 TRP_INFOREC *rec=NULL;
786 for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
787 switch (trp_inforec_set_next_hop(rec, cpy=tr_new_name(hostname))) {
789 /* Success, the TR_NAME in cpy is now stored with the inforec */
792 case TRP_UNSUPPORTED:
793 /* No error, but the inforec does not accept a next_hop. Free our copy. */
798 tr_err("trp_upd_set_next_hop: error setting next hop.");
805 void trp_upd_add_to_provenance(TRP_UPD *upd, TR_NAME *name)
807 TRP_INFOREC *rec=NULL;
809 /* add it to all inforecs */
810 for (rec=trp_upd_get_inforec(upd); rec!=NULL; rec=trp_inforec_get_next(rec)) {
811 if (TRP_SUCCESS!=trp_inforec_add_to_provenance(rec, name))
812 tr_err("trp_upd_set_peer: error adding peer to provenance list.");
817 static void trp_inforec_route_print(TRP_INFOREC_DATA *data)
819 if (data->route!=NULL) {
820 tr_info(" trust_router=%.*s\n metric=%d\n interval=%d]\n",
821 data->route->trust_router->len, data->route->trust_router->buf,
822 data->route->metric, data->route->interval);
826 static void trp_inforec_comm_print(TRP_INFOREC_DATA *data)
828 if (data->comm!=NULL) {
829 tr_info(" type=%s\n role=%s\n owner=%.*s\n contact=%.*s]\n",
830 tr_comm_type_to_str(data->comm->comm_type),
831 tr_realm_role_to_str(data->comm->role),
832 data->comm->owner_realm->len, data->comm->owner_realm->buf,
833 data->comm->owner_contact->len, data->comm->owner_contact->buf);
834 /* TODO: print apcs */