Merge pull request #46 from LarryLaffer-dev/diameter_response_codes

Diameter response codes

Author: LarryLaffer-dev

modules/aaa_diameter/aaa_diameter.c

@@ -357,26 +357,21 @@ static int dm_send_request(struct sip_msg *msg, int *app_id, int *cmd_code,
 	if (_dm_send_message(NULL, dmsg, &rpl) != 0)
 		goto ret;
 
-	{
-		diameter_reply dm_rpl;
-		memset(&dm_rpl, 0, sizeof dm_rpl);
-		rc = _dm_get_message_response(rpl,
-				(rpl_avps_pv ? &rpl_avps : NULL), &dm_rpl);
-
-		dm_last_result_code = dm_rpl.rc;
-		dm_last_experimental_rc = dm_rpl.experimental_rc;
-
-		if (rpl_avps_pv) {
-			pv_value_t val = {(str){rpl_avps, strlen(rpl_avps)}, 0, PV_VAL_STR};
-			if (pv_set_value(msg, rpl_avps_pv, 0, &val) != 0)
-				LM_ERR("failed to set output rpl_avps pv to: %s\n", rpl_avps);
-			_dm_release_message_response(rpl, rpl_avps);
-		}
+	rc = _dm_get_message_response(rpl, (rpl_avps_pv ? &rpl_avps : NULL));
 
-		if (rc != 0) {
-			LM_ERR("Diameter request failed (rc: %d)\n", rc);
-			return -3;
-		}
+	dm_last_result_code = rpl->rpl.rc;
+	dm_last_experimental_rc = rpl->experimental_rc;
+
+	if (rpl_avps_pv) {
+		pv_value_t val = {(str){rpl_avps, strlen(rpl_avps)}, 0, PV_VAL_STR};
+		if (pv_set_value(msg, rpl_avps_pv, 0, &val) != 0)
+			LM_ERR("failed to set output rpl_avps pv to: %s\n", rpl_avps);
+		_dm_release_message_response(rpl, rpl_avps);
+	}
+
+	if (rc != 0) {
+		LM_ERR("Diameter request failed (rc: %d)\n", rc);
+		return -3;
 	}
 
 	return 1;
@@ -541,15 +536,12 @@ static int dm_send_request_async_reply(int fd,
 		LM_ERR("could not resume async route!\n");
 		goto error;
 	}
-	{
-		diameter_reply dm_rpl;
-		memset(&dm_rpl, 0, sizeof dm_rpl);
-		ret = _dm_get_message_response(amsg->cond,
-				(amsg->ret ? &rpl_avps : NULL), &dm_rpl);
 
-		dm_last_result_code = dm_rpl.rc;
-		dm_last_experimental_rc = dm_rpl.experimental_rc;
-	}
+	ret = _dm_get_message_response(amsg->cond,
+			(amsg->ret ? &rpl_avps : NULL));
+
+	dm_last_result_code = amsg->cond->rpl.rc;
+	dm_last_experimental_rc = amsg->cond->experimental_rc;
 
 	if (ret == 0)
 		ret = 1;

modules/aaa_diameter/diameter_api.h

@@ -34,9 +34,7 @@
 typedef struct {
 	cJSON *json;
 	int is_error;
-	int rc;                /* Result-Code AVP (268) */
-	int experimental_rc;   /* Experimental-Result-Code AVP (298) */
-	int vendor_id;         /* Vendor-Id from Experimental-Result (266) */
+	int rc;
 } diameter_reply;
 
 /*

modules/aaa_diameter/dm_impl.c

@@ -720,15 +720,17 @@ static int dm_receive_msg(struct msg **_msg, struct avp * avp, struct session *
 	if (rc == 0 && a) {
 		struct avp *child = NULL;
 		fd_msg_browse(a, MSG_BRW_FIRST_CHILD, &child, NULL);
+
 		while (child) {
 			rc = fd_msg_avp_hdr(child, &h);
 			if (rc == 0) {
-				if (h->avp_code == 298) {
-					rpl_cond->rpl.experimental_rc = h->avp_value->u32;
-					LM_DBG("Experimental-Result-Code: %u\n", h->avp_value->u32);
+			if (h->avp_code == 298 && h->avp_value) {
+				rpl_cond->experimental_rc = h->avp_value->u32;
+				LM_DBG("Experimental-Result-Code: %u\n", h->avp_value->u32);
+				if (h->avp_value->u32 < 2000 || h->avp_value->u32 >= 3000)
 					rpl_cond->rpl.is_error = 1;
-				} else if (h->avp_code == 266) {
-					rpl_cond->rpl.vendor_id = h->avp_value->u32;
+				} else if (h->avp_code == 266 && h->avp_value) {
+					rpl_cond->vendor_id = h->avp_value->u32;
 					LM_DBG("Experimental-Result Vendor-Id: %u\n", h->avp_value->u32);
 				}
 			}
@@ -2031,8 +2033,7 @@ static int _dm_get_message_reply(struct dm_cond *cond, diameter_reply *rpl)
 	return (cond->rpl.is_error?-1:0);
 }
 
-int _dm_get_message_response(struct dm_cond *cond, char **rpl_avps,
-		diameter_reply *rpl_out)
+int _dm_get_message_response(struct dm_cond *cond, char **rpl_avps)
 {
 	cJSON *obj;
 	diameter_reply rpl;
@@ -2043,10 +2044,6 @@ int _dm_get_message_response(struct dm_cond *cond, char **rpl_avps,
 		*rpl_avps = cJSON_PrintUnformatted(obj);
 		LM_DBG("AVPs: %s\n", *rpl_avps);
 	}
-
-	if (rpl_out)
-		*rpl_out = rpl;
-
 	return rc;
 }
 

modules/aaa_diameter/dm_impl.h

@@ -141,6 +141,9 @@ struct dm_cond {
 	} sync;
 
 	diameter_reply rpl;
+
+	int experimental_rc;   /* Experimental-Result-Code AVP (298) */
+	int vendor_id;         /* Vendor-Id from Experimental-Result (266) */
 };
 int init_mutex_cond(pthread_mutex_t *mutex, pthread_cond_t *cond);
 
@@ -173,8 +176,7 @@ int dm_build_avps(struct list_head *subavps, cJSON *array);
 int dm_send_message(aaa_conn *_, aaa_message *req, aaa_message **__);
 int _dm_send_message(aaa_conn *_, aaa_message *req, struct dm_cond **reply_cond);
 int _dm_send_message_async(aaa_conn *_, aaa_message *req, int *fd);
-int _dm_get_message_response(struct dm_cond *cond, char **rpl_avps,
-		diameter_reply *rpl_out);
+int _dm_get_message_response(struct dm_cond *cond, char **rpl_avps);
 void _dm_release_message_response(struct dm_cond *cond, char *rpl_avps);
 int dm_destroy_message(aaa_conn *con, aaa_message *msg);
 void _dm_destroy_message(aaa_message *msg);

modules/tm/async.c

@@ -131,7 +131,8 @@ int t_resume_async_request(int fd, void*param, int was_timeout)
 	backup_list = set_avp_list( &t->user_avps );
 	/* set default send address to the saved value */
 	backup_si = bind_address;
-	bind_address = TM_BRANCH( t, 0).request.dst.send_sock;
+	if (t->uac[0])
+		bind_address = TM_BRANCH( t, 0).request.dst.send_sock;
 
 	async_status = ASYNC_DONE; /* assume default status as done */
 	/* call the resume function in order to read and handle data */