windows-nt/Source/XPSP1/NT/multimedia/media/audiosrv/mme.cpp

986 lines
29 KiB
C++
Raw Normal View History

2020-09-26 03:20:57 -05:00
/* mme.cpp
* Handles pnp, etc. for MME APIs
* Created by FrankYe on 2/14/2001
* Copyright (c) 2001-2001 Microsoft Corporation
*/
#include <nt.h>
#include <ntrtl.h>
#include <nturtl.h>
#include <windows.h>
#include <regstr.h>
#include <mmsystem.h>
#include <mmsysp.h>
#include <ks.h>
#include <ksmedia.h>
#include <setupapi.h>
#include "debug.h"
#include "reg.h"
#include "service.h"
#include "audiosrv.h"
//=============================================================================
//=== file scope constants ===
//=============================================================================
#define REGSTR_VAL_SETUPPREFERREDAUDIODEVICES TEXT("SetupPreferredAudioDevices")
#define REGSTR_VAL_SETUPPREFERREDAUDIODEVICESCOUNT TEXT("SetupPreferredAudioDevicesCount")
#define PNPINFOSIZE (256 * 1024)
//=============================================================================
//=== file scope globals ===
//=============================================================================
RTL_RESOURCE PnpInfoResource;
BOOL gfPnpInfoResource = FALSE;
HANDLE hPnpInfo = NULL;
PMMPNPINFO pPnpInfo = NULL;
//=============================================================================
//=== security helpers ===
//=============================================================================
PSECURITY_DESCRIPTOR BuildSecurityDescriptor(DWORD AccessMask)
{
PSECURITY_DESCRIPTOR pSd;
PSID pSidSystem;
PSID pSidEveryone;
PACL pDacl;
ULONG cbDacl;
BOOL fSuccess;
BOOL f;
SID_IDENTIFIER_AUTHORITY AuthorityNt = SECURITY_NT_AUTHORITY;
SID_IDENTIFIER_AUTHORITY AuthorityWorld = SECURITY_WORLD_SID_AUTHORITY;
fSuccess = FALSE;
pSd = HeapAlloc(hHeap, 0, SECURITY_DESCRIPTOR_MIN_LENGTH);
if (pSd)
{
if (InitializeSecurityDescriptor(pSd, SECURITY_DESCRIPTOR_REVISION))
{
if (AllocateAndInitializeSid(&AuthorityNt, 1, SECURITY_LOCAL_SYSTEM_RID, 0, 0, 0, 0, 0, 0, 0, &pSidSystem))
{
ASSERT(IsValidSid(pSidSystem));
if (AllocateAndInitializeSid(&AuthorityWorld, 1, SECURITY_WORLD_RID, 0, 0, 0, 0, 0, 0, 0, &pSidEveryone))
{
ASSERT(IsValidSid(pSidEveryone));
cbDacl = sizeof(ACL) +
2 * (sizeof(ACCESS_ALLOWED_ACE) - sizeof(DWORD)) +
GetLengthSid(pSidSystem) +
GetLengthSid(pSidEveryone);
pDacl = (PACL)HeapAlloc(hHeap, 0, cbDacl);
if (pDacl) {
if (InitializeAcl(pDacl, cbDacl, ACL_REVISION))
{
if (AddAccessAllowedAce(pDacl, ACL_REVISION, GENERIC_ALL, pSidSystem))
{
if (AddAccessAllowedAce(pDacl, ACL_REVISION, AccessMask, pSidEveryone))
{
if (SetSecurityDescriptorDacl(pSd, TRUE, pDacl, FALSE))
{
fSuccess = TRUE;
} else {
dprintf(TEXT("BuildSD: SetSecurityDescriptorDacl failed\n"));
}
} else {
dprintf(TEXT("BuildSD: AddAccessAlloweAce for Everyone failed\n"));
}
} else {
dprintf(TEXT("BuildSD: AddAccessAllowedAce for System failed\n"));
}
} else {
dprintf(TEXT("BuildSD: InitializeAcl failed\n"));
}
if (!fSuccess) {
f = HeapFree(hHeap, 0, pDacl);
ASSERT(f);
}
}
FreeSid(pSidEveryone);
} else {
dprintf(TEXT("BuildSD: AllocateAndInitizeSid failed for Everyone\n"));
}
FreeSid(pSidSystem);
} else {
dprintf(TEXT("BuildSD: AllocateAndInitizeSid failed for System\n"));
}
} else {
dprintf(TEXT("BuildSD: InitializeSecurityDescriptor failed\n"));
}
if (!fSuccess) {
f = HeapFree(hHeap, 0, pSd);
ASSERT(f);
}
}
return fSuccess ? pSd : NULL;
}
void DestroySecurityDescriptor(PSECURITY_DESCRIPTOR pSd)
{
PACL pDacl;
BOOL fDaclPresent;
BOOL fDaclDefaulted;
BOOL f;
if (GetSecurityDescriptorDacl(pSd, &fDaclPresent, &pDacl, &fDaclDefaulted))
{
if (fDaclPresent)
{
f = HeapFree(hHeap, 0, pDacl);
ASSERT(f);
}
} else {
dprintf(TEXT("DestroySD: GetSecurityDescriptorDacl failed\n"));
}
f = HeapFree(hHeap, 0, pSd);
ASSERT(f);
return;
}
//=============================================================================
//=== ===
//=============================================================================
//--------------------------------------------------------------------------;
//
// PTSTR BroadcastWinmmDeviceChange
//
// Arguments:
//
// Return value:
//
// History:
// 11/9/98 FrankYe Created
// 2/15/2001 FrankYe Moved from winmm to audiosrv
//
//--------------------------------------------------------------------------;
void BroadcastWinmmDeviceChange(void)
{
static UINT uWinmmDeviceChange = 0;
if (!uWinmmDeviceChange) {
uWinmmDeviceChange = RegisterWindowMessage(WINMMDEVICECHANGEMSGSTRING);
// dprintf(TEXT("BroadcastWinmmDeviceChange: WINMMDEVICECHANGEMSG = %d\n"), uWinmmDeviceChange);
if (!uWinmmDeviceChange) {
dprintf(TEXT("BroadcastWinmmDeviceChange: RegisterWindowMessage failed!\n"));
}
}
if (uWinmmDeviceChange) {
DWORD dwRecipients = BSM_APPLICATIONS | BSM_ALLDESKTOPS;
long result;
// dprintf(TEXT("BroadcastWinmmDeviceChange: BroadcastSystemMessage\n"));
result = BroadcastSystemMessage(BSF_POSTMESSAGE, &dwRecipients,
uWinmmDeviceChange, 0, 0);
if (result < 0) {
dprintf(TEXT("BroadcastWinmmDeviceChange: BroadcastSystemMessage failed\n"));
}
}
return;
}
//--------------------------------------------------------------------------;
//
// PTSTR MakeRendererDeviceInstanceIdFromDeviceInterface
//
// Arguments:
//
// Return value:
//
// History:
// 11/9/98 FrankYe Created
//
//--------------------------------------------------------------------------;
PTSTR MakeRendererDeviceInstanceIdFromDeviceInterface(PWSTR DeviceInterface)
{
PTSTR DeviceInstanceId;
HDEVINFO hdi;
DWORD dwLastError;
// dprintf(TEXT("MRDIIFDI: DeviceInterface=%ls\n"), DeviceInterface);
DeviceInstanceId = NULL;
hdi = SetupDiCreateDeviceInfoList(NULL, NULL);
if (INVALID_HANDLE_VALUE != hdi)
{
SP_DEVICE_INTERFACE_DATA DeviceInterfaceData;
// dprintf(TEXT("MRDIIFDI: Created empty DeviceInfoList\n"));
DeviceInterfaceData.cbSize = sizeof(DeviceInterfaceData);
if (SetupDiOpenDeviceInterface(hdi, DeviceInterface, 0,
&DeviceInterfaceData))
{
SP_DEVINFO_DATA DeviceInfoData;
DWORD cbDeviceInterfaceDetail;
// dprintf(TEXT("MRDIIFDI: Opened DeviceInterface\n"));
DeviceInfoData.cbSize = sizeof(DeviceInfoData);
if (SetupDiGetDeviceInterfaceDetail(hdi, &DeviceInterfaceData, NULL, 0, &cbDeviceInterfaceDetail, &DeviceInfoData) ||
(ERROR_INSUFFICIENT_BUFFER == GetLastError()))
{
DWORD cchDeviceInstanceId;
// dprintf(TEXT("MRDIIFDI: Got DeviceInfoData\n"));
if (SetupDiGetDeviceInstanceId(hdi, &DeviceInfoData, NULL, 0, &cchDeviceInstanceId) ||
ERROR_INSUFFICIENT_BUFFER == GetLastError())
{
// dprintf(TEXT("MRDIIFDI: DeviceInstanceId is %d characters\n"), cchDeviceInstanceId);
DeviceInstanceId = (PTSTR)HeapAlloc(hHeap, 0, cchDeviceInstanceId * sizeof(DeviceInstanceId[0]));
if (DeviceInstanceId)
{
// dprintf(TEXT("MRDIIFDI: Allocated storage for DeviceInstanceId\n"));
if (SetupDiGetDeviceInstanceId(hdi, &DeviceInfoData,
DeviceInstanceId,
cchDeviceInstanceId, NULL))
{
// dprintf(TEXT("MRDIIFDI: DeviceInstanceId=%ls\n"), DeviceInstanceId);
} else {
BOOL f;
dwLastError = GetLastError();
dprintf(TEXT("MRDIIFDI: Couldn't query DeviceInstanceId, LastError=%d\n"), dwLastError);
f = HeapFree(hHeap, 0, DeviceInstanceId);
DeviceInstanceId = NULL;
ASSERT(f);
}
} else {
dprintf(TEXT("MRDIIFDI: Could not allocate storage for DeviceInstanceId\n"));
}
} else {
dwLastError = GetLastError();
dprintf(TEXT("MRDIIFDI: Couldn't query size of DeviceInstanceId, LastError=%d\n"), dwLastError);
}
} else {
dwLastError = GetLastError();
dprintf(TEXT("MRDIIFDI: SetupDiGetDeviceInterfaceDetail failed LastError=%d\n"), dwLastError);
}
} else {
dwLastError = GetLastError();
dprintf(TEXT("MRDIIFDI: SetupDiOpenDeviceInterface failed, LastError=%d\n"), dwLastError);
}
if (!SetupDiDestroyDeviceInfoList(hdi))
{
dwLastError = GetLastError();
dprintf(TEXT("MRDIIFDI: SetupDiDestroyDeviceInfoList failed, LastError=%d\n"), dwLastError);
}
} else {
dwLastError = GetLastError();
dprintf(TEXT("MRDIIFDI: SetupDiCreateDeviceInfoList failed, LastError=%d\n"), dwLastError);
}
return DeviceInstanceId;
}
//--------------------------------------------------------------------------;
//
// MigrageNewDeviceInterfaceSetup
//
// Arguments:
//
// Return value:
//
// History:
// 1/19/99 FrankYe Created
//
//--------------------------------------------------------------------------;
void MigrateNewDeviceInterfaceSetup(PMMDEVICEINTERFACEINFO pdii, DWORD CurrentSetupCount)
{
HDEVINFO hdi;
DWORD dwLastError;
DWORD SetupCount;
SetupCount = 0;
hdi = SetupDiCreateDeviceInfoList(NULL, NULL);
if (INVALID_HANDLE_VALUE != hdi)
{
SP_DEVICE_INTERFACE_DATA DeviceInterfaceData;
DeviceInterfaceData.cbSize = sizeof(DeviceInterfaceData);
if (SetupDiOpenDeviceInterface(hdi, pdii->szName, 0, &DeviceInterfaceData))
{
HKEY hkeyDeviceInterface;
hkeyDeviceInterface = SetupDiCreateDeviceInterfaceRegKeyW(hdi, &DeviceInterfaceData, 0, KEY_SET_VALUE | KEY_QUERY_VALUE, NULL, NULL);
if (INVALID_HANDLE_VALUE != hkeyDeviceInterface)
{
LONG result;
result = RegQueryDwordValue(hkeyDeviceInterface, REGSTR_VAL_SETUPPREFERREDAUDIODEVICESCOUNT, &SetupCount);
if ((ERROR_SUCCESS != result) || (SetupCount < CurrentSetupCount))
{
SetupCount = CurrentSetupCount;
if (ERROR_SUCCESS == RegSetDwordValue(hkeyDeviceInterface, REGSTR_VAL_SETUPPREFERREDAUDIODEVICESCOUNT, SetupCount))
{
// dprintf(TEXT("MNDIS: Success\n"));
} else {
dwLastError = GetLastError();
dprintf(TEXT("MNDIS: RegSetValueEx failed, LastError=%d\n"), dwLastError);
}
}
result = RegCloseKey(hkeyDeviceInterface);
ASSERT(ERROR_SUCCESS == result);
} else {
dwLastError = GetLastError();
dprintf(TEXT("MNDIS: SetupDiCreateDeviceInterfaceRegKey failed, LastError=%d\n"), dwLastError);
}
} else {
dwLastError = GetLastError();
dprintf(TEXT("MRDIIFDI: SetupDiOpenDeviceInterface failed, LastError=%d\n"), dwLastError);
}
if (!SetupDiDestroyDeviceInfoList(hdi))
{
dwLastError = GetLastError();
dprintf(TEXT("MRDIIFDI: SetupDiDestroyDeviceInfoList failed, LastError=%d\n"), dwLastError);
}
} else {
dwLastError = GetLastError();
dprintf(TEXT("MRDIIFDI: SetupDiCreateDeviceInfoList failed, LastError=%d\n"), dwLastError);
}
pdii->SetupPreferredAudioCount = SetupCount;
return;
}
//--------------------------------------------------------------------------;
//
// MigrateNewDeviceInstanceSetup
//
// Arguments:
//
// Return value:
//
// History:
// 11/9/98 FrankYe Created
//
//--------------------------------------------------------------------------;
void MigrateNewDeviceInstanceSetup(PTSTR DeviceInstanceId, PDWORD pSetupCountOut)
{
HDEVINFO hdi;
DWORD dwLastError;
*pSetupCountOut = 0;
hdi = SetupDiCreateDeviceInfoList(NULL, NULL);
if (INVALID_HANDLE_VALUE != hdi)
{
SP_DEVINFO_DATA DeviceInfoData;
// dprintf(TEXT("MNDS: Created empty DeviceInfoList\n"));
DeviceInfoData.cbSize = sizeof(DeviceInfoData);
if (SetupDiOpenDeviceInfoW(hdi, DeviceInstanceId, NULL, 0, &DeviceInfoData))
{
HKEY hkeyDriver;
hkeyDriver = SetupDiOpenDevRegKey(hdi, &DeviceInfoData, DICS_FLAG_GLOBAL, 0, DIREG_DRV, KEY_QUERY_VALUE | KEY_SET_VALUE);
if (INVALID_HANDLE_VALUE != hkeyDriver) {
HKEY hkeyMmeDrivers;
dwLastError = RegOpenKeyEx(hkeyDriver, TEXT("Drivers"), 0, KEY_QUERY_VALUE, &hkeyMmeDrivers);
if (ERROR_SUCCESS == dwLastError)
{
DWORD fNewInstall;
BOOL fSetupPreferredAudioDevices;
DWORD cbfSetupPreferredAudioDevices;
DWORD SetupCount;
// Read the driver's existing setup count. If it doesn't exist
// then this is a new install.
dwLastError = RegQueryDwordValue(hkeyDriver, REGSTR_VAL_SETUPPREFERREDAUDIODEVICESCOUNT, &SetupCount);
if (ERROR_SUCCESS == dwLastError) {
fNewInstall = FALSE;
} else if (ERROR_FILE_NOT_FOUND == dwLastError) {
fNewInstall = TRUE;
SetupCount = 0;
dwLastError = ERROR_SUCCESS;
} else {
fNewInstall = FALSE;
SetupCount = 0;
}
if (ERROR_SUCCESS == dwLastError)
{
// Read the driver's SetupPreferredAudioDevices flag.
cbfSetupPreferredAudioDevices = sizeof(fSetupPreferredAudioDevices);
dwLastError = RegQueryValueEx(hkeyDriver, REGSTR_VAL_SETUPPREFERREDAUDIODEVICES, NULL, NULL, (PBYTE)&fSetupPreferredAudioDevices, &cbfSetupPreferredAudioDevices);
if (ERROR_FILE_NOT_FOUND == dwLastError) {
fSetupPreferredAudioDevices = FALSE;
dwLastError = ERROR_SUCCESS;
}
// If this is a new install AND the driver .inf set the
// fSetupPreferredAudioDevices flag, then let's try to
// increment the machine setupcount and write the driver
// setupcount.
if ((ERROR_SUCCESS == dwLastError) && fNewInstall && fSetupPreferredAudioDevices)
{
HKEY hkeySetupPreferredAudioDevices;
dwLastError = RegCreateKey(HKEY_LOCAL_MACHINE, REGSTR_PATH_MEDIARESOURCES TEXT("\\") REGSTR_VAL_SETUPPREFERREDAUDIODEVICES, &hkeySetupPreferredAudioDevices);
if (ERROR_SUCCESS == dwLastError)
{
dwLastError = RegQueryDwordValue(hkeySetupPreferredAudioDevices, REGSTR_VAL_SETUPPREFERREDAUDIODEVICESCOUNT, &SetupCount);
if (ERROR_FILE_NOT_FOUND == dwLastError) {
SetupCount = 0;
dwLastError = ERROR_SUCCESS;
}
if (ERROR_SUCCESS == dwLastError)
{
SetupCount++;
dwLastError = RegSetDwordValue(hkeySetupPreferredAudioDevices, REGSTR_VAL_SETUPPREFERREDAUDIODEVICESCOUNT, SetupCount);
if (ERROR_SUCCESS != dwLastError) dprintf(TEXT("MNDIS: Couldn't set count\n"));
}
RegCloseKey(hkeySetupPreferredAudioDevices);
} else {
dprintf(TEXT("MNDIS: Couldn't create hklm\\...\\SetupPreferredAudioDevices\n"));
}
}
if (ERROR_SUCCESS == dwLastError) {
// We've successfully read, incremented, and written the
// setup version to HKLM, or we've done nothing because we
// didn't have to.
if (fNewInstall) RegSetDwordValue(hkeyDriver, REGSTR_VAL_SETUPPREFERREDAUDIODEVICESCOUNT, SetupCount);
if (fSetupPreferredAudioDevices) RegDeleteValue(hkeyDriver, REGSTR_VAL_SETUPPREFERREDAUDIODEVICES);
}
}
// Return the SetupCount for the driver
*pSetupCountOut = SetupCount;
RegCloseKey(hkeyMmeDrivers);
}
RegCloseKey(hkeyDriver);
} else {
dwLastError = GetLastError();
dprintf(TEXT("MNDS: SetupDiCreateDevRegKey failed, LastError=%d\n"), dwLastError);
}
} else {
dwLastError = GetLastError();
dprintf(TEXT("MNDS: SetupDiOpenDeviceInfo failed, LastError=%d\n"), dwLastError);
}
if (!SetupDiDestroyDeviceInfoList(hdi))
{
dwLastError = GetLastError();
dprintf(TEXT("MNDS: SetupDiDestroyDeviceInfoList failed, LastError=%d\n"), dwLastError);
}
} else {
dwLastError = GetLastError();
dprintf(TEXT("MNDS: SetupDiCreateDeviceInfoList failed, LastError=%d\n"), dwLastError);
}
return;
}
//------------------------------------------------------------------------------
//
//
// MigrateAutoSetupPreferredAudio
//
//
//------------------------------------------------------------------------------
void MigrateAutoSetupPreferredAudio(PMMDEVICEINTERFACEINFO pdii)
{
PTSTR pstrRendererDeviceInstanceId;
DWORD SetupCount;
SetupCount = 0;
pstrRendererDeviceInstanceId = MakeRendererDeviceInstanceIdFromDeviceInterface(pdii->szName);
if (pstrRendererDeviceInstanceId) {
BOOL f;
MigrateNewDeviceInstanceSetup(pstrRendererDeviceInstanceId, &SetupCount);
f = HeapFree(hHeap, 0, pstrRendererDeviceInstanceId);
ASSERT(f);
}
MigrateNewDeviceInterfaceSetup(pdii, SetupCount);
return;
}
PMMDEVICEINTERFACEINFO pnpServerInstallDevice
(
PCTSTR pszDeviceInterface,
BOOL fRemove
)
{
PMMDEVICEINTERFACEINFO pdii;
PWSTR pszDev;
UINT ii;
if (NULL == pPnpInfo)
{
dprintf(TEXT("pnpServerInstallDevice called at bad time\n"));
ASSERT(FALSE);
return NULL;
}
pdii = (PMMDEVICEINTERFACEINFO)&(pPnpInfo[1]);
pdii = (PMMDEVICEINTERFACEINFO)PAD_POINTER(pdii);
for (ii = pPnpInfo->cDevInterfaces; ii; ii--)
{
// Searching for the device interface...
pszDev = (PWSTR)(&(pdii->szName[0]));
if (0 == lstrcmpi(pszDev, pszDeviceInterface))
{
break;
}
pdii = (PMMDEVICEINTERFACEINFO)(pszDev + lstrlenW(pszDev) + 1);
pdii = (PMMDEVICEINTERFACEINFO)PAD_POINTER(pdii);
}
// Getting the current settings...
if (0 == ii)
{
PMMDEVICEINTERFACEINFO pdiiNext;
SIZE_T sizePnpInfo;
// Does not exist, create it, first ensuring there is enough room
pdiiNext = (PMMDEVICEINTERFACEINFO)(pdii->szName + ((lstrlenW(pszDeviceInterface) + 1)));
pdiiNext = (PMMDEVICEINTERFACEINFO)PAD_POINTER(pdiiNext);
sizePnpInfo = ((PBYTE)pdiiNext) - ((PBYTE)pPnpInfo);
if (sizePnpInfo < PNPINFOSIZE)
{
// dprintf(TEXT("pnpServerInstallDevice: note: sizePnpInfo = %p\n", sizePnpInfo);
pdii->cPnpEvents = 0;
pdii->fdwInfo = 0L;
pszDev = (PWSTR)(&(pdii->szName[0]));
lstrcpy(pszDev, pszDeviceInterface);
pPnpInfo->cDevInterfaces++;
pPnpInfo->cbSize = (DWORD)sizePnpInfo;
} else {
dprintf(TEXT("pnpServerInstallDevice RAN OUT OF PNPINO STORAGE!!!\n"));
pdii = NULL;
}
}
else
{
// Already exists, increment the event count.
pdii->cPnpEvents++;
}
// Set or clear the "removed" bit.
if (pdii)
{
if (fRemove)
{
pdii->fdwInfo |= MMDEVICEINFO_REMOVED;
// dprintf("pnpServerInstallDevice removed [%ls]\n", pszDeviceInterface);
}
else
{
pdii->fdwInfo &= (~MMDEVICEINFO_REMOVED);
// dprintf("pnpServerInstallDevice added [%ls]\n", pszDeviceInterface);
}
}
return pdii;
}
BOOL PnpInfoEnum
(
void
)
{
UINT cDevs = 0;
HDEVINFO hDevInfo;
BOOL fEnum;
DWORD ii, dw;
DWORD cbSize;
GUID guidClass = KSCATEGORY_AUDIO;
SP_DEVICE_INTERFACE_DATA did;
SP_DEVINFO_DATA DevInfoData;
PSP_DEVICE_INTERFACE_DETAIL_DATA pdidd;
cbSize = 300 * sizeof(TCHAR) + sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);
pdidd = (PSP_DEVICE_INTERFACE_DETAIL_DATA)HeapAlloc(hHeap, 0, cbSize);
if (NULL == pdidd)
{
HeapFree(hHeap, 0, pdidd);
return FALSE;
}
pdidd->cbSize = sizeof(SP_DEVICE_INTERFACE_DETAIL_DATA);
hDevInfo = SetupDiGetClassDevs(
&guidClass,
NULL,
NULL,
DIGCF_DEVICEINTERFACE | DIGCF_PRESENT);
if (INVALID_HANDLE_VALUE == hDevInfo)
{
dprintf(TEXT("SetupDiGetClasDevs failed [0x%08lx].\n"), GetLastError());
HeapFree(hHeap, 0, pdidd);
return FALSE;
}
ZeroMemory(&did, sizeof(did));
did.cbSize = sizeof(did);
ZeroMemory(&DevInfoData, sizeof(SP_DEVINFO_DATA));
DevInfoData.cbSize = sizeof(SP_DEVINFO_DATA);
DevInfoData.ClassGuid = KSCATEGORY_AUDIO;
for (ii = 0; ;ii++, cDevs++)
{
PMMDEVICEINTERFACEINFO pdii;
fEnum = SetupDiEnumDeviceInterfaces(
hDevInfo,
NULL,
&guidClass,
ii,
&did);
if (!fEnum)
{
break;
}
dw = 0;
fEnum = SetupDiGetDeviceInterfaceDetail(
hDevInfo,
&did,
pdidd,
cbSize,
&dw,
&DevInfoData);
if (!fEnum)
{
dprintf(TEXT("SetupDiGetDeviceInterfaceDetail failed (0x%08lx).\n"), GetLastError());
break;
}
// dprintf(TEXT("PnpInfoEnum: Enumerated[%ls]\n"), pdidd->DevicePath);
pdii = pnpServerInstallDevice(pdidd->DevicePath, FALSE);
if (pdii) MigrateAutoSetupPreferredAudio(pdii);
}
if (!SetupDiDestroyDeviceInfoList(hDevInfo))
{
ASSERT(!"wdmEnumerateInstalledDevices: SetupDiDestroyDeviceInfoList failed.");
}
HeapFree(hHeap, 0, pdidd);
SetLastError(ERROR_SUCCESS);
return TRUE;
} // PnpInfoEnum()
BOOL InitializePnpInfo
(
void
)
{
SECURITY_ATTRIBUTES sa;
PSECURITY_DESCRIPTOR pSd;
BOOL result;
ASSERT(!gfPnpInfoResource);
ASSERT(!hPnpInfo);
ASSERT(!pPnpInfo);
result = FALSE;
__try {
RtlInitializeResource(&PnpInfoResource);
gfPnpInfoResource = TRUE;
} __except(EXCEPTION_EXECUTE_HANDLER) {
gfPnpInfoResource = FALSE;;
}
if (!gfPnpInfoResource) return FALSE;
pSd = BuildSecurityDescriptor(FILE_MAP_READ);
if (pSd)
{
sa.nLength = sizeof(SECURITY_ATTRIBUTES);
sa.lpSecurityDescriptor = pSd;
sa.bInheritHandle = FALSE;
hPnpInfo = CreateFileMapping(GetCurrentProcess(), &sa, PAGE_READWRITE, 0, PNPINFOSIZE, MMGLOBALPNPINFONAME);
DestroySecurityDescriptor(pSd);
if (hPnpInfo)
{
pPnpInfo = (PMMPNPINFO)MapViewOfFile(hPnpInfo, FILE_MAP_WRITE, 0, 0, 0);
if (pPnpInfo)
{
ZeroMemory(pPnpInfo, PNPINFOSIZE);
pPnpInfo->cbSize = sizeof(MMPNPINFO);
pPnpInfo->cPnpEvents = 0;
result = TRUE;
} else {
dprintf(TEXT("InitializePnpInfo: MapViewOfFile failed!\n"));
}
} else {
dprintf(TEXT("InitializePnpInfo: CreateFileMappingFailed!\n"));
}
} else {
dprintf(TEXT("InitializePnpInfo: BuildSecurityDescriptor failed!\n"));
}
if (!result)
{
if (pPnpInfo) UnmapViewOfFile(pPnpInfo);
if (hPnpInfo) CloseHandle(hPnpInfo);
if (gfPnpInfoResource) RtlDeleteResource(&PnpInfoResource);
pPnpInfo = NULL;
hPnpInfo = NULL;
gfPnpInfoResource = FALSE;
}
return result;
} // InitializePnpInfo()
void DeletePnpInfo(void)
{
ASSERT(gfPnpInfoResource);
ASSERT(hPnpInfo);
ASSERT(pPnpInfo);
UnmapViewOfFile(pPnpInfo);
CloseHandle(hPnpInfo);
RtlDeleteResource(&PnpInfoResource);
pPnpInfo = NULL;
hPnpInfo = NULL;
gfPnpInfoResource = FALSE;
return;
}
//=============================================================================
//=== rpc functions ===
//=============================================================================
long s_wdmDriverOpenDrvRegKey(IN DWORD dwProcessId, IN LPCTSTR DeviceInterface, IN ULONG samDesired, OUT RHANDLE *phkeyClient)
{
HDEVINFO hdi;
HKEY hkey;
RPC_STATUS status;
// We impersonate the client while calling setupapi so that we are sure
// the client actually has access to open the driver reg key
status = RpcImpersonateClient(NULL);
if (status) return status;
hdi = SetupDiCreateDeviceInfoList(NULL, NULL);
if (INVALID_HANDLE_VALUE != hdi)
{
SP_DEVICE_INTERFACE_DATA DeviceInterfaceData;
DeviceInterfaceData.cbSize = sizeof(DeviceInterfaceData);
if (SetupDiOpenDeviceInterface(hdi, DeviceInterface, 0,
&DeviceInterfaceData))
{
SP_DEVINFO_DATA DeviceInfoData;
DWORD cbDeviceInterfaceDetail;
DeviceInfoData.cbSize = sizeof(DeviceInfoData);
if (SetupDiGetDeviceInterfaceDetail(hdi, &DeviceInterfaceData, NULL, 0, &cbDeviceInterfaceDetail, &DeviceInfoData) ||
(ERROR_INSUFFICIENT_BUFFER == GetLastError()))
{
hkey = SetupDiOpenDevRegKey(hdi, &DeviceInfoData, DICS_FLAG_GLOBAL, 0, DIREG_DRV, samDesired);
if (INVALID_HANDLE_VALUE != hkey)
{
status = ERROR_SUCCESS;
} else {
status = GetLastError();
dprintf(TEXT("s_wdmDriverOpenDrvRegKey: SetupDiOpenDevRegKey failed, Error=%d\n"), status);
}
} else {
status = GetLastError();
dprintf(TEXT("s_wdmDriverOpenDrvRegKey: SetupDiGetDeviceInterfaceDetail failed, Error=%d\n"), status);
}
} else {
status = GetLastError();
dprintf(TEXT("s_wdmDriverOpenDrvRegKey: SetupDiOpenDeviceInterface failed, Error=%d\n"), status);
}
SetupDiDestroyDeviceInfoList(hdi);
} else {
status = GetLastError();
dprintf(TEXT("s_wdmDriverOpenDrvRegKey: SetupDiCreateDeviceInfoList failed, Error=%d\n"), status);
}
// We stop impersonating here because the remaining operations
// should not depend on the client's privileges.
RpcRevertToSelf();
if (ERROR_SUCCESS == status)
{
HANDLE hClientProcess;
hClientProcess = OpenProcess(PROCESS_DUP_HANDLE, FALSE, dwProcessId);
if (hClientProcess)
{
HANDLE hkeyClient;
if (DuplicateHandle(GetCurrentProcess(), hkey, hClientProcess, &hkeyClient, 0, FALSE, DUPLICATE_SAME_ACCESS))
{
// dprintf(TEXT("s_wdmDriverOpenDrvRegKey hkeyClient=%p\n"), hkeyClient);
*phkeyClient = (RHANDLE)hkeyClient;
} else {
status = GetLastError();
dprintf(TEXT("s_wdmDriverOpenDrvRegKey: DuplicateHandle failed, Error=%d\n"), status);
}
CloseHandle(hClientProcess);
} else {
status = GetLastError();
dprintf(TEXT("s_wdmDriverOpenDrvRegKey: OpenProcess failed, Error=%d\n"), status);
}
RegCloseKey(hkey);
}
return status;
}
void s_winmmAdvisePreferredDeviceChange(void)
{
// dprintf(TEXT("s_winmmAdvisePreferredDeviceChange\n"));
ASSERT(pPnpInfo);
InterlockedIncrement(&pPnpInfo->cPreferredDeviceChanges);
BroadcastWinmmDeviceChange();
return;
}
long s_winmmGetPnpInfo(OUT LONG *pcbPnpInfo, OUT BYTE **ppPnpInfoOut)
{
static BOOL fEnumDone = FALSE;
PBYTE pPnpInfoOut;
LONG result;
// dprintf(TEXT("s_winmmGetPnpInfo\n"));
ASSERT(pPnpInfo);
RtlAcquireResourceShared(&PnpInfoResource, TRUE);
pPnpInfoOut = (PBYTE)HeapAlloc(hHeap, 0, pPnpInfo->cbSize);
if (pPnpInfoOut)
{
CopyMemory(pPnpInfoOut, pPnpInfo, pPnpInfo->cbSize);
*pcbPnpInfo = pPnpInfo->cbSize;
*ppPnpInfoOut = pPnpInfoOut;
result = NO_ERROR;
} else {
// ISSUE-2001/02/15-FrankYe Do we need to zero the out params?
result = ERROR_OUTOFMEMORY;
}
RtlReleaseResource(&PnpInfoResource);
return result;
}
//=============================================================================
//=== pnp interface handlers ===
//=============================================================================
void MME_AudioInterfaceArrival(PCTSTR DeviceInterface)
{
PMMDEVICEINTERFACEINFO pdii;
// dprintf(TEXT("MME_AudioInterfaceArrival\n"));
RtlAcquireResourceExclusive(&PnpInfoResource, TRUE);
pdii = pnpServerInstallDevice(DeviceInterface, FALSE);
if (pdii) MigrateAutoSetupPreferredAudio(pdii);
InterlockedIncrement(&pPnpInfo->cPnpEvents);
InterlockedIncrement(&pPnpInfo->cPreferredDeviceChanges);
RtlReleaseResource(&PnpInfoResource);
if (pdii) BroadcastWinmmDeviceChange();
return;
}
void MME_AudioInterfaceRemove(PCTSTR DeviceInterface)
{
PMMDEVICEINTERFACEINFO pdii;
// dprintf(TEXT("MME_AudioInterfaceRemove\n"));
RtlAcquireResourceExclusive(&PnpInfoResource, TRUE);
pdii = pnpServerInstallDevice(DeviceInterface, TRUE);
InterlockedIncrement(&pPnpInfo->cPnpEvents);
InterlockedIncrement(&pPnpInfo->cPreferredDeviceChanges);
RtlReleaseResource(&PnpInfoResource);
if (pdii) BroadcastWinmmDeviceChange();
return;
}
LONG MME_ServiceStart(void)
{
ASSERT(pPnpInfo);
// dprintf(TEXT("MME_ServiceStart\n"));
RtlAcquireResourceExclusive(&PnpInfoResource, TRUE);
if (!PnpInfoEnum()) dprintf(TEXT("MME_ServiceStart: PnpInfoEnum failed!\n"));
InterlockedIncrement(&pPnpInfo->cPnpEvents);
RtlReleaseResource(&PnpInfoResource);
return NO_ERROR;
}
//=============================================================================
//=== DLL attach/detach ===
//=============================================================================
BOOL MME_DllProcessAttach(void)
{
return InitializePnpInfo();
}
void MME_DllProcessDetach(void)
{
return DeletePnpInfo();
}