125 lines
2.6 KiB
C++
125 lines
2.6 KiB
C++
// DrvSearchSet.cpp : Implementation of CDrvSearchSet
|
|
#include "stdafx.h"
|
|
#include "DevCon2.h"
|
|
#include "DrvSearchSet.h"
|
|
#include "Drivers.h"
|
|
#include "Driver.h"
|
|
#include "Device.h"
|
|
#include "DevInfoSet.h"
|
|
#include "DeviceConsole.h"
|
|
|
|
/////////////////////////////////////////////////////////////////////////////
|
|
// CDrvSearchSet
|
|
|
|
CDrvSearchSet::~CDrvSearchSet()
|
|
{
|
|
if(pActualDevice) {
|
|
pActualDevice->Release();
|
|
}
|
|
if(pTempDevice) {
|
|
pTempDevice->Release();
|
|
}
|
|
}
|
|
|
|
HRESULT CDrvSearchSet::Init(CDevice *device,DWORD searchType)
|
|
{
|
|
if(pActualDevice) {
|
|
pActualDevice->Release();
|
|
pActualDevice = NULL;
|
|
}
|
|
if(pTempDevice) {
|
|
pTempDevice->Release();
|
|
pTempDevice = NULL;
|
|
}
|
|
//
|
|
// determine machine name that device resides on
|
|
//
|
|
HDEVINFO hDevInfo = device->GetDevInfoSet();
|
|
if(hDevInfo == INVALID_HANDLE_VALUE) {
|
|
return E_UNEXPECTED;
|
|
}
|
|
SP_DEVINFO_LIST_DETAIL_DATA devInfoListDetail;
|
|
HRESULT hr;
|
|
|
|
devInfoListDetail.cbSize = sizeof(devInfoListDetail);
|
|
if(!SetupDiGetDeviceInfoListDetail(hDevInfo,&devInfoListDetail)) {
|
|
DWORD Err = GetLastError();
|
|
return HRESULT_FROM_SETUPAPI(Err);
|
|
}
|
|
|
|
//
|
|
// create empty device info set for same machine as device
|
|
//
|
|
hDevInfo = SetupDiCreateDeviceInfoListEx(NULL,
|
|
NULL,
|
|
devInfoListDetail.RemoteMachineName[0]
|
|
? devInfoListDetail.RemoteMachineName
|
|
: NULL,
|
|
NULL);
|
|
if(hDevInfo == INVALID_HANDLE_VALUE) {
|
|
DWORD Err = GetLastError();
|
|
return HRESULT_FROM_SETUPAPI(Err);
|
|
}
|
|
|
|
CComObject<CDevInfoSet> *pDevInfoSet = NULL;
|
|
hr = CComObject<CDevInfoSet>::CreateInstance(&pDevInfoSet);
|
|
if(FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
CComPtr<IDevInfoSet> pDevInfoSetPtr = pDevInfoSet;
|
|
hr = pDevInfoSet->Init(hDevInfo);
|
|
if(FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
|
|
//
|
|
// create a single entry device in new set
|
|
//
|
|
CComObject<CDevice> *pDevice = NULL;
|
|
hr = CComObject<CDevice>::CreateInstance(&pDevice);
|
|
if(FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
CComPtr<IDevice> pDevicePtr = pDevice;
|
|
|
|
//
|
|
// make a note of real device
|
|
//
|
|
device->AddRef();
|
|
pActualDevice = device;
|
|
//
|
|
// and make a copy of it for temp device
|
|
//
|
|
BSTR Instance = NULL;
|
|
hr = device->get_InstanceId(&Instance);
|
|
if(FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
hr = pDevice->Init(pDevInfoSet,Instance,device->DeviceConsole);
|
|
SysFreeString(Instance);
|
|
if(FAILED(hr)) {
|
|
return hr;
|
|
}
|
|
pDevicePtr.Detach();
|
|
pTempDevice = pDevice;
|
|
SearchType = searchType;
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
HDEVINFO CDrvSearchSet::GetDevInfoSet()
|
|
{
|
|
if(!pTempDevice) {
|
|
return INVALID_HANDLE_VALUE;
|
|
}
|
|
return pTempDevice->GetDevInfoSet();
|
|
}
|
|
|
|
PSP_DEVINFO_DATA CDrvSearchSet::GetDevInfoData()
|
|
{
|
|
if(!pTempDevice) {
|
|
return NULL;
|
|
}
|
|
return &pTempDevice->DevInfoData;
|
|
}
|