Merge pull request #671 from ejohnstown/msgid-filter2

Messaging Filtering Followup
wolfSSL · Mar 21, 2024 · 295e5ab · 295e5ab
2 parents 863714a + 26c8b58
commit 295e5ab
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/internal.c b/src/internal.c
@@ -586,6 +586,14 @@ INLINE static int IsMessageAllowedServer(WOLFSSH *ssh, byte msg)
             return 0;
         }
     }
+    else {
+        if (msg >= MSGID_USERAUTH_RESTRICT && msg < MSGID_USERAUTH_LIMIT) {
+            WLOG(WS_LOG_DEBUG, "Message ID %u not allowed by server "
+                    "after user authentication", msg);
+            return 0;
+        }
+    }
+
     return 1;
 }
 #endif /* NO_WOLFSSH_SERVER */
@@ -617,6 +625,13 @@ INLINE static int IsMessageAllowedClient(WOLFSSH *ssh, byte msg)
             return 0;
         }
     }
+    else {
+        if (msg >= MSGID_USERAUTH_RESTRICT && msg < MSGID_USERAUTH_LIMIT) {
+            WLOG(WS_LOG_DEBUG, "Message ID %u not allowed by client "
+                    "after user authentication", msg);
+            return 0;
+        }
+    }
     return 1;
 }
 #endif /* NO_WOLFSSH_CLIENT */