Error check for invalid TDS login request

amazon-aurora · Aug 28, 2024 · 4b4d697 · 4b4d697
1 parent 3605cae
commit 4b4d697
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/contrib/babelfishpg_tds/src/backend/tds/tdslogin.c b/contrib/babelfishpg_tds/src/backend/tds/tdslogin.c
@@ -759,7 +759,8 @@ FetchLoginRequest(LoginRequest request)
 		return STATUS_ERROR;
 
 	/* Check we indeed got the correct packet */
-	Assert(TdsCheckMessageType(TDS_LOGIN7));
+	if (!TdsCheckMessageType(TDS_LOGIN7))
+		return STATUS_ERROR;
 
 	/* fix the client version now */
 	request->clientProVersion = pg_bswap32(request->clientProVersion);