X-Git-Url: http://www.project-moonshot.org/gitweb/?a=blobdiff_plain;f=common%2Ftr_msg.c;h=6f8610881a56d77e0a32dce8fb9667a022c1321c;hb=6491f2a660c3183b210b968a939eaa18f1b44a09;hp=f771fb83762fa12fadc456cd2ec439f0d40b89da;hpb=3f2f7c0b61b33e22ca937809e5a80e68c7490609;p=trust_router.git diff --git a/common/tr_msg.c b/common/tr_msg.c index f771fb8..6f86108 100644 --- a/common/tr_msg.c +++ b/common/tr_msg.c @@ -36,17 +36,62 @@ #include #include #include +#include #include #include #include +#include +#include +#include +#include #include -#include -#include +#include #include +#include #include +/* JSON helpers */ +/* Read attribute attr from msg as an integer. Returns nonzero on error. */ +static int tr_msg_get_json_integer(json_t *jmsg, const char *attr, int *dest) +{ + json_t *obj; + + obj=json_object_get(jmsg, attr); + if (obj == NULL) { + return -1; + } + /* check type */ + if (!json_is_integer(obj)) { + return -1; + } + + (*dest)=json_integer_value(obj); + return 0; +} + +/* Read attribute attr from msg as a string. Copies string into mem_ctx context so jmsg can + * be destroyed safely. Returns nonzero on error. */ +static TRP_RC tr_msg_get_json_string(json_t *jmsg, const char *attr, char **dest, TALLOC_CTX *mem_ctx) +{ + json_t *obj; + + obj=json_object_get(jmsg, attr); + if (obj == NULL) + return TRP_ERROR; + + /* check type */ + if (!json_is_string(obj)) + return TRP_ERROR; + + *dest=talloc_strdup(mem_ctx, json_string_value(obj)); + if (*dest==NULL) + return TRP_ERROR; + + return TRP_SUCCESS; +} + enum msg_type tr_msg_get_msg_type(TR_MSG *msg) { return msg->msg_type; @@ -83,23 +128,80 @@ void tr_msg_set_resp(TR_MSG *msg, TID_RESP *resp) msg->msg_type = TID_RESPONSE; } +MON_REQ *tr_msg_get_mon_req(TR_MSG *msg) +{ + if (msg->msg_type == MON_REQUEST) + return (MON_REQ *)msg->msg_rep; + return NULL; +} + +void tr_msg_set_mon_req(TR_MSG *msg, MON_REQ *req) +{ + msg->msg_rep = req; + msg->msg_type = MON_REQUEST; +} + +MON_RESP *tr_msg_get_mon_resp(TR_MSG *msg) +{ + if (msg->msg_type == MON_RESPONSE) + return (MON_RESP *)msg->msg_rep; + return NULL; +} + +void tr_msg_set_mon_resp(TR_MSG *msg, MON_RESP *resp) +{ + msg->msg_rep = resp; + msg->msg_type = MON_RESPONSE; +} + +TRP_UPD *tr_msg_get_trp_upd(TR_MSG *msg) +{ + if (msg->msg_type == TRP_UPDATE) + return (TRP_UPD *)msg->msg_rep; + return NULL; +} + +void tr_msg_set_trp_upd(TR_MSG *msg, TRP_UPD *update) +{ + msg->msg_rep=update; + talloc_steal(NULL, update); /* should attach to msg, but TR_MSG not usually talloc'ed */ + msg->msg_type=TRP_UPDATE; +} + +TRP_REQ *tr_msg_get_trp_req(TR_MSG *msg) +{ + if (msg->msg_type == TRP_REQUEST) + return (TRP_REQ *)msg->msg_rep; + return NULL; +} + +void tr_msg_set_trp_req(TR_MSG *msg, TRP_REQ *req) +{ + msg->msg_rep=req; + msg->msg_type=TRP_REQUEST; +} + static json_t *tr_msg_encode_dh(DH *dh) { json_t *jdh = NULL; json_t *jbn = NULL; + char *s=NULL; if ((!dh) || (!dh->p) || (!dh->g) || (!dh->pub_key)) return NULL; jdh = json_object(); - jbn = json_string(BN_bn2hex(dh->p)); + jbn = json_string(s=BN_bn2hex(dh->p)); + OPENSSL_free(s); json_object_set_new(jdh, "dh_p", jbn); - jbn = json_string(BN_bn2hex(dh->g)); + jbn = json_string(s=BN_bn2hex(dh->g)); + OPENSSL_free(s); json_object_set_new(jdh, "dh_g", jbn); - jbn = json_string(BN_bn2hex(dh->pub_key)); + jbn = json_string(s=BN_bn2hex(dh->pub_key)); + OPENSSL_free(s); json_object_set_new(jdh, "dh_pub_key", jbn); return jdh; @@ -112,19 +214,17 @@ static DH *tr_msg_decode_dh(json_t *jdh) json_t *jg = NULL; json_t *jpub_key = NULL; - if (!(dh = malloc(sizeof(DH)))) { + if (!(dh=tr_dh_new())) { tr_crit("tr_msg_decode_dh(): Error allocating DH structure."); return NULL; } - memset(dh, 0, sizeof(DH)); - /* store required fields from dh object */ if ((NULL == (jp = json_object_get(jdh, "dh_p"))) || (NULL == (jg = json_object_get(jdh, "dh_g"))) || (NULL == (jpub_key = json_object_get(jdh, "dh_pub_key")))) { tr_debug("tr_msg_decode_dh(): Error parsing dh_info."); - free(dh); + tr_dh_destroy(dh); return NULL; } @@ -145,17 +245,17 @@ static json_t * tr_msg_encode_tidreq(TID_REQ *req) assert(jreq = json_object()); - jstr = json_string(req->rp_realm->buf); + jstr = tr_name_to_json_string(req->rp_realm); json_object_set_new(jreq, "rp_realm", jstr); - jstr = json_string(req->realm->buf); + jstr = tr_name_to_json_string(req->realm); json_object_set_new(jreq, "target_realm", jstr); - jstr = json_string(req->comm->buf); + jstr = tr_name_to_json_string(req->comm); json_object_set_new(jreq, "community", jstr); if (req->orig_coi) { - jstr = json_string(req->orig_coi->buf); + jstr = tr_name_to_json_string(req->orig_coi); json_object_set_new(jreq, "orig_coi", jstr); } @@ -173,7 +273,7 @@ static json_t * tr_msg_encode_tidreq(TID_REQ *req) return jreq; } -static TID_REQ *tr_msg_decode_tidreq(json_t *jreq) +static TID_REQ *tr_msg_decode_tidreq(TALLOC_CTX *mem_ctx, json_t *jreq) { TID_REQ *treq = NULL; json_t *jrp_realm = NULL; @@ -188,7 +288,8 @@ static TID_REQ *tr_msg_decode_tidreq(json_t *jreq) tr_crit("tr_msg_decode_tidreq(): Error allocating TID_REQ structure."); return NULL; } - + talloc_steal(mem_ctx, treq); + /* store required fields from request */ if ((NULL == (jrp_realm = json_object_get(jreq, "rp_realm"))) || (NULL == (jrealm = json_object_get(jreq, "target_realm"))) || @@ -249,20 +350,19 @@ static json_t *tr_msg_encode_one_server(TID_SRVR_BLK *srvr) jsrvr = json_object(); - /* Server IP Address -- TBD handle IPv6 */ - jstr = json_string(inet_ntoa(srvr->aaa_server_addr)); + jstr = json_string(srvr->aaa_server_addr); json_object_set_new(jsrvr, "server_addr", jstr); json_object_set_new(jsrvr, "key_expiration", json_string(time_str)); g_free(time_str); /* Server DH Block */ - jstr = json_string(srvr->key_name->buf); + jstr = tr_name_to_json_string(srvr->key_name); json_object_set_new(jsrvr, "key_name", jstr); json_object_set_new(jsrvr, "server_dh", tr_msg_encode_dh(srvr->aaa_server_dh)); if (srvr->path) /* The path is owned by the srvr, so grab an extra ref*/ - json_object_set(jsrvr, "path", srvr->path); + json_object_set(jsrvr, "path", (json_t *)(srvr->path)); return jsrvr; } @@ -283,12 +383,11 @@ static int tr_msg_decode_one_server(json_t *jsrvr, TID_SRVR_BLK *srvr) tr_notice("tr_msg_decode_one_server(): Error parsing required fields."); return -1; } - - /* TBD -- handle IPv6 Addresses */ - inet_aton(json_string_value(jsrvr_addr), &(srvr->aaa_server_addr)); + + srvr->aaa_server_addr=talloc_strdup(srvr, json_string_value(jsrvr_addr)); srvr->key_name = tr_new_name((char *)json_string_value(jsrvr_kn)); srvr->aaa_server_dh = tr_msg_decode_dh(jsrvr_dh); - srvr->path = json_object_get(jsrvr, "path"); + tid_srvr_blk_set_path(srvr, (TID_PATH *) json_object_get(jsrvr, "path")); jsrvr_expire = json_object_get(jsrvr, "key_expiration"); if (jsrvr_expire && json_is_string(jsrvr_expire)) { if (!g_time_val_from_iso8601(json_string_value(jsrvr_expire), @@ -320,9 +419,11 @@ static json_t *tr_msg_encode_servers(TID_RESP *resp) return jservers; } -static TID_SRVR_BLK *tr_msg_decode_servers(void * ctx, json_t *jservers, size_t *out_len) +static TID_SRVR_BLK *tr_msg_decode_servers(TALLOC_CTX *mem_ctx, json_t *jservers) { - TID_SRVR_BLK *servers = NULL; + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TID_SRVR_BLK *servers=NULL; + TID_SRVR_BLK *new_srvr=NULL; json_t *jsrvr; size_t i, num_servers; @@ -331,20 +432,30 @@ static TID_SRVR_BLK *tr_msg_decode_servers(void * ctx, json_t *jservers, size_t if (0 == num_servers) { tr_debug("tr_msg_decode_servers(): Server array is empty."); - return NULL; + goto cleanup; } - servers = talloc_zero_array(ctx, TID_SRVR_BLK, num_servers); for (i = 0; i < num_servers; i++) { jsrvr = json_array_get(jservers, i); - if (0 != tr_msg_decode_one_server(jsrvr, &servers[i])) { - talloc_free(servers); - return NULL; - } + new_srvr=tid_srvr_blk_new(tmp_ctx); + if (new_srvr==NULL) { + servers=NULL; /* it's all in tmp_ctx, so we can just let go */ + goto cleanup; + } + + if (0 != tr_msg_decode_one_server(jsrvr, new_srvr)) { + servers=NULL; /* it's all in tmp_ctx, so we can just let go */ + goto cleanup; + } + tid_srvr_blk_add(servers, new_srvr); } - *out_len = num_servers; + + talloc_steal(mem_ctx, servers); + +cleanup: + talloc_free(tmp_ctx); return servers; } @@ -363,7 +474,7 @@ static json_t * tr_msg_encode_tidresp(TID_RESP *resp) jstr = json_string("error"); json_object_set_new(jresp, "result", jstr); if (resp->err_msg) { - jstr = json_string(resp->err_msg->buf); + jstr = tr_name_to_json_string(resp->err_msg); json_object_set_new(jresp, "err_msg", jstr); } } @@ -372,17 +483,17 @@ static json_t * tr_msg_encode_tidresp(TID_RESP *resp) json_object_set_new(jresp, "result", jstr); } - jstr = json_string(resp->rp_realm->buf); + jstr = tr_name_to_json_string(resp->rp_realm); json_object_set_new(jresp, "rp_realm", jstr); - jstr = json_string(resp->realm->buf); + jstr = tr_name_to_json_string(resp->realm); json_object_set_new(jresp, "target_realm", jstr); - jstr = json_string(resp->comm->buf); + jstr = tr_name_to_json_string(resp->comm); json_object_set_new(jresp, "comm", jstr); if (resp->orig_coi) { - jstr = json_string(resp->orig_coi->buf); + jstr = tr_name_to_json_string(resp->orig_coi); json_object_set_new(jresp, "orig_coi", jstr); } @@ -400,7 +511,7 @@ static json_t * tr_msg_encode_tidresp(TID_RESP *resp) return jresp; } -static TID_RESP *tr_msg_decode_tidresp(json_t *jresp) +static TID_RESP *tr_msg_decode_tidresp(TALLOC_CTX *mem_ctx, json_t *jresp) { TID_RESP *tresp = NULL; json_t *jresult = NULL; @@ -411,11 +522,10 @@ static TID_RESP *tr_msg_decode_tidresp(json_t *jresp) json_t *jservers = NULL; json_t *jerr_msg = NULL; - if (!(tresp = talloc_zero(NULL, TID_RESP))) { + if (!(tresp=tid_resp_new(mem_ctx))) { tr_crit("tr_msg_decode_tidresp(): Error allocating TID_RESP structure."); return NULL; } - /* store required fields from response */ if ((NULL == (jresult = json_object_get(jresp, "result"))) || @@ -435,7 +545,7 @@ static TID_RESP *tr_msg_decode_tidresp(json_t *jresp) tr_debug("tr_msg_decode_tidresp(): Success! result = %s.", json_string_value(jresult)); if ((NULL != (jservers = json_object_get(jresp, "servers"))) || (!json_is_array(jservers))) { - tresp->servers = tr_msg_decode_servers(tresp, jservers, &tresp->num_servers); + tresp->servers = tr_msg_decode_servers(tresp, jservers); } else { talloc_free(tresp); @@ -448,62 +558,701 @@ static TID_RESP *tr_msg_decode_tidresp(json_t *jresp) tr_debug("tr_msg_decode_tidresp(): Error! result = %s.", json_string_value(jresult)); if ((NULL != (jerr_msg = json_object_get(jresp, "err_msg"))) || (!json_is_string(jerr_msg))) { - tresp->err_msg = tr_new_name((char *)json_string_value(jerr_msg)); - } + tresp->err_msg = tr_new_name(json_string_value(jerr_msg)); + } else + tresp->err_msg = tr_new_name("No error message set."); + + if (NULL !=(tresp->error_path = json_object_get(jresp, "error_path"))) + json_incref(tresp->error_path); } - tresp->rp_realm = tr_new_name((char *)json_string_value(jrp_realm)); - tresp->realm = tr_new_name((char *)json_string_value(jrealm)); - tresp->comm = tr_new_name((char *)json_string_value(jcomm)); + tresp->rp_realm = tr_new_name(json_string_value(jrp_realm)); + tresp->realm = tr_new_name(json_string_value(jrealm)); + tresp->comm = tr_new_name(json_string_value(jcomm)); /* store optional "orig_coi" field */ if ((NULL != (jorig_coi = json_object_get(jresp, "orig_coi"))) && (!json_is_object(jorig_coi))) { - tresp->orig_coi = tr_new_name((char *)json_string_value(jorig_coi)); + tresp->orig_coi = tr_new_name(json_string_value(jorig_coi)); } return tresp; } -char *tr_msg_encode(TR_MSG *msg) + +/* Information records for TRP update msg + * requires that jrec already be allocated */ +static TRP_RC tr_msg_encode_inforec_route(json_t *jrec, TRP_INFOREC *rec) { - json_t *jmsg; - json_t *jmsg_type; + json_t *jstr=NULL; + json_t *jint=NULL; + char *s=NULL; + + if (rec==NULL) + return TRP_BADTYPE; + + if (trp_inforec_get_trust_router(rec)==NULL) + return TRP_ERROR; + + s=tr_name_strdup(trp_inforec_get_trust_router(rec)); + if (s==NULL) + return TRP_NOMEM; + jstr=json_string(s); + free(s);s=NULL; + if(jstr==NULL) + return TRP_ERROR; + json_object_set_new(jrec, "trust_router", jstr); + + jint=json_integer(trp_inforec_get_metric(rec)); + if(jint==NULL) + return TRP_ERROR; + json_object_set_new(jrec, "metric", jint); + + jint=json_integer(trp_inforec_get_interval(rec)); + if(jint==NULL) + return TRP_ERROR; + json_object_set_new(jrec, "interval", jint); + + return TRP_SUCCESS; +} + +/* returns a json array */ +static json_t *tr_msg_encode_apcs(TR_APC *apcs) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TR_APC_ITER *iter=tr_apc_iter_new(tmp_ctx); + TR_APC *apc=NULL; + json_t *jarray=NULL; + json_t *jid=NULL; + + if (iter==NULL) + goto cleanup; + + jarray=json_array(); + if (jarray==NULL) + goto cleanup; + + for (apc=tr_apc_iter_first(iter, apcs); apc!=NULL; apc=tr_apc_iter_next(iter)) { + jid=tr_name_to_json_string(tr_apc_get_id(apc)); + if ((jid==NULL) || (json_array_append_new(jarray, jid)!=0)) { + json_decref(jarray); + jarray=NULL; + goto cleanup; + } + } + +cleanup: + talloc_free(tmp_ctx); + return jarray; +} + +static TR_APC *tr_msg_decode_apcs(TALLOC_CTX *mem_ctx, json_t *jarray, TRP_RC *rc) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + size_t ii=0; + TR_APC *apc_list=NULL; + TR_APC *new=NULL; + json_t *jstr=NULL; + + *rc=TRP_ERROR; + + for (ii=0; iitype) { + case TRP_INFOREC_TYPE_ROUTE: + if (TRP_SUCCESS!=tr_msg_encode_inforec_route(jrec, rec)) { + json_decref(jrec); + return NULL; + } + break; + case TRP_INFOREC_TYPE_COMMUNITY: + if (TRP_SUCCESS!=tr_msg_encode_inforec_comm(jrec, rec)) { + json_decref(jrec); + return NULL; + } + break; + default: + json_decref(jrec); + return NULL; + } + return jrec; +} + +static TRP_RC tr_msg_decode_trp_inforec_route(json_t *jrecord, TRP_INFOREC *rec) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TRP_RC rc=TRP_ERROR; + char *s=NULL; + int num=0; + + rc=tr_msg_get_json_string(jrecord, "trust_router", &s, tmp_ctx); + if (rc != TRP_SUCCESS) + goto cleanup; + if (TRP_SUCCESS!=trp_inforec_set_trust_router(rec, tr_new_name(s))) { + rc=TRP_ERROR; + goto cleanup; + } + talloc_free(s); s=NULL; + + trp_inforec_set_next_hop(rec, NULL); /* make sure this is null (filled in later) */ + + rc=tr_msg_get_json_integer(jrecord, "metric", &num); + if ((rc != TRP_SUCCESS) || (TRP_SUCCESS!=trp_inforec_set_metric(rec,num))) + goto cleanup; + + rc=tr_msg_get_json_integer(jrecord, "interval", &num); + if ((rc != TRP_SUCCESS) || (TRP_SUCCESS!=trp_inforec_set_interval(rec,num))) + goto cleanup; + +cleanup: + talloc_free(tmp_ctx); + return rc; +} + +static TRP_RC tr_msg_decode_trp_inforec_comm(json_t *jrecord, TRP_INFOREC *rec) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TRP_RC rc=TRP_ERROR; + char *s=NULL; + int num=0; + TR_APC *apcs=NULL; + + rc=tr_msg_get_json_string(jrecord, "type", &s, tmp_ctx); + if (rc != TRP_SUCCESS) + goto cleanup; + if (TRP_SUCCESS!=trp_inforec_set_comm_type(rec, tr_comm_type_from_str(s))) { + rc=TRP_ERROR; + goto cleanup; + } + talloc_free(s); s=NULL; + + rc=tr_msg_get_json_string(jrecord, "role", &s, tmp_ctx); + if (rc != TRP_SUCCESS) + goto cleanup; + if (TRP_SUCCESS!=trp_inforec_set_role(rec, tr_realm_role_from_str(s))) { + rc=TRP_ERROR; + goto cleanup; + } + talloc_free(s); s=NULL; + + apcs=tr_msg_decode_apcs(rec, json_object_get(jrecord, "apcs"), &rc); + if (rc!=TRP_SUCCESS) { + rc=TRP_ERROR; + goto cleanup; + } + trp_inforec_set_apcs(rec, apcs); + + rc=tr_msg_get_json_integer(jrecord, "interval", &num); + tr_debug("tr_msg_decode_trp_inforec_comm: interval=%u", num); + if ((rc != TRP_SUCCESS) || (TRP_SUCCESS!=trp_inforec_set_interval(rec,num))) + goto cleanup; + + trp_inforec_set_provenance(rec, json_object_get(jrecord, "provenance")); + + /* optional */ + rc=tr_msg_get_json_string(jrecord, "owner_realm", &s, tmp_ctx); + if (rc == TRP_SUCCESS) { + if (TRP_SUCCESS!=trp_inforec_set_owner_realm(rec, tr_new_name(s))) { + rc=TRP_ERROR; + goto cleanup; + } + if (s!=NULL) { + talloc_free(s); + s=NULL; + } + } + + rc=tr_msg_get_json_string(jrecord, "owner_contact", &s, tmp_ctx); + if (rc == TRP_SUCCESS) { + if (TRP_SUCCESS!=trp_inforec_set_owner_contact(rec, tr_new_name(s))) { + rc=TRP_ERROR; + goto cleanup; + } + if (s!=NULL) { + talloc_free(s); + s=NULL; + } + } + +cleanup: + talloc_free(tmp_ctx); + return rc; +} + +/* decode a single record */ +static TRP_INFOREC *tr_msg_decode_trp_inforec(TALLOC_CTX *mem_ctx, json_t *jrecord) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + TRP_INFOREC_TYPE rectype; + TRP_INFOREC *rec=NULL; + TRP_RC rc=TRP_ERROR; + char *s=NULL; + + if (TRP_SUCCESS!=tr_msg_get_json_string(jrecord, "record_type", &s, tmp_ctx)) + goto cleanup; + + rectype=trp_inforec_type_from_string(s); + talloc_free(s); s=NULL; + + rec=trp_inforec_new(tmp_ctx, rectype); + if (rec==NULL) { + rc=TRP_NOMEM; + goto cleanup; + } + + tr_debug("tr_msg_decode_trp_inforec: '%s' record found.", trp_inforec_type_to_string(rec->type)); + + switch(trp_inforec_get_type(rec)) { + case TRP_INFOREC_TYPE_ROUTE: + rc=tr_msg_decode_trp_inforec_route(jrecord, rec); + break; + case TRP_INFOREC_TYPE_COMMUNITY: + rc=tr_msg_decode_trp_inforec_comm(jrecord, rec); + break; + default: + rc=TRP_UNSUPPORTED; + goto cleanup; + } + + talloc_steal(mem_ctx, rec); + rc=TRP_SUCCESS; + +cleanup: + if (rc != TRP_SUCCESS) { + trp_inforec_free(rec); + rec=NULL; + } + talloc_free(tmp_ctx); + return rec; +} + +/* TRP update msg */ +static json_t *tr_msg_encode_trp_upd(TRP_UPD *update) +{ + json_t *jupdate=NULL; + json_t *jrecords=NULL; + json_t *jrec=NULL; + json_t *jstr=NULL; + TRP_INFOREC *rec; + char *s=NULL; + + if (update==NULL) + return NULL; + + jupdate=json_object(); + if (jupdate==NULL) + return NULL; + + s=tr_name_strdup(trp_upd_get_comm(update)); + if (s==NULL) { + json_decref(jupdate); + return NULL; + } + jstr=json_string(s); + free(s);s=NULL; + if(jstr==NULL) { + json_decref(jupdate); + return NULL; + } + json_object_set_new(jupdate, "community", jstr); + + s=tr_name_strdup(trp_upd_get_realm(update)); + if (s==NULL) { + json_decref(jupdate); + return NULL; + } + jstr=json_string(s); + free(s);s=NULL; + if(jstr==NULL) { + json_decref(jupdate); + return NULL; + } + json_object_set_new(jupdate, "realm", jstr); + + jrecords=json_array(); + if (jrecords==NULL) { + json_decref(jupdate); + return NULL; + } + json_object_set_new(jupdate, "records", jrecords); /* jrecords now a "borrowed" reference */ + for (rec=trp_upd_get_inforec(update); rec!=NULL; rec=trp_inforec_get_next(rec)) { + tr_debug("tr_msg_encode_trp_upd: encoding inforec."); + jrec=tr_msg_encode_inforec(rec); + if (jrec==NULL) { + json_decref(jupdate); /* also decs jrecords and any elements */ + return NULL; + } + if (0!=json_array_append_new(jrecords, jrec)) { + json_decref(jupdate); /* also decs jrecords and any elements */ + json_decref(jrec); /* this one did not get added so dec explicitly */ + return NULL; + } + } + + return jupdate; +} + +/* Creates a linked list of records in the msg->body talloc context. + * An error will be returned if any unparseable records are encountered. + */ +static TRP_UPD *tr_msg_decode_trp_upd(TALLOC_CTX *mem_ctx, json_t *jupdate) +{ + TALLOC_CTX *tmp_ctx=talloc_new(NULL); + json_t *jrecords=NULL; + size_t ii=0; + TRP_UPD *update=NULL; + TRP_INFOREC *new_rec=NULL; + TRP_INFOREC *list_tail=NULL; + char *s=NULL; + TR_NAME *name; + TRP_RC rc=TRP_ERROR; + + update=trp_upd_new(tmp_ctx); + if (update==NULL) { + rc=TRP_NOMEM; + goto cleanup; + } + + rc=tr_msg_get_json_string(jupdate, "community", &s, tmp_ctx); + if (rc != TRP_SUCCESS) { + tr_debug("tr_msg_decode_trp_upd: no community in TRP update message."); + rc=TRP_NOPARSE; + goto cleanup; + } + name=tr_new_name(s); + if (name==NULL) { + tr_debug("tr_msg_decode_trp_upd: could not allocate community name."); + rc=TRP_NOMEM; + goto cleanup; + } + talloc_free(s); s=NULL; + trp_upd_set_comm(update, name); + + rc=tr_msg_get_json_string(jupdate, "realm", &s, tmp_ctx); + if (rc != TRP_SUCCESS) { + tr_debug("tr_msg_decode_trp_upd: no realm in TRP update message."); + rc=TRP_NOPARSE; + goto cleanup; + } + name=tr_new_name(s); + if (name==NULL) { + tr_debug("tr_msg_decode_trp_upd: could not allocate realm name."); + rc=TRP_NOMEM; + goto cleanup; + } + talloc_free(s); s=NULL; + trp_upd_set_realm(update, name); + + jrecords=json_object_get(jupdate, "records"); + if ((jrecords==NULL) || (!json_is_array(jrecords))) { + rc=TRP_NOPARSE; + goto cleanup; + } + + tr_debug("tr_msg_decode_trp_upd: found %d records", json_array_size(jrecords)); + /* process the array */ + for (ii=0; iimsg_type) - { + switch (msg->msg_type) { case TID_REQUEST: jmsg_type = json_string("tid_request"); json_object_set_new(jmsg, "msg_type", jmsg_type); - json_object_set_new(jmsg, "msg_body", tr_msg_encode_tidreq(tr_msg_get_req(msg))); + tidreq=tr_msg_get_req(msg); + json_object_set_new(jmsg, "msg_body", tr_msg_encode_tidreq(tidreq)); break; case TID_RESPONSE: jmsg_type = json_string("tid_response"); json_object_set_new(jmsg, "msg_type", jmsg_type); - json_object_set_new(jmsg, "msg_body", tr_msg_encode_tidresp(tr_msg_get_resp(msg))); + tidresp=tr_msg_get_resp(msg); + json_object_set_new(jmsg, "msg_body", tr_msg_encode_tidresp(tidresp)); + break; + + case TRP_UPDATE: + jmsg_type = json_string("trp_update"); + json_object_set_new(jmsg, "msg_type", jmsg_type); + trpupd=tr_msg_get_trp_upd(msg); + json_object_set_new(jmsg, "msg_body", tr_msg_encode_trp_upd(trpupd)); + break; + + case TRP_REQUEST: + jmsg_type = json_string("trp_request"); + json_object_set_new(jmsg, "msg_type", jmsg_type); + trpreq=tr_msg_get_trp_req(msg); + json_object_set_new(jmsg, "msg_body", tr_msg_encode_trp_req(trpreq)); break; - /* TBD -- Add TR message types */ + case MON_REQUEST: + jmsg_type = json_string("mon_request"); + json_object_set_new(jmsg, "msg_type", jmsg_type); + monreq=tr_msg_get_mon_req(msg); + json_object_set_new(jmsg, "msg_body", mon_req_encode(monreq)); + break; + + case MON_RESPONSE: + jmsg_type = json_string("mon_response"); + json_object_set_new(jmsg, "msg_type", jmsg_type); + monresp=tr_msg_get_mon_resp(msg); + json_object_set_new(jmsg, "msg_body", mon_resp_encode(monresp)); + break; default: json_decref(jmsg); return NULL; - } - - return(json_dumps(jmsg, 0)); + } + + /* We should perhaps use json_set_alloc_funcs to automatically use talloc, but for + * now, we'll encode to a malloc'ed buffer, then copy that to a talloc'ed buffer. */ + encoded_tmp=json_dumps(jmsg, 0); // malloc'ed version + json_decref(jmsg); // free the JSON structure + encoded = talloc_strdup(mem_ctx, encoded_tmp); // get the talloc'ed version + free(encoded_tmp); // free the malloc'ed version + + tr_debug("tr_msg_encode: outgoing msg=%s", encoded); + return encoded; } -TR_MSG *tr_msg_decode(char *jbuf, size_t buflen) +TR_MSG *tr_msg_decode(TALLOC_CTX *mem_ctx, const char *jbuf, size_t buflen) { - TR_MSG *msg; + TR_MSG *msg=NULL; json_t *jmsg = NULL; json_error_t rc; - json_t *jtype; - json_t *jbody; + json_t *jtype=NULL; + json_t *jbody=NULL; const char *mtype = NULL; if (NULL == (jmsg = json_loadb(jbuf, buflen, JSON_DISABLE_EOF_CHECK, &rc))) { @@ -511,14 +1260,12 @@ TR_MSG *tr_msg_decode(char *jbuf, size_t buflen) return NULL; } - if (!(msg = malloc(sizeof(TR_MSG)))) { + if (!(msg = talloc_zero(mem_ctx, TR_MSG))) { tr_debug("tr_msg_decode(): Error allocating TR_MSG structure."); json_decref(jmsg); return NULL; } - memset(msg, 0, sizeof(TR_MSG)); - if ((NULL == (jtype = json_object_get(jmsg, "msg_type"))) || (NULL == (jbody = json_object_get(jmsg, "msg_body")))) { tr_debug("tr_msg_decode(): Error parsing message header."); @@ -531,29 +1278,47 @@ TR_MSG *tr_msg_decode(char *jbuf, size_t buflen) if (0 == strcmp(mtype, "tid_request")) { msg->msg_type = TID_REQUEST; - tr_msg_set_req(msg, tr_msg_decode_tidreq(jbody)); + tr_msg_set_req(msg, tr_msg_decode_tidreq(msg, jbody)); } else if (0 == strcmp(mtype, "tid_response")) { msg->msg_type = TID_RESPONSE; - tr_msg_set_resp(msg, tr_msg_decode_tidresp(jbody)); + tr_msg_set_resp(msg, tr_msg_decode_tidresp(msg, jbody)); + } + else if (0 == strcmp(mtype, "trp_update")) { + msg->msg_type = TRP_UPDATE; + tr_msg_set_trp_upd(msg, tr_msg_decode_trp_upd(msg, jbody)); /* null talloc context for now */ + } + else if (0 == strcmp(mtype, "trp_request")) { + msg->msg_type = TRP_UPDATE; + tr_msg_set_trp_req(msg, tr_msg_decode_trp_req(msg, jbody)); /* null talloc context for now */ + } + else if (0 == strcmp(mtype, "mon_request")) { + msg->msg_type = MON_REQUEST; + tr_msg_set_mon_req(msg, mon_req_decode(msg, jbody)); + } + /* We do not currently handle monitoring responses */ + else if (0 == strcmp(mtype, "mon_response")) { + msg->msg_type = MON_RESPONSE; + tr_msg_set_mon_resp(msg, mon_resp_decode(msg, jbody)); } else { msg->msg_type = TR_UNKNOWN; msg->msg_rep = NULL; } + + json_decref(jmsg); + return msg; } void tr_msg_free_encoded(char *jmsg) { if (jmsg) - free (jmsg); + talloc_free(jmsg); } void tr_msg_free_decoded(TR_MSG *msg) { if (msg) - free (msg); + talloc_free(msg); } - -