|
|
/******************************************************************************
** (C) COPYRIGHT MICROSOFT CORP., 2000** TITLE: Exports.cpp** VERSION: 1.0** AUTHOR: KeisukeT** DATE: 27 Mar, 2000** DESCRIPTION:* Exported functions.*********************************************************************************/
//
// Precompiled header
//
#include "precomp.h"
#pragma hdrstop
//
// Include
//
#include "sti_ci.h"
#include "exports.h"
#include "device.h"
#include "portsel.h"
#include <devguid.h>
#include <stdio.h>
#include <shlobj.h>
#include <objbase.h>
#include <icm.h>
#include <stiregi.h>
#include <stisvc.h>
//
// Global
//
extern HINSTANCE g_hDllInstance;
//
// Function
//
DLLEXPORTHANDLEWINAPIWiaAddDevice( VOID ){ return ExecCommandLine(STR_ADD_DEVICE, EXECDIR_SYSTEM32);} // WiaAddDevice()
DLLEXPORTBOOLWINAPIWiaRemoveDevice( PSTI_DEVICE_INFORMATION pStiDeviceInformation ){ if(NULL == pStiDeviceInformation){ DebugTrace(TRACE_ERROR,(("WiaRemoveDevice: ERROR!! Invalid argument.\r\n"))); return FALSE; } // if(NULL == pStiDeviceInformation)
return (RemoveDevice(NULL, g_hDllInstance, pStiDeviceInformation->szDeviceInternalName, 0));} // WiaRemoveDevice(
HANDLEWiaInstallerProcess( LPTSTR lpCommandLine ){ BOOL bRet = FALSE; HANDLE hProcess = NULL; PROCESS_INFORMATION ProcessInfo; STARTUPINFO SetupInfo = {sizeof SetupInfo, NULL, NULL, NULL, 0, 0, 0, 0, 0, 0, 0, STARTF_FORCEONFEEDBACK, SW_SHOWNORMAL, 0, NULL, NULL, NULL, NULL};
//
// Create install wizard process.
//
DebugTrace(TRACE_STATUS,(("WiaInstallerProcess: Executing \"%ws\".\r\n"), lpCommandLine)); bRet = CreateProcess(NULL, lpCommandLine, NULL, NULL, FALSE, NORMAL_PRIORITY_CLASS, NULL, NULL, &SetupInfo, &ProcessInfo);
if(bRet){ DebugTrace(TRACE_STATUS,(("WiaInstallerProcess: Installer process successfully created.\r\n"))); CloseHandle(ProcessInfo.hThread); hProcess = ProcessInfo.hProcess; } else { DebugTrace(TRACE_ERROR,(("WiaInstallerProcess: ERROR!! Unable to create a process. Err=0x%x.\r\n"), GetLastError())); hProcess = NULL; }
return hProcess;}
BOOLWINAPICreateWiaShortcut( VOID ){
HRESULT hres; IShellLink *psl; LONG err; HKEY khWindowsCurrentVersion; DWORD dwType; DWORD dwSize;
TCHAR pszSystemPath[MAX_PATH+1]; // path to system32 folder.
TCHAR pszShortcutPath[MAX_PATH+1]; // path to creating shortcut.
TCHAR pszProgramPath[MAX_PATH+1]; // path to ProgramFiles folder.
TCHAR pszAccessoriesPath[MAX_PATH+1]; // path to Accessories folder.
TCHAR pszWizardPath[MAX_PATH+1]; // path to wiaacmgr.exe.
TCHAR pszSticiPath[MAX_PATH+1]; // path to sti_ci.dll.
TCHAR pszWizardName[MAX_PATH+1]; // Menu name ofcreating shortcut.
TCHAR pszWizardLinkName[MAX_PATH+1]; // filename ofcreating shortcut.
TCHAR pszWizardDesc[MAX_PATH+1]; // description of creating shortcut.
TCHAR pszAccessoriesName[MAX_PATH+1]; // localized "Accessories" folder name.
BOOL bRet;
//
// Init locals.
//
bRet = FALSE; psl = NULL; err = 0; khWindowsCurrentVersion = NULL;
memset(pszSystemPath, 0, sizeof(pszSystemPath)); memset(pszShortcutPath, 0, sizeof(pszShortcutPath)); memset(pszProgramPath, 0, sizeof(pszProgramPath)); memset(pszAccessoriesPath, 0, sizeof(pszAccessoriesPath)); memset(pszWizardPath, 0, sizeof(pszWizardPath)); memset(pszSticiPath, 0, sizeof(pszSticiPath)); memset(pszWizardName, 0, sizeof(pszWizardName)); memset(pszWizardLinkName, 0, sizeof(pszWizardLinkName)); memset(pszWizardDesc, 0, sizeof(pszWizardDesc)); memset(pszAccessoriesName, 0, sizeof(pszAccessoriesName));
//
// Get path to the "ProgramFiles" folder.
//
hres = SHGetFolderPath(NULL, CSIDL_COMMON_PROGRAMS | CSIDL_FLAG_CREATE, NULL, 0, pszProgramPath); if(!SUCCEEDED(hres)){ DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! Can't get ProgramFiles folder.\r\n"))); bRet = FALSE; goto CreateWiaShortcut_return;
}
//
// Get localized "Accessoies" folder name from reistry.
//
err = RegOpenKey(HKEY_LOCAL_MACHINE, REGKEY_WINDOWS_CURRENTVERSION, &khWindowsCurrentVersion); if(ERROR_SUCCESS != err){ DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! Can't open Windows\\CurrentVersion.Err=0x%x \r\n"), err)); bRet = FALSE; goto CreateWiaShortcut_return; }
dwSize = sizeof(pszAccessoriesName); err = RegQueryValueEx(khWindowsCurrentVersion, REGSTR_VAL_ACCESSORIES_NAME, NULL, &dwType, (LPBYTE)pszAccessoriesName, &dwSize); if(err){ DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! Can't get %ws value.Err=0x%x\r\n"), REGSTR_VAL_ACCESSORIES_NAME, err));
//
// Unable to get "Accessories" name from registry. Let's take it from resource.
//
if( (NULL == g_hDllInstance) || (0 == LoadString(g_hDllInstance, LocalAccessoriesFolderName, pszAccessoriesName, MAX_PATH)) ) { bRet = FALSE; goto CreateWiaShortcut_return; } // if(0 == LoadString(g_hDllInstance, AccessoriesFolderName, pszAccessoriesName, MAX_PATH))
} // if(err)
//
// Load localizable string from resource.
//
if(NULL != g_hDllInstance){ LoadString(g_hDllInstance, WiaWizardName, pszWizardName, MAX_PATH); } else { DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! No DLL instance\r\n")));
bRet = FALSE; goto CreateWiaShortcut_return; }
//
// Get System path.
//
if( 0== GetSystemDirectory(pszSystemPath, MAX_PATH)){ DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! GetSystemDirectory failed. Err=0x%x\r\n"), GetLastError()));
bRet = FALSE; goto CreateWiaShortcut_return; }
//
// Create shortcut/program name.
//
_sntprintf(pszAccessoriesPath, ARRAYSIZE(pszAccessoriesPath)-1, TEXT("%ws\\%ws"), pszProgramPath, pszAccessoriesName); _sntprintf(pszWizardLinkName, ARRAYSIZE(pszWizardLinkName)-1, TEXT("%ws.lnk"), WIAWIZARDCHORCUTNAME); _sntprintf(pszShortcutPath, ARRAYSIZE(pszShortcutPath)-1, TEXT("%ws\\%ws"), pszAccessoriesPath, pszWizardLinkName); _sntprintf(pszWizardPath, ARRAYSIZE(pszWizardPath)-1, TEXT("%ws\\%ws"), pszSystemPath, WIAACMGR_PATH); _sntprintf(pszSticiPath, ARRAYSIZE(pszSticiPath)-1, TEXT("%ws\\%ws"), pszSystemPath, WIAINSTALLERFILENAME); _sntprintf(pszWizardDesc, ARRAYSIZE(pszWizardDesc)-1, TEXT("@%ws,-%d"), pszSticiPath, WiaWizardDescription);
//
// Create an IShellLink object and get a pointer to the IShellLink
// interface (returned from CoCreateInstance).
//
hres = CoInitialize(NULL); if(!SUCCEEDED(hres)){ DebugTrace(TRACE_ERROR,(("CoInitialize failed. hres=0x%x\r\n"), hres)); bRet = FALSE; goto CreateWiaShortcut_return; }
hres = CoCreateInstance(CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLink, (LPVOID *)&psl); if (SUCCEEDED(hres)){
IPersistFile *ppf;
//
// Query IShellLink for the IPersistFile interface for
// saving the shortcut in persistent storage.
//
hres = psl->QueryInterface(IID_IPersistFile, (void **)&ppf); if (SUCCEEDED(hres)){
// Set the path to the shortcut target.
hres = psl->SetPath(pszWizardPath);
if (SUCCEEDED(hres)){ // Set the argument to the shortcut target.
hres = psl->SetArguments(WIAACMGR_ARG);
if (SUCCEEDED(hres)){ // Set the description of the shortcut.
hres = psl->SetDescription(pszWizardDesc);
if (SUCCEEDED(hres)){ // Save the shortcut via the IPersistFile::Save member function.
hres = ppf->Save(pszShortcutPath, TRUE);
if (SUCCEEDED(hres)){ //
// Shortcut created. Set MUI name.
//
hres = SHSetLocalizedName(pszShortcutPath, pszSticiPath, WiaWizardName); if (SUCCEEDED(hres)){ //
// Operation succeeded.
//
bRet = TRUE; } else { DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! SHSetLocalizedName failed. hRes=0x%x\r\n"), hres)); } } else { DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! Save failed. hRes=0x%x\r\n"), hres)); } } else { DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! SetDescription failed. hRes=0x%x\r\n"), hres)); } } else { DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! SetArguments failed. hRes=0x%x\r\n"), hres)); } } else { DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! SetPath failed. hRes=0x%x\r\n"), hres)); }
// Release the pointer to IPersistFile.
ppf->Release(); } else { DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! QueryInterface(IID_IPersistFile) failed.\r\n"))); }
// Release the pointer to IShellLink.
psl->Release();
CoUninitialize();
} else { // if (SUCCEEDED(hres))
DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: ERROR!! CoCreateInstance(IID_IShellLink) failed.\r\n")));
switch(hres){
case REGDB_E_CLASSNOTREG : DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: REGDB_E_CLASSNOTREG.\r\n"))); break; case CLASS_E_NOAGGREGATION : DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: CLASS_E_NOAGGREGATION.\r\n"))); break; case E_NOINTERFACE : DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: E_NOINTERFACE.\r\n"))); break;
default: DebugTrace(TRACE_ERROR,(("CreateWiaShortcut: default.(hres=0x%x).\r\n hres=0x%x"), hres)); break; }
bRet = FALSE; goto CreateWiaShortcut_return; } // if (SUCCEEDED(hres))
CreateWiaShortcut_return:
if(FALSE == bRet){ CString csCmdLine;
//
// Try it again after next reboot.
//
csCmdLine.MakeSystemPath(STI_CI32_ENTRY_WIZMANU); csCmdLine = TEXT(" ") + csCmdLine; csCmdLine = RUNDLL32 + csCmdLine;
SetRunonceKey(REGSTR_VAL_WIZMENU, csCmdLine); } // if(FALSE == bRet)
//
// Clean up
//
if(NULL != khWindowsCurrentVersion){ RegCloseKey(khWindowsCurrentVersion); }
return bRet;}
BOOLWINAPIDeleteWiaShortcut( VOID ){
HRESULT hres; IShellLink *psl; LONG err; HKEY khWindowsCurrentVersion; DWORD dwType; DWORD dwSize;
TCHAR pszShortcutPath[MAX_PATH]; TCHAR pszAccessoriesName[MAX_PATH]; // localized "Accessories" folder name.
TCHAR pszProgramPath[MAX_PATH]; // path to ProgramFiles folder.
BOOL bRet;
//
// Init locals.
//
bRet = FALSE; psl = NULL; err = 0; khWindowsCurrentVersion = NULL;
memset(pszShortcutPath, 0, sizeof(pszShortcutPath)); memset(pszAccessoriesName, 0, sizeof(pszAccessoriesName)); memset(pszProgramPath, 0, sizeof(pszProgramPath));
//
// Get path to the "ProgramFiles" folder.
//
hres = SHGetFolderPath(NULL, CSIDL_COMMON_PROGRAMS | CSIDL_FLAG_CREATE, NULL, 0, pszProgramPath); if(!SUCCEEDED(hres)){ DebugTrace(TRACE_ERROR,(("DeleteWiaShortcut: ERROR!! Can't get ProgramFiles folder.\r\n"), hres));
bRet = FALSE; goto DeleteWiaShortcut_return;
}
//
// Get localized "Accessoies" folder name from reistry.
//
err = RegOpenKey(HKEY_LOCAL_MACHINE, REGKEY_WINDOWS_CURRENTVERSION, &khWindowsCurrentVersion); if(err){ DebugTrace(TRACE_ERROR,(("DeleteWiaShortcut: ERROR!! Can't open Windows\\CurrentVersion key.Err=0x%x\r\n"), err));
bRet = FALSE; goto DeleteWiaShortcut_return; }
dwSize = sizeof(pszAccessoriesName)-sizeof(TCHAR); err = RegQueryValueEx(khWindowsCurrentVersion, REGSTR_VAL_ACCESSORIES_NAME, NULL, &dwType, (LPBYTE)pszAccessoriesName, &dwSize); if(err){ DebugTrace(TRACE_ERROR,(("DeleteWiaShortcut: ERROR!! Can't get %ws value.Err=0x%x\r\n"), REGSTR_VAL_ACCESSORIES_NAME, err));
//
// Unable to get "Accessories" name from registry. Let's take it from resource.
//
if( (NULL == g_hDllInstance) || (0 == LoadString(g_hDllInstance, LocalAccessoriesFolderName, pszAccessoriesName, MAX_PATH)) ) { bRet = FALSE; goto DeleteWiaShortcut_return; } // if(0 == LoadString(g_hDllInstance, AccessoriesFolderName, pszAccessoriesName, MAX_PATH))
}
//
// Create shortcut/program name.
//
_sntprintf(pszShortcutPath, ARRAYSIZE(pszShortcutPath)-1, TEXT("%ws\\%ws\\%ws.lnk"), pszProgramPath, pszAccessoriesName, WIAWIZARDCHORCUTNAME);
if(!DeleteFile((LPCTSTR)pszShortcutPath)){ DebugTrace(TRACE_ERROR,(("ERROR!! DeleteFile failed. Err=0x%x\r\n"), GetLastError()));
bRet = FALSE; goto DeleteWiaShortcut_return; }
//
// Operation succeeded.
//
bRet = TRUE;
DeleteWiaShortcut_return:
//
// Clean up
//
if(NULL != khWindowsCurrentVersion){ RegCloseKey(khWindowsCurrentVersion); }
return bRet;}
DLLEXPORTVOIDCALLBACKWiaCreateWizardMenu( HWND hwnd, HINSTANCE hinst, LPTSTR lpszCmdLine, int nCmdShow ){ CreateWiaShortcut();}
DLLEXPORTVOIDCALLBACKAddDevice( HWND hWnd, HINSTANCE hInst, LPSTR lpszCmdLine, int nCmdShow ){
HANDLE hDevInfo; HWND hDlg; GUID Guid; SP_DEVINFO_DATA spDevInfoData; SP_INSTALLWIZARD_DATA InstallWizard; SP_DEVINSTALL_PARAMS spDevInstallParams; TCHAR ClassName[LINE_LEN]; DWORD err; DWORD dwRequired; HANDLE hMutex;
CString csTitle; CString csSubTitle; CString csInstruction; CString csListLabel;
DebugTrace(TRACE_PROC_ENTER,(("AddDevice: Enter...\r\n")));
//
// Initialize locals.
//
hDevInfo = INVALID_HANDLE_VALUE; hDlg = hWnd; Guid = GUID_DEVCLASS_IMAGE; err = ERROR_SUCCESS; dwRequired = 0; hMutex = NULL;
memset(&spDevInfoData, 0, sizeof(spDevInfoData)); memset(&InstallWizard, 0, sizeof(InstallWizard)); memset(&spDevInstallParams, 0, sizeof(spDevInstallParams)); memset(ClassName, 0, sizeof(ClassName));
//
// Acquire Mutex.
//
CInstallerMutex CMutex(&hMutex, WIAINSTALLWIZMUTEX, 0); if(!CMutex.Succeeded()){
HWND hwndAnotherWizard; CString csWindowTitle;
hwndAnotherWizard = NULL;
//
// Other instance is running. Just activate that window and quit.
//
csWindowTitle.FromTable (MessageTitle); hwndAnotherWizard = FindWindow(NULL, (LPTSTR)csWindowTitle); if(NULL != hwndAnotherWizard){ if(!SetForegroundWindow(hwndAnotherWizard)){ DebugTrace(TRACE_ERROR, ("AddDevice: ERROR!! SetForegroundWindow() failed. Err=0x%x.\r\n", GetLastError())); } // if(!SetForegroundWindow(hwndAnotherWizard))
} else { // if(NULL != hwndAnotherWizard)
//
// Mutex acquisition was failed but didn't find Window.
// Continue.
//
DebugTrace(TRACE_WARNING, ("AddDevice: WARNING!! Mutex acquisition was failed but didn't find Window.\r\n")); } // else (NULL != hwndAnotherWizard)
goto AddDevice_Err; } // if(!CMutex.Succeeded())
//
// Create Device Information Set from guid.
//
hDevInfo = SetupDiCreateDeviceInfoList(&Guid, hDlg);
if (hDevInfo == INVALID_HANDLE_VALUE) { err=GetLastError(); goto AddDevice_Err; }
// //
// // Get class install parameter.
// //
//
// if(!SetupDiGetClassInstallParams(hDevInfo,
// NULL,
// &spSelectDeviceParams.ClassInstallHeader,
// sizeof(spSelectDeviceParams),
// &dwRequired)){
// err=GetLastError();
// goto AddDevice_Err;
// }
//
// Get class name from Guid.
//
if(!SetupDiClassNameFromGuid(&Guid, ClassName, sizeof(ClassName)/sizeof(TCHAR), NULL )){ err=GetLastError(); goto AddDevice_Err; }
//
// Create a new device information element to install
//
spDevInfoData.cbSize = sizeof(SP_DEVINFO_DATA); if(!SetupDiCreateDeviceInfo(hDevInfo, ClassName, &Guid, NULL, hDlg, DICD_GENERATE_ID, &spDevInfoData )){ err=GetLastError(); goto AddDevice_Err; }
//
// Set new element as selected device
//
if(!SetupDiSetSelectedDevice(hDevInfo, &spDevInfoData )){ err=GetLastError(); goto AddDevice_Err; }
//
// Get device install parameters
//
spDevInstallParams.cbSize = sizeof(SP_DEVINSTALL_PARAMS); if(!SetupDiGetDeviceInstallParams(hDevInfo, &spDevInfoData, &spDevInstallParams )){ err=GetLastError(); goto AddDevice_Err; }
//
// Set device install parameters
//
spDevInstallParams.Flags |= DI_SHOWOEM ; spDevInstallParams.Flags |= DI_USECI_SELECTSTRINGS;
spDevInstallParams.hwndParent = hDlg;
if(!SetupDiSetDeviceInstallParams(hDevInfo, &spDevInfoData, &spDevInstallParams )){ err=GetLastError(); goto AddDevice_Err; }
//
// Set class install parameter.
//
InstallWizard.ClassInstallHeader.InstallFunction = DIF_INSTALLWIZARD; InstallWizard.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER); InstallWizard.hwndWizardDlg = hDlg;
//
// TRUE = Show first page
//
InstallWizard.PrivateFlags = SCIW_PRIV_CALLED_FROMCPL | SCIW_PRIV_SHOW_FIRST;
if(!SetupDiSetClassInstallParams(hDevInfo, &spDevInfoData, &InstallWizard.ClassInstallHeader, sizeof(SP_INSTALLWIZARD_DATA) )) { err=GetLastError(); goto AddDevice_Err; }
//
// Call class installer to retrieve wizard pages
//
if(!SetupDiCallClassInstaller(DIF_INSTALLWIZARD, hDevInfo, &spDevInfoData )){ err=GetLastError(); goto AddDevice_Err; }
//
// Get result from class installer
//
if(!SetupDiGetClassInstallParams(hDevInfo, &spDevInfoData, &InstallWizard.ClassInstallHeader, sizeof(SP_INSTALLWIZARD_DATA), NULL )) { err=GetLastError(); goto AddDevice_Err; }
//
// Prepare UI parameters to be used by DevSelect page,
//
csTitle.FromTable(SelDevTitle); csSubTitle.FromTable(SelDevSubTitle); csInstruction.FromTable(SelDevInstructions); csListLabel.FromTable(SelDevListLabel);
if(!SetSelectDevTitleAndInstructions(hDevInfo, &spDevInfoData, (LPTSTR)csTitle, (LPTSTR)csSubTitle, (LPTSTR)csInstruction, (LPTSTR)csListLabel)) { err=GetLastError(); goto AddDevice_Err; }
//
// Get device selection page
//
InstallWizard.DynamicPageFlags = DYNAWIZ_FLAG_PAGESADDED; InstallWizard.DynamicPages[InstallWizard.NumDynamicPages++] = SetupDiGetWizardPage(hDevInfo, &spDevInfoData, &InstallWizard, SPWPT_SELECTDEVICE, 0);
//
// Create installer property sheet
//
{ PROPSHEETHEADER PropSheetHeader; DWORD Pages; HPROPSHEETPAGE SelectDevicePage;
PropSheetHeader.dwSize = sizeof(PropSheetHeader); PropSheetHeader.dwFlags = PSH_WIZARD | PSH_USECALLBACK | PSH_WIZARD97 | PSH_STRETCHWATERMARK | PSH_WATERMARK | PSH_HEADER; PropSheetHeader.pszbmWatermark = MAKEINTRESOURCE(WizardBitmap); PropSheetHeader.pszbmHeader = MAKEINTRESOURCE(IDB_BANNERBMP); PropSheetHeader.hwndParent = hDlg; PropSheetHeader.hInstance = g_hDllInstance; PropSheetHeader.pszIcon = NULL; //MAKEINTRESOURCE(IDI_NEWDEVICEICON);
PropSheetHeader.pszCaption = MAKEINTRESOURCE(MessageTitle); PropSheetHeader.nStartPage = 0; PropSheetHeader.nPages = InstallWizard.NumDynamicPages; PropSheetHeader.phpage = InstallWizard.DynamicPages; PropSheetHeader.pfnCallback = iHdwWizardDlgCallback;
if(PropertySheet(&PropSheetHeader) < 0){ err=GetLastError(); }
}
AddDevice_Err:
//
// Free allocated memory
//
if(IS_VALID_HANDLE(hDevInfo)){
//
// Set install parameter.
//
InstallWizard.ClassInstallHeader.InstallFunction = DIF_DESTROYWIZARDDATA; InstallWizard.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER); if(!SetupDiSetClassInstallParams(hDevInfo, &spDevInfoData, &InstallWizard.ClassInstallHeader, sizeof(SP_INSTALLWIZARD_DATA)) ) { DebugTrace(TRACE_ERROR,(("AddDevice: ERROR!! SetupDiSetClassInstallParams() failed with active hDevInfo.\r\n"))); }
//
// Let isntaller free context data.
//
SetupDiCallClassInstaller(DIF_DESTROYWIZARDDATA, hDevInfo, &spDevInfoData );
//
// Destroy infoset.
//
SetupDiDestroyDeviceInfoList(hDevInfo); }
DebugTrace(TRACE_PROC_LEAVE,(("AddDevice: Leaving... Ret=VOID.\r\n"))); return;}
BOOLCALLBACKRemoveDevice( HWND hWnd, HINSTANCE hInst, LPTSTR lpszCmdLine, int nCmdShow ){
HANDLE hDevInfo; SP_DEVINFO_DATA spDevInfoData; SP_REMOVEDEVICE_PARAMS spRemoveDeviceParams; BOOL bStatus; BOOL bIsInterfaceOnly; DWORD err; DWORD dwDeviceIndex; TCHAR szTemp[MAX_FRIENDLYNAME+1];
DebugTrace(TRACE_PROC_ENTER,(("RemoveDevice: Enter...\r\n")));
//
// Initialize local.
//
hDevInfo = INVALID_HANDLE_VALUE; bStatus = FALSE; err = ERROR_SUCCESS; bIsInterfaceOnly = FALSE; dwDeviceIndex = INVALID_DEVICE_INDEX;
memset (&spDevInfoData, 0, sizeof(SP_DEVINFO_DATA)); memset((void *)&spRemoveDeviceParams, 0, sizeof(SP_REMOVEDEVICE_PARAMS)); memset(szTemp, 0, sizeof(szTemp));
//
// Check the argument.
//
if(NULL == lpszCmdLine){ DebugTrace(TRACE_ERROR,(("RemoveDevice: ERROR!! Invalid argumet.\r\n"))); goto RemoveDevice_Err; } // if(NULL == lpszCmdLine)
lstrcpyn(szTemp, lpszCmdLine, ARRAYSIZE(szTemp)-1); DebugTrace(TRACE_STATUS,(("RemoveDevice: Removing \"%ws\".\r\n"), szTemp));
//
// Get removing device element.
//
hDevInfo = SelectDevInfoFromDeviceId(szTemp);
if(INVALID_HANDLE_VALUE != hDevInfo){ spDevInfoData.cbSize = sizeof(SP_DEVINFO_DATA); SetupDiGetSelectedDevice(hDevInfo, &spDevInfoData); } else {
//
// See if it's "Interface-only" device.
//
hDevInfo = GetDeviceInterfaceIndex(szTemp, &dwDeviceIndex); if( (INVALID_HANDLE_VALUE == hDevInfo) || (INVALID_DEVICE_INDEX == dwDeviceIndex) ) { DebugTrace(TRACE_ERROR,(("RemoveDevice: ERROR!! Can't find \"%ws\".\r\n"), szTemp)); goto RemoveDevice_Err; }
//
// This is "Interface-only" device.
//
bIsInterfaceOnly = TRUE;
} // if(INVALID_HANDLE_VALUE != hDevInfo)
if(bIsInterfaceOnly){ DebugTrace(TRACE_STATUS,(("RemoveDevice: Uninstalling interface-only device.\r\n")));
//
// Uninstalling "Interface-only" device.
//
CDevice cdThis(hDevInfo, dwDeviceIndex); bStatus = (NO_ERROR == cdThis.Remove(NULL));
} else { // if(bIsInterfaceOnly)
DebugTrace(TRACE_STATUS,(("RemoveDevice: Uninstalling a device w/ devnode.\r\n")));
//
// Uninstalling device w/ devnode.
//
if(!SetupDiSetSelectedDevice(hDevInfo, &spDevInfoData )){ err=GetLastError(); goto RemoveDevice_Err; }
//
// Call class installer to remove selected device.
//
spRemoveDeviceParams.ClassInstallHeader.InstallFunction = DIF_REMOVE; spRemoveDeviceParams.ClassInstallHeader.cbSize = sizeof(SP_CLASSINSTALL_HEADER); spRemoveDeviceParams.Scope = DI_REMOVEDEVICE_GLOBAL;
if(!SetupDiSetClassInstallParams(hDevInfo, &spDevInfoData, &spRemoveDeviceParams.ClassInstallHeader, sizeof(SP_REMOVEDEVICE_PARAMS) )){ err=GetLastError(); goto RemoveDevice_Err; }
if(!SetupDiCallClassInstaller(DIF_REMOVE, hDevInfo, &spDevInfoData )){ err=GetLastError(); goto RemoveDevice_Err; }
//
// Removal succeeded.
//
bStatus = TRUE;
} // if(bIsInterfaceOnly)
RemoveDevice_Err:
if(IS_VALID_HANDLE(hDevInfo)){ SetupDiDestroyDeviceInfoList(hDevInfo); } DebugTrace(TRACE_PROC_LEAVE,(("RemoveDevice... Ret=0x%x Err=0x%x.\r\n"), bStatus, err)); return bStatus;} // RemoveDevice()
DLLEXPORTVOIDCALLBACKInstallWiaService( HWND hwnd, HINSTANCE hinst, LPTSTR lpszCmdLine, int nCmdShow ){ DWORD dwError; DWORD dwStiCount; DWORD dwWiaCount; HANDLE hProcess;
DebugTrace(TRACE_PROC_ENTER,(("InstallWiaService: Enter...\r\n")));
//
// Remove old service entry.
//
GetDeviceCount(&dwWiaCount, &dwStiCount);
//
// Install WIA service. This will install only if the service failed to install during processing of STI.INF, else
// it will simply change the StartType.
//
dwError = StiServiceInstall(FALSE, TRUE, // Disable by default
TEXT("NT AUTHORITY\\LocalService"), TEXT("")); if(NOERROR != dwError){ DebugTrace(TRACE_ERROR,(("InstallWiaService: ERROR!! Unable to install service. Err=0x%x\n"), dwError)); } // if(NOERROR != dwError)
//
// Register WIA DLLs.
//
hProcess = ExecCommandLine(TEXT("regsvr32.exe /s wiaservc.dll"), EXECDIR_SYSTEM32); if(IS_VALID_HANDLE(hProcess)){ CloseHandle(hProcess); }
hProcess = ExecCommandLine(TEXT("regsvr32.exe /s sti.dll"), EXECDIR_SYSTEM32); if(IS_VALID_HANDLE(hProcess)){ CloseHandle(hProcess); }
hProcess = ExecCommandLine(TEXT("regsvr32.exe /s wiascr.dll"), EXECDIR_SYSTEM32); if(IS_VALID_HANDLE(hProcess)){ CloseHandle(hProcess); }
hProcess = ExecCommandLine(TEXT("regsvr32.exe /s wiashext.dll"), EXECDIR_SYSTEM32); if(IS_VALID_HANDLE(hProcess)){ CloseHandle(hProcess); }
hProcess = ExecCommandLine(TEXT("regsvr32.exe /s camocx.dll"), EXECDIR_SYSTEM32); if(IS_VALID_HANDLE(hProcess)){ CloseHandle(hProcess); }
hProcess = ExecCommandLine(TEXT("regsvr32.exe /s wiadefui.dll"), EXECDIR_SYSTEM32); if(IS_VALID_HANDLE(hProcess)){ CloseHandle(hProcess); }
hProcess = ExecCommandLine(TEXT("wiaacmgr.exe /RegServer"), EXECDIR_SYSTEM32); if(IS_VALID_HANDLE(hProcess)){ CloseHandle(hProcess); }
hProcess = ExecCommandLine(TEXT("regsvr32.exe /s wiavusd.dll"), EXECDIR_SYSTEM32); if(IS_VALID_HANDLE(hProcess)){ CloseHandle(hProcess); }
hProcess = ExecCommandLine(TEXT("regsvr32.exe /s wiasf.ax"), EXECDIR_SYSTEM32); if(IS_VALID_HANDLE(hProcess)){ CloseHandle(hProcess); }
hProcess = ExecCommandLine(TEXT("rundll32.exe sti.dll,MigrateRegisteredSTIAppsForWIAEvents %%l"), EXECDIR_SYSTEM32); if(IS_VALID_HANDLE(hProcess)){ CloseHandle(hProcess); }
DebugTrace(TRACE_PROC_LEAVE,(("InstallWiaService: Leaving... Ret=VOID.\r\n")));
} // InstallWiaService()
HANDLESelectDevInfoFromDeviceId( LPTSTR pszDeviceId ){ TCHAR szTemp[MAX_DEVICE_ID]; HANDLE hDevInfo; GUID Guid; DWORD Idx; SP_DEVINFO_DATA spDevInfoData; BOOL bFound; HKEY hKeyDevice; ULONG cbData; LONG lResult;
DebugTrace(TRACE_PROC_ENTER,(("SelectDevInfoFromDeviceId: Enter...\r\n")));
//
// Initialize local.
//
hDevInfo = INVALID_HANDLE_VALUE; Guid = GUID_DEVCLASS_IMAGE;// Guid = KSCATEGORY_CAPTURE;
Idx = 0; cbData = 0; bFound = FALSE; hKeyDevice = NULL; lResult = ERROR_SUCCESS;
memset(szTemp, 0, sizeof(szTemp)); memset(&spDevInfoData, 0, sizeof(spDevInfoData));
//
// Check argument.
//
if(NULL == pszDeviceId){ DebugTrace(TRACE_ERROR,(("SelectDevInfoFromDeviceId: Invalid arbument.\r\n")));
hDevInfo = INVALID_HANDLE_VALUE; goto SelectDevInfoFromDeviceId_return; }
//
// Get device info set of specified class.
//
hDevInfo = SetupDiGetClassDevs (&Guid, NULL, NULL, DIGCF_PROFILE); if (hDevInfo == INVALID_HANDLE_VALUE) { DebugTrace(TRACE_ERROR,(("SelectDevInfoFromDeviceId: SetupDiGetClassDevs failed. Err=0x%x\r\n"), GetLastError()));
hDevInfo = INVALID_HANDLE_VALUE; goto SelectDevInfoFromDeviceId_return; }
spDevInfoData.cbSize = sizeof (SP_DEVINFO_DATA); for (Idx = 0; SetupDiEnumDeviceInfo (hDevInfo, Idx, &spDevInfoData); Idx++) {
DebugTrace(TRACE_STATUS,(("SelectDevInfoFromDeviceId: Checking Device(0x%x)\r\n"), Idx)); hKeyDevice = SetupDiOpenDevRegKey (hDevInfo, &spDevInfoData, DICS_FLAG_GLOBAL, 0, DIREG_DRV, KEY_READ);
if (hKeyDevice != INVALID_HANDLE_VALUE) {
//
// Is DeviceId == pszDeviceId?
//
cbData = sizeof(szTemp)-sizeof(TEXT('\0')); lResult = RegQueryValueEx(hKeyDevice, REGSTR_VAL_DEVICE_ID, NULL, NULL, (LPBYTE)szTemp, &cbData); if(ERROR_SUCCESS == lResult){
if(_tcsicmp((LPCTSTR)pszDeviceId, (LPCTSTR)szTemp) != 0) {
//
// Doesn't match, skip this one.
//
RegCloseKey(hKeyDevice); continue; } } else { DebugTrace(TRACE_ERROR,(("SelectDevInfoFromDeviceId: can't get DeviceId. Err=0x%x\r\n"), GetLastError())); RegCloseKey(hKeyDevice); continue; }
//
// Found the target!
//
bFound = TRUE; RegCloseKey(hKeyDevice); break; } else { DebugTrace(TRACE_ERROR,(("SelectDevInfoFromDeviceId: Invalid handle.\r\n"), GetLastError())); } // if (hKeyDevice != INVALID_HANDLE_VALUE)
} //for (Idx = 0; SetupDiEnumDeviceInfo (hDevInfo, Idx, &spDevInfoData); Idx++)
SelectDevInfoFromDeviceId_return:
if(!bFound){
//
// FriendleName is not found.
//
if (IS_VALID_HANDLE(hDevInfo)) { SetupDiDestroyDeviceInfoList(hDevInfo); hDevInfo = INVALID_HANDLE_VALUE; } } else {
//
// Device found. Select found device.
//
SetupDiSetSelectedDevice(hDevInfo, &spDevInfoData); }
DebugTrace(TRACE_PROC_LEAVE,(("SelectDevInfoFromDeviceId: Leaving... Ret=0x%x\r\n"), hDevInfo));
return hDevInfo;} // SelectDevInfoFromDeviceId()
HANDLEGetDeviceInterfaceIndex( LPTSTR pszDeviceId, DWORD *pdwIndex ){ TCHAR szTemp[MAX_DEVICE_ID]; HANDLE hDevInfo; GUID Guid; DWORD Idx; SP_DEVICE_INTERFACE_DATA spDevInterfaceData; BOOL bFound; HKEY hKeyInterface; ULONG cbData; LONG lResult;
DebugTrace(TRACE_PROC_ENTER,(("GetDeviceInterfaceIndex: Enter... DeviceId=%ws\r\n"), pszDeviceId));
//
// Initialize local.
//
hDevInfo = INVALID_HANDLE_VALUE; Guid = GUID_DEVCLASS_IMAGE; Idx = 0; cbData = 0; bFound = FALSE; hKeyInterface = NULL; lResult = ERROR_SUCCESS;
memset(szTemp, 0, sizeof(szTemp)); memset(&spDevInterfaceData, 0, sizeof(spDevInterfaceData));
//
// Check argument.
//
if(NULL == pszDeviceId){ DebugTrace(TRACE_ERROR,(("GetDeviceInterfaceIndex: Invalid arbument.\r\n")));
hDevInfo = INVALID_HANDLE_VALUE; goto GetDeviceInterfaceIndex_return; }
//
// Get device info set of specified class interface.
//
hDevInfo = SetupDiGetClassDevs (&Guid, NULL, NULL, DIGCF_DEVICEINTERFACE | DIGCF_PROFILE); if (hDevInfo == INVALID_HANDLE_VALUE) { DebugTrace(TRACE_ERROR,(("GetDeviceInterfaceIndex: SetupDiGetClassDevs failed. Err=0x%x\r\n"), GetLastError()));
hDevInfo = INVALID_HANDLE_VALUE; goto GetDeviceInterfaceIndex_return; }
spDevInterfaceData.cbSize = sizeof (SP_DEVICE_INTERFACE_DATA); for (Idx = 0; SetupDiEnumDeviceInterfaces (hDevInfo, NULL, &Guid, Idx, &spDevInterfaceData); Idx++) {
DebugTrace(TRACE_STATUS,(("GetDeviceInterfaceIndex: Checking Interface(0x%x)\r\n"), Idx)); hKeyInterface = SetupDiOpenDeviceInterfaceRegKey(hDevInfo, &spDevInterfaceData, 0, KEY_ALL_ACCESS); if (INVALID_HANDLE_VALUE != hKeyInterface) {
//
// Is FriendlyName == pszLocalName?
//
cbData = sizeof(szTemp)-sizeof(TEXT('\0')); lResult = RegQueryValueEx(hKeyInterface, REGSTR_VAL_DEVICE_ID, NULL, NULL, (LPBYTE)szTemp, &cbData); if(ERROR_SUCCESS == lResult){
if(_tcsicmp((LPCTSTR)pszDeviceId, (LPCTSTR)szTemp) == 0) {
//
// Found the target!
//
bFound = TRUE; RegCloseKey(hKeyInterface); break; } } else { // if(ERROR_SUCCESS == lResult)
DebugTrace(TRACE_STATUS,(("GetDeviceInterfaceIndex: can't get DeviceID. Err=0x%x\r\n"), GetLastError())); } // if(ERROR_SUCCESS == lResult)
RegCloseKey(hKeyInterface); hKeyInterface = NULL; } else { // if (hKeyDevice != INVALID_HANDLE_VALUE)
DWORD Err; Err = GetLastError(); DebugTrace(TRACE_ERROR,(("GetDeviceInterfaceIndex: Invalid handle. Err=0x%x.\r\n"), Err)); } // if (hKeyDevice != INVALID_HANDLE_VALUE)
} //for (Idx = 0; SetupDiEnumDeviceInterface (hDevInfo, NULL, &Guid, Idx, &spDevInterfaceData); Idx++)
GetDeviceInterfaceIndex_return:
if(FALSE == bFound){ if (IS_VALID_HANDLE(hDevInfo)) { SetupDiDestroyDeviceInfoList(hDevInfo); hDevInfo = INVALID_HANDLE_VALUE; }
*pdwIndex = INVALID_DEVICE_INDEX;
} else {
//
// Interface found.
//
*pdwIndex = Idx; }
DebugTrace(TRACE_PROC_LEAVE,(("GetDeviceInterfaceIndex: Leaving... Ret=0x%x\r\n"), hDevInfo));
return hDevInfo;} // GetDeviceInterfaceIndex()
INT CALLBACKiHdwWizardDlgCallback( IN HWND hwndDlg, IN UINT uMsg, IN LPARAM lParam )/*++
Routine Description:
Call back used to remove the "?" from the wizard page.
Arguments:
hwndDlg - Handle to the property sheet dialog box.
uMsg - Identifies the message being received. This parameter is one of the following values:
PSCB_INITIALIZED - Indicates that the property sheet is being initialized. The lParam value is zero for this message.
PSCB_PRECREATE Indicates that the property sheet is about to be created. The hwndDlg parameter is NULL and the lParam parameter is a pointer to a dialog template in memory. This template is in the form of a DLGTEMPLATE structure followed by one or more DLGITEMTEMPLATE structures.
lParam - Specifies additional information about the message. The meaning of this value depends on the uMsg parameter.
Return Value:
The function returns zero.
--*/{
switch( uMsg ) {
case PSCB_INITIALIZED: break;
case PSCB_PRECREATE: if( lParam ){
DLGTEMPLATE *pDlgTemplate = (DLGTEMPLATE *)lParam; pDlgTemplate->style &= ~DS_CONTEXTHELP; } break; }
return FALSE;}
BOOLSetSelectDevTitleAndInstructions( HDEVINFO hDevInfo, PSP_DEVINFO_DATA pspDevInfoData, LPCTSTR pszTitle, LPCTSTR pszSubTitle, LPCTSTR pszInstn, LPCTSTR pszListLabel )/*++
Routine Description:
Arguments:
Return Value:
Side effects:
--*/{ SP_SELECTDEVICE_PARAMS SelectDevParams; BOOL fRet;
memset((void *)&SelectDevParams, 0, sizeof(SelectDevParams));
if ( pszTitle && (lstrlen(pszTitle) + 1 > MAX_TITLE_LEN ) ) {
SetLastError(ERROR_INVALID_PARAMETER); return FALSE; }
if ( pszSubTitle && (lstrlen(pszSubTitle) + 1 > MAX_SUBTITLE_LEN )) {
SetLastError(ERROR_INVALID_PARAMETER); return FALSE; }
if ( pszInstn && (lstrlen(pszInstn) + 1 > MAX_INSTRUCTION_LEN )) {
SetLastError(ERROR_INVALID_PARAMETER); return FALSE; }
if ( pszListLabel && (lstrlen(pszListLabel) + 1 > MAX_LABEL_LEN )) {
SetLastError(ERROR_INVALID_PARAMETER); return FALSE; }
SelectDevParams.ClassInstallHeader.cbSize = sizeof(SelectDevParams.ClassInstallHeader);
if ( !SetupDiGetClassInstallParams(hDevInfo, pspDevInfoData, &SelectDevParams.ClassInstallHeader, sizeof(SelectDevParams), NULL) ) { DWORD dwErr = GetLastError();
if (ERROR_NO_CLASSINSTALL_PARAMS != dwErr ) { return FALSE; } }
if ( pszTitle ){ lstrcpyn(SelectDevParams.Title, pszTitle, ARRAYSIZE(SelectDevParams.Title)); SelectDevParams.Title[ARRAYSIZE(SelectDevParams.Title)-1] = TEXT('\0'); }
if ( pszSubTitle ){ lstrcpyn(SelectDevParams.SubTitle, pszSubTitle, ARRAYSIZE(SelectDevParams.SubTitle)); SelectDevParams.SubTitle[ARRAYSIZE(SelectDevParams.SubTitle)-1] = TEXT('\0'); }
if ( pszInstn ){ lstrcpyn(SelectDevParams.Instructions, pszInstn, ARRAYSIZE(SelectDevParams.Instructions)); SelectDevParams.Instructions[ARRAYSIZE(SelectDevParams.Instructions)-1] = TEXT('\0'); } if ( pszListLabel ){ lstrcpyn(SelectDevParams.ListLabel, pszListLabel, ARRAYSIZE(SelectDevParams.ListLabel)); SelectDevParams.ListLabel[ARRAYSIZE(SelectDevParams.ListLabel)-1] = TEXT('\0'); }
SelectDevParams.ClassInstallHeader.InstallFunction = DIF_SELECTDEVICE; fRet = SetupDiSetClassInstallParams(hDevInfo, pspDevInfoData, &SelectDevParams.ClassInstallHeader, sizeof(SelectDevParams));
return fRet;
}
BOOLWINAPIWiaDeviceEnum( VOID ){ BOOL rVal = FALSE; SC_HANDLE hSvcMgr = NULL; SC_HANDLE hService = NULL; SERVICE_STATUS ServiceStatus; UINT uiRetry = 10;
DebugTrace(TRACE_PROC_ENTER,(("WiaDeviceEnum: Enter... \r\n")));
//
// Open Service Control Manager.
//
hSvcMgr = OpenSCManager( NULL, NULL, SC_MANAGER_ALL_ACCESS ); if (!hSvcMgr) { DebugTrace(TRACE_ERROR,(("WiaDeviceEnum: ERROR!! OpenSCManager failed. Err=0x%x\n"), GetLastError())); goto exit; }
//
// Open WIA service.
//
hService = OpenService( hSvcMgr, STI_SERVICE_NAME, SERVICE_ALL_ACCESS );
if (!hService) { DebugTrace(TRACE_ERROR,(("WiaDeviceEnum: ERROR!! OpenService failed. Err=0x%x\n"), GetLastError())); goto exit; }
//
// Inform WIA service to refresh its device list.
//
rVal = ControlService(hService, STI_SERVICE_CONTROL_REFRESH, &ServiceStatus); if (!rVal) { DebugTrace(TRACE_WARNING,(("WiaDeviceEnum: WARNING!! ControlService failed. Err=0x%x\n"), GetLastError())); goto exit; }
exit: if(NULL != hService){ CloseServiceHandle( hService ); } if(NULL != hSvcMgr){ CloseServiceHandle( hSvcMgr ); }
DebugTrace(TRACE_PROC_LEAVE,(("WiaDeviceEnum: Leaving... Ret=0x%x\n"), rVal)); return rVal;
} // WiaDeviceEnum()
DLLEXPORTPWIA_PORTLISTWINAPIWiaCreatePortList( LPWSTR szDeviceId ){
PWIA_PORTLIST pReturn; GUID PortGuid; HDEVINFO hPortDevInfo = NULL; DWORD Idx; DWORD dwRequired; DWORD dwNumberOfPorts; DWORD dwSize; CStringArray csaPortName; TCHAR szPortName[MAX_DESCRIPTION]; TCHAR szPortFriendlyName[MAX_DESCRIPTION];
BOOL bIsSerial; BOOL bIsParallel; BOOL bIsAutoCapable; BOOL bIsPortSelectable; //
// Initialize local.
//
Idx = 0; hPortDevInfo = NULL; pReturn = NULL; dwSize = 0; dwRequired = 0; dwNumberOfPorts = 0;
bIsSerial = TRUE; bIsParallel = TRUE; bIsAutoCapable = FALSE; bIsPortSelectable = TRUE;
memset(szPortName, 0, sizeof(szPortName)); memset(szPortFriendlyName, 0, sizeof(szPortFriendlyName));
//
// Convert from WCHAR to TCHAR
//
#ifndef UNICODE
//
// #pragma message("Not optimal conversion - reimplement if running on nonUNICODE system")
//
// TCHAR szDeviceIdConverted[STI_MAX_INTERNAL_NAME_LENGTH+1];
//
// szDeviceIdConverted[0] = TEXT('\0');
// MultiByteToWideChar(CP_ACP,
// 0,
// szDeviceIdConverted,
// -1,
// szDeviceId,
// sizeof(szDeviceId));
//
#else
// On UNICODE system use the same buffer
#define szDeviceIdConverted szDeviceId
#endif
if(!CheckPortForDevice(szDeviceIdConverted, &bIsSerial, &bIsParallel, &bIsAutoCapable, &bIsPortSelectable)){ DebugTrace(TRACE_ERROR,(("WiaGetPortList: ERROR!! Unable to get port info for device.\r\n")));
pReturn = NULL; goto WiaGetPortList_return; }
if(bIsAutoCapable){ dwNumberOfPorts++; csaPortName.Add(AUTO); }
//
// Enumerate all Port class devices if "PortSelect=NO" is not specified.
//
if(bIsPortSelectable){
//
// Get GUID of port device.
//
if(!SetupDiClassGuidsFromName (PORTS, &PortGuid, sizeof(GUID), &dwRequired)){ DebugTrace(TRACE_ERROR,(("WiaGetPortList: ERROR!! SetupDiClassGuidsFromName Failed. Err=0x%lX\r\n"), GetLastError()));
pReturn = NULL; goto WiaGetPortList_return; } // if(!SetupDiClassGuidsFromName (PORTS, &Guid, sizeof(GUID), &dwRequired))
//
// Get device info set of port devices.
//
hPortDevInfo = SetupDiGetClassDevs (&PortGuid, NULL, NULL, DIGCF_PRESENT | DIGCF_PROFILE); if (hPortDevInfo == INVALID_HANDLE_VALUE) { DebugTrace(TRACE_ERROR,(("WiaGetPortList: ERROR!! SetupDiGetClassDevs Failed. Err=0x%lX\r\n"), GetLastError()));
pReturn = NULL; goto WiaGetPortList_return; }
//
// Process all of device element listed in device info set.
//
for(Idx = 0; GetPortNamesFromIndex(hPortDevInfo, Idx, szPortName, szPortFriendlyName); Idx++){
//
// Add valid Port CreateFile/Friendly Name to array.
//
if(0 == lstrlen(szPortName)){ DebugTrace(TRACE_ERROR,(("WiaGetPortList: ERROR!! Invalid Port/Friendly Name.\r\n")));
szPortName[0] = TEXT('\0'); continue; }
DebugTrace(TRACE_STATUS,(("WiaGetPortList: Found Port %d: %ws.\r\n"), Idx, szPortName));
//
// Check it's port type.
//
if(_tcsstr((const TCHAR *)szPortName, TEXT("LPT"))){ if(!bIsParallel){ szPortName[0] = TEXT('\0'); continue; } }
if(_tcsstr((const TCHAR *)szPortName, TEXT("COM"))){ if(!bIsSerial){ szPortName[0] = TEXT('\0'); continue; } }
dwNumberOfPorts++; csaPortName.Add(szPortName);
szPortName[0] = TEXT('\0');
} // for(Idx = 0; GetPortNamesFromIndex(hPortDevInfo, Idx, szPortName, szPortFriendlyName); Idx++)
} // if(bIsPortSelectable)
if(0 != dwNumberOfPorts){
//
// Allocate memory for returning structure.
//
dwSize = sizeof(DWORD) + sizeof(LPTSTR)*dwNumberOfPorts; pReturn = (PWIA_PORTLIST)new BYTE[dwSize]; if(NULL == pReturn){ goto WiaGetPortList_return; } memset(pReturn, 0, dwSize);
//
// Fill in the info.
//
pReturn->dwNumberOfPorts = dwNumberOfPorts; for(Idx = 0; Idx < dwNumberOfPorts; Idx++){ pReturn->szPortName[Idx] = (LPTSTR)new BYTE[lstrlen(csaPortName[Idx])*sizeof(TCHAR)+sizeof(TEXT('\0'))]; if(NULL != pReturn->szPortName[Idx]){ lstrcpy(pReturn->szPortName[Idx], csaPortName[Idx]); } else { WiaDestroyPortList(pReturn); pReturn = NULL; break; } // if(NULL != pReturn->szPortName[Idx])
} // for(Idx = 0; Idx < dwNumberOfPorts; Idx++)
} // if(0 != dwNumberOfPorts)
WiaGetPortList_return:
//
// Cleanup
//
if ( IS_VALID_HANDLE(hPortDevInfo) ) { SetupDiDestroyDeviceInfoList(hPortDevInfo); }
return pReturn;
} // WiaCreatePortList()
DLLEXPORTVOIDWINAPIWiaDestroyPortList( PWIA_PORTLIST pWiaPortList ){ DWORD Idx;
if(NULL == pWiaPortList){ return; } // if(NULL == pWiaPortList)
for(Idx = 0; Idx < pWiaPortList->dwNumberOfPorts; Idx++){ if(NULL != pWiaPortList->szPortName[Idx]){ delete pWiaPortList->szPortName[Idx]; } // if(NULL != pWiaPortList->szPortName[Idx])
} // for(Idx = 0; Idx < pWiaPortList; Idx++)
delete pWiaPortList;
return;
} // WiaDestroyPortList()
BOOLCheckPortForDevice( LPTSTR szDeviceId, BOOL *pbIsSerial, BOOL *pbIsParallel, BOOL *pbIsAutoCapable, BOOL *pbIsPortSelectable ){
GUID WiaGuid; HDEVINFO hWiaDevInfo; DWORD dwCapability; SP_DEVINFO_DATA spDevInfoData; CString csConnection; CString csPortSelect; DWORD dwDeviceIndex; HKEY hkDevice; BOOL bCapabilityAcquired; BOOL bRet;
BOOL bIsSerial; BOOL bIsParallel; BOOL bIsAutoCapable; BOOL bIsPortSelectable; DWORD dwIsPnp;
//
// Initialize locals.
//
dwCapability = 0; dwIsPnp = 0; hWiaDevInfo = INVALID_HANDLE_VALUE; dwDeviceIndex = INVALID_DEVICE_INDEX; WiaGuid = GUID_DEVCLASS_IMAGE;
bRet = FALSE; bCapabilityAcquired = FALSE;
bIsSerial = TRUE; bIsParallel = TRUE; bIsAutoCapable = FALSE; bIsPortSelectable = TRUE;
memset(&spDevInfoData, 0, sizeof(spDevInfoData));
//
// Get specified device property.
//
hWiaDevInfo = SelectDevInfoFromDeviceId(szDeviceId);
if(INVALID_HANDLE_VALUE != hWiaDevInfo){ spDevInfoData.cbSize = sizeof(SP_DEVINFO_DATA); SetupDiGetSelectedDevice(hWiaDevInfo, &spDevInfoData);
hkDevice = SetupDiOpenDevRegKey(hWiaDevInfo, &spDevInfoData, DICS_FLAG_GLOBAL, 0, DIREG_DRV, KEY_READ);
if(INVALID_HANDLE_VALUE != hkDevice){ csConnection.Load(hkDevice, CONNECTION); csPortSelect.Load(hkDevice, PORTSELECT); GetDwordFromRegistry(hkDevice, ISPNP, &dwIsPnp);
if(GetDwordFromRegistry(hkDevice, CAPABILITIES, &dwCapability)){
bCapabilityAcquired = TRUE;
} // if(GetDwordFromRegistry(hkDevice, CAPABILITIES, &dwCapability))
RegCloseKey(hkDevice); hkDevice = (HKEY)INVALID_HANDLE_VALUE;
} // if(INVALID_HANDLE_VALUE != hkDevice)
SetupDiDestroyDeviceInfoList(hWiaDevInfo); hWiaDevInfo = INVALID_HANDLE_VALUE;
} else { // if(INVALID_HANDLE_VALUE != hDevInfo)
SP_DEVICE_INTERFACE_DATA spDeviceInterfaceData;
//
// See if it's "Interface-only" device.
//
hWiaDevInfo = GetDeviceInterfaceIndex(szDeviceId, &dwDeviceIndex); if( (INVALID_HANDLE_VALUE == hWiaDevInfo) || (INVALID_DEVICE_INDEX == dwDeviceIndex) ) { DebugTrace(TRACE_ERROR,(("CheckPortForDevice: ERROR!! Can't find \"%ws\".\r\n"), szDeviceId)); bRet = FALSE; goto CheckPortForDevice_return; }
spDeviceInterfaceData.cbSize = sizeof(SP_DEVICE_INTERFACE_DATA); if(!SetupDiEnumDeviceInterfaces(hWiaDevInfo, NULL, &WiaGuid, dwDeviceIndex, &spDeviceInterfaceData)){ DebugTrace(TRACE_ERROR,(("CheckPortForDevice: ERROR!! SetupDiEnumDeviceInterfaces() failed. Err=0x%x.\r\n"), GetLastError())); bRet = FALSE; goto CheckPortForDevice_return; }
hkDevice = SetupDiOpenDeviceInterfaceRegKey(hWiaDevInfo, &spDeviceInterfaceData, 0, KEY_READ); if(INVALID_HANDLE_VALUE == hkDevice){ DebugTrace(TRACE_ERROR,(("CheckPortForDevice: ERROR!! SetupDiOpenDeviceInterfaceRegKey() failed. Err=0x%x.\r\n"), GetLastError())); bRet = FALSE; goto CheckPortForDevice_return; }
csConnection.Load(hkDevice, CONNECTION); csPortSelect.Load(hkDevice, PORTSELECT); GetDwordFromRegistry(hkDevice, ISPNP, &dwIsPnp); if(GetDwordFromRegistry(hkDevice, CAPABILITIES, &dwCapability)){
bCapabilityAcquired = TRUE;
} // if(GetDwordFromRegistry(hkDevice, CAPABILITIES, &dwCapability))
RegCloseKey(hkDevice); hkDevice = (HKEY)INVALID_HANDLE_VALUE;
} // else (INVALID_HANDLE_VALUE != hDevInfo)
//
// Check what port should be shown.
//
if(0 != dwIsPnp){ //
// This is PnP device. No port should be available.
//
bRet = FALSE; goto CheckPortForDevice_return; }
if(bCapabilityAcquired){
if(csConnection.IsEmpty()){ bIsSerial = TRUE; bIsParallel = TRUE; } else { if(0 == _tcsicmp((LPTSTR)csConnection, SERIAL)){ bIsSerial = TRUE; bIsParallel = FALSE; } if(0 == _tcsicmp((LPTSTR)csConnection, PARALLEL)){ bIsSerial = FALSE; bIsParallel = TRUE; } }
if(dwCapability & STI_GENCAP_AUTO_PORTSELECT){ bIsAutoCapable = TRUE; } else { bIsAutoCapable = FALSE; }
if(0 == MyStrCmpi(csPortSelect, NO)){ bIsPortSelectable = FALSE; } else {// if(0 == lstrcmpi(csPortSelect, NO))
bIsPortSelectable = TRUE; } } else { DebugTrace(TRACE_ERROR,(("CheckPortForDevice: ERROR!! Unable to acquire info from registry.\r\n"))); bRet = FALSE; goto CheckPortForDevice_return; }
//
// Operation succeeded.
//
bRet = TRUE;
CheckPortForDevice_return:
if(IS_VALID_HANDLE(hWiaDevInfo)){ SetupDiDestroyDeviceInfoList(hWiaDevInfo); }
if(bRet){ *pbIsSerial = bIsSerial; *pbIsParallel = bIsParallel; *pbIsAutoCapable = bIsAutoCapable; *pbIsPortSelectable = bIsPortSelectable; }
return bRet;
} // CheckPortForDevice()
DLLEXPORTBOOLWINAPIMigrateDevice( PDEVICE_INFO pMigratingDevice ){ BOOL bSucceeded;
//
// Initialize local.
//
bSucceeded = TRUE;
//
// Validate migrating device.
//
if(!IsValidMigrationDevice(pMigratingDevice)){ DebugTrace(TRACE_ERROR,(("MigrateDevice: ERROR!! Invalid migration device.\r\n"))); bSucceeded = FALSE; return FALSE; } // if(!IsValidMigrationDevice(pMigratingDevice))
//
// Create Device class object.
//
CDevice cdThis(pMigratingDevice);
//
// Set default devnode selector.
//
cdThis.SetDevnodeSelectCallback((DEVNODESELCALLBACK)GetDevinfoFromPortName);
//
// Generate FriendlyName.
//
cdThis.NameDefaultUniqueName();
//
// Install(migrate) the device.
//
bSucceeded = cdThis.PreInstall(); if(bSucceeded){ bSucceeded = cdThis.Install(); }
//
// Do final touch. Clean up if failed, or finish installation.
//
cdThis.PostInstall(bSucceeded);
//MigrateDevice_return:
return bSucceeded;} // MigrateDevice()
BOOLIsValidMigrationDevice( PDEVICE_INFO pMigratingDevice ){ BOOL bRet; DWORD dwCounter; PARAM_LIST CurrentParam; PVOID pNext; HINF hInf;
//
// Initialize local.
//
bRet = FALSE; dwCounter = 0; pNext = NULL; hInf = INVALID_HANDLE_VALUE; memset(&CurrentParam, 0, sizeof(CurrentParam)); //
// See if given pointer is NULL.
//
if(NULL == pMigratingDevice){ DebugTrace(TRACE_ERROR,(("IsValidMigrationDevice: ERROR!! Passed pointer is NULL.\r\n"))); bRet = FALSE; goto IsValidMigrationDevice_return; } // if(NULL == pMigratingDevice)
//
// See if all members are valid.
//
_try { //
// It must have valid FridendlyName.
//
if( (NULL == pMigratingDevice->pszFriendlyName) || (!IsValidFriendlyName(pMigratingDevice->pszFriendlyName)) ) { DebugTrace(TRACE_ERROR,(("IsValidMigrationDevice: ERROR!! Invalid FriendlyName.\r\n"))); bRet = FALSE; goto IsValidMigrationDevice_return; }
//
// It must have valid INF path and section.
//
if( (NULL == pMigratingDevice->pszInfPath) || (NULL == pMigratingDevice->pszInfSection) ) { DebugTrace(TRACE_ERROR,(("IsValidMigrationDevice: ERROR!! Invalid INF info.\r\n"))); bRet = FALSE; goto IsValidMigrationDevice_return; }
//
// See if INF exists or not.
//
hInf = SetupOpenInfFileA(pMigratingDevice->pszInfPath, NULL, INF_STYLE_WIN4, NULL); if(INVALID_HANDLE_VALUE == hInf){ DebugTrace(TRACE_ERROR,(("IsValidMigrationDevice: ERROR!! INF doesn't exist.\r\n"))); bRet = FALSE; goto IsValidMigrationDevice_return; } else { SetupCloseInfFile(hInf); hInf = INVALID_HANDLE_VALUE; }
//
// It must have valid DeviceData parameters.
//
dwCounter = pMigratingDevice->dwNumberOfDeviceDataKey; pNext = pMigratingDevice->pDeviceDataParam; for(;dwCounter != 0; dwCounter--){ CurrentParam = *((PPARAM_LIST)pNext); pNext = CurrentParam.pNext; } // for(;dwCounter != 0; dwCounter--)
} // _try
_except (EXCEPTION_EXECUTE_HANDLER) { DebugTrace(TRACE_ERROR,(("IsValidMigrationDevice: ERROR!! Access to given data caused AV.\r\n"))); bRet = FALSE; goto IsValidMigrationDevice_return; } // _except (EXCEPTION_EXECUTE_HANDLER)
//
// Data validated.
//
bRet = TRUE; IsValidMigrationDevice_return: return bRet;
} // IsValidMigrationDevice()
|
