Add state to the on_session_message parameter.

    * When state is closed, it implies that packet is null and the session will be destroyed

1 files changed, 7 insertions, 2 deletions

diff --git a/test/lpi_plus/gtest_lpip_module.c b/test/lpi_plus/gtest_lpip_module.c
index abcbe61..49382a8 100644
--- a/test/lpi_plus/gtest_lpip_module.c
+++ b/test/lpi_plus/gtest_lpip_module.c
@@ -130,9 +130,14 @@ static void gtest_lpip_on_appid_msg(struct session *sess, enum APPID_ORIGIN orig
     return;
 }
 
-static void on_session(struct session *sess, struct packet *pkt, void *args)
+static void on_session(struct session *sess, enum session_state state, struct packet *pkt, void *args)
 {
-    if(sess==NULL || pkt==NULL || args==NULL)return;
+    if (state == SESSION_STATE_CLOSED)
+    {
+        assert(pkt == NULL);
+        return;
+    }
+
     struct test_lpip_env *env = (struct test_lpip_env *)args;
     if (session_get_current_state(sess) == SESSION_STATE_OPENING)
     {