mctpd: update supports_vdm check

Updated supports_vdm check to be decided based on supported
PCIE/IANA message type instead of formatID.

Signed-off-by: rutaabhyankar <ruta.abhyankar@intel.com>

Author: Ruta01

src/mctpd.c

@@ -3163,10 +3163,8 @@ static int query_peer_properties(struct peer *peer)
 	}
 
 	for (unsigned int i = 0; i < peer->num_message_types; i++) {
-		if (peer->message_types[i] ==
-			    MCTP_GET_VDM_SUPPORT_IANA_FORMAT_ID ||
-		    peer->message_types[i] ==
-			    MCTP_GET_VDM_SUPPORT_PCIE_FORMAT_ID) {
+		if (peer->message_types[i] == MCTP_TYPE_VENDOR_IANA ||
+		    peer->message_types[i] == MCTP_TYPE_VENDOR_PCIE) {
 			supports_vdm = true;
 			break;
 		}

tests/test_mctpd.py

@@ -641,7 +641,7 @@ async def test_query_vdm_types(dbus, mctpd):
     """Test that VendorDefinedMessageTypes is queried and populated."""
     iface = mctpd.system.interfaces[0]
     vdm_support = [[0, 0x1234, 0x5678], [1, 0xABCDEF12, 0x3456]]
-    ep = Endpoint(iface, bytes([0x1E]), eid=15, vdm_msg_types=vdm_support)
+    ep = Endpoint(iface, bytes([0x1E]), eid=15, types=[0, 0x7E, 0x7F], vdm_msg_types=vdm_support)
     mctpd.network.add_endpoint(ep)
 
     mctp = await mctpd_mctp_iface_obj(dbus, iface)