diff --git a/drivers/wdm/audio/hdaudbus/fdo.cpp b/drivers/wdm/audio/hdaudbus/fdo.cpp index 84686e3bd3e..64a0c6aef0b 100644 --- a/drivers/wdm/audio/hdaudbus/fdo.cpp +++ b/drivers/wdm/audio/hdaudbus/fdo.cpp @@ -222,6 +222,11 @@ HDA_InitCodec( DPRINT1("NodeId %u GroupType %x\n", NodeId, GroupType); if ((GroupType & FUNCTION_GROUP_NODETYPE_MASK) == FUNCTION_GROUP_NODETYPE_AUDIO) { + if (Entry->AudioGroupCount >= HDA_MAX_AUDIO_GROUPS) + { + DPRINT1("Too many audio groups in node %u. Skipping.\n", NodeId); + break; + } AudioGroup = (PHDA_CODEC_AUDIO_GROUP)AllocateItem(NonPagedPool, sizeof(HDA_CODEC_AUDIO_GROUP)); if (!AudioGroup) @@ -682,6 +687,7 @@ HDA_FDORemoveDevice( continue; } + ASSERT(CodecEntry->AudioGroupCount <= HDA_MAX_AUDIO_GROUPS); for (AFGIndex = 0; AFGIndex < CodecEntry->AudioGroupCount; AFGIndex++) { ChildPDO = CodecEntry->AudioGroups[AFGIndex]->ChildPDO; @@ -743,6 +749,7 @@ HDA_FDOQueryBusRelations( continue; Codec = DeviceExtension->Codecs[CodecIndex]; + ASSERT(Codec->AudioGroupCount <= HDA_MAX_AUDIO_GROUPS); for (AFGIndex = 0; AFGIndex < Codec->AudioGroupCount; AFGIndex++) { DeviceRelations->Objects[DeviceRelations->Count] = Codec->AudioGroups[AFGIndex]->ChildPDO; diff --git a/drivers/wdm/audio/hdaudbus/hdaudbus.cpp b/drivers/wdm/audio/hdaudbus/hdaudbus.cpp index 3e1ef9d526b..aa62a820bab 100644 --- a/drivers/wdm/audio/hdaudbus/hdaudbus.cpp +++ b/drivers/wdm/audio/hdaudbus/hdaudbus.cpp @@ -63,6 +63,7 @@ HDA_FdoPnp( { CodecEntry = FDODeviceExtension->Codecs[CodecIndex]; + ASSERT(CodecEntry->AudioGroupCount <= HDA_MAX_AUDIO_GROUPS); for (AFGIndex = 0; AFGIndex < CodecEntry->AudioGroupCount; AFGIndex++) { ChildDeviceExtension = static_cast(CodecEntry->AudioGroups[AFGIndex]->ChildPDO->DeviceExtension);