From: Tobias Brunner <tobias@strongswan.org>
Date: Mon, 30 May 2016 13:39:38 +0000 (+0200)
Subject: unit-tests: Only deliver messages to the SA they are addressed to
X-Git-Tag: 5.5.0dr1~4^2~37
X-Git-Url: http://git.ipfire.org/gitweb/gitweb.cgi?a=commitdiff_plain;h=40d9a4c892bb18440bc463ba7bc63e0fa31195c3;p=thirdparty%2Fstrongswan.git

unit-tests: Only deliver messages to the SA they are addressed to
---

diff --git a/src/libcharon/tests/utils/exchange_test_helper.c b/src/libcharon/tests/utils/exchange_test_helper.c
index ab562271f6..f32906d5dc 100644
--- a/src/libcharon/tests/utils/exchange_test_helper.c
+++ b/src/libcharon/tests/utils/exchange_test_helper.c
@@ -182,16 +182,24 @@ METHOD(backend_t, create_peer_cfg_enumerator, enumerator_t*,
 METHOD(exchange_test_helper_t, process_message, status_t,
 	private_exchange_test_helper_t *this, ike_sa_t *ike_sa, message_t *message)
 {
-	status_t status;
+	status_t status = FAILED;
+	ike_sa_id_t *id;
 
 	if (!message)
 	{
 		message = this->public.sender->dequeue(this->public.sender);
 	}
-	charon->bus->set_sa(charon->bus, ike_sa);
-	status = ike_sa->process_message(ike_sa, message);
-	charon->bus->set_sa(charon->bus, NULL);
+	id = message->get_ike_sa_id(message);
+	id = id->clone(id);
+	id->switch_initiator(id);
+	if (!id->get_responder_spi(id) || id->equals(id, ike_sa->get_id(ike_sa)))
+	{
+		charon->bus->set_sa(charon->bus, ike_sa);
+		status = ike_sa->process_message(ike_sa, message);
+		charon->bus->set_sa(charon->bus, NULL);
+	}
 	message->destroy(message);
+	id->destroy(id);
 	return status;
 }