safeguards

src/baseprotocol.c

@@ -228,6 +228,7 @@ PEP_STATUS try_base_prepare_message(
 {
     PEP_STATUS status = PEP_STATUS_OK;
 
+    assert(session && session->messageToSend && session->notifyHandshake);
     assert(me);
     assert(partner);
     assert(payload);
@@ -235,6 +236,9 @@ PEP_STATUS try_base_prepare_message(
     assert(result);
     assert(type == BASE_SYNC || type == BASE_KEYRESET);
 
+    if (!(session && session->messageToSend && session->notifyHandshake))
+        return PEP_ILLEGAL_VALUE;
+
     if (!(me && partner && payload && size && result && type))
         return PEP_ILLEGAL_VALUE;
 

src/message_api.c

@@ -5322,11 +5322,12 @@ PEP_STATUS try_encrypt_message(
 {
     PEP_STATUS status = PEP_STATUS_OK;
 
-    assert(session && session->messageToSend);
+    assert(session && session->messageToSend && session->notifyHandshake);
     assert(src && src->from);
     assert(dst);
 
-    if (!(session && session->messageToSend && src && src->from && dst))
+    if (!(session && session->messageToSend && session->notifyHandshake && src
+                && src->from && dst))
         return PEP_ILLEGAL_VALUE;
 
     if (src->dir == PEP_dir_incoming)