312 lines
8.5 KiB
C++
312 lines
8.5 KiB
C++
|
#include <tchar.h>
|
||
|
|
#include <windows.h>
|
||
|
|
#include <windowsx.h>
|
||
|
|
#include <objbase.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <stdarg.h>
|
||
|
|
#include <wia.h>
|
||
|
|
#include <stdio.h>
|
||
|
|
|
||
|
|
#ifdef _DEBUG
|
||
|
|
#define TRACE(x) Trace x
|
||
|
|
#else
|
||
|
|
#define TRACE(x) Trace x
|
||
|
|
#endif
|
||
|
|
|
||
|
|
#define RELEASE(x)\
|
||
|
|
do {\
|
||
|
|
if((x)) {\
|
||
|
|
(x)->Release(); \
|
||
|
|
(x) = NULL;\
|
||
|
|
}\
|
||
|
|
}while(0)
|
||
|
|
|
||
|
|
#define REQUIRE(x)\
|
||
|
|
do {\
|
||
|
|
if(!(x)) {\
|
||
|
|
TRACE((_T("%hs(%d): %hs failed with LastError() = %d\r\n"),\
|
||
|
|
__FILE__, __LINE__, #x, GetLastError()));\
|
||
|
|
goto Cleanup;\
|
||
|
|
}\
|
||
|
|
} while(0)
|
||
|
|
|
||
|
|
|
||
|
|
#define REQUIRE_S_OK(x)\
|
||
|
|
do {\
|
||
|
|
hr = (x);\
|
||
|
|
\
|
||
|
|
if(hr != S_OK) {\
|
||
|
|
TRACE((_T("%hs(%d): %hs failed with hr = %x\r\n"),\
|
||
|
|
__FILE__, __LINE__, #x, hr));\
|
||
|
|
goto Cleanup;\
|
||
|
|
}\
|
||
|
|
} while(0)
|
||
|
|
|
||
|
|
#define REQUIRE_SUCCESS(x)\
|
||
|
|
do {\
|
||
|
|
DWORD dwResult = (x);\
|
||
|
|
\
|
||
|
|
if(dwResult != ERROR_SUCCESS) {\
|
||
|
|
TRACE((_T("%hs(%d): %hs failed with status = %x\r\n"),\
|
||
|
|
__FILE__, __LINE__, #x, dwResult));\
|
||
|
|
goto Cleanup;\
|
||
|
|
}\
|
||
|
|
} while(0)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
void Trace(LPCTSTR fmt, ...)
|
||
|
|
{
|
||
|
|
TCHAR buffer[2048];
|
||
|
|
va_list marker;
|
||
|
|
|
||
|
|
va_start(marker, fmt);
|
||
|
|
wvsprintf(buffer, fmt, marker);
|
||
|
|
|
||
|
|
printf("\nError: %ws", buffer);
|
||
|
|
}
|
||
|
|
|
||
|
|
struct regVals_t {
|
||
|
|
ULONG propId;
|
||
|
|
LPWSTR propName;
|
||
|
|
} regVals[] = {
|
||
|
|
{WIA_DIP_DEV_ID, WIA_DIP_DEV_ID_STR},
|
||
|
|
{WIA_DIP_VEND_DESC, WIA_DIP_VEND_DESC_STR},
|
||
|
|
{WIA_DIP_DEV_DESC, WIA_DIP_DEV_DESC_STR},
|
||
|
|
{WIA_DIP_DEV_TYPE, WIA_DIP_DEV_TYPE_STR},
|
||
|
|
{WIA_DIP_PORT_NAME, WIA_DIP_PORT_NAME_STR},
|
||
|
|
{WIA_DIP_DEV_NAME, WIA_DIP_DEV_NAME_STR},
|
||
|
|
{WIA_DIP_SERVER_NAME, WIA_DIP_SERVER_NAME_STR},
|
||
|
|
{WIA_DIP_REMOTE_DEV_ID, WIA_DIP_REMOTE_DEV_ID_STR},
|
||
|
|
{WIA_DIP_UI_CLSID, WIA_DIP_UI_CLSID_STR}
|
||
|
|
};
|
||
|
|
#define NREGVALS (sizeof(regVals) / sizeof(regVals[0]))
|
||
|
|
|
||
|
|
HRESULT RegisterRemoteScanners(LPCWSTR servername)
|
||
|
|
{
|
||
|
|
HRESULT hr;
|
||
|
|
IWiaDevMgr *pDevMgr = NULL;
|
||
|
|
IEnumWIA_DEV_INFO *pEnumInfo = NULL;
|
||
|
|
IWiaPropertyStorage *pWiaPropStg = NULL;
|
||
|
|
COSERVERINFO csi;
|
||
|
|
HKEY hDevList = NULL;
|
||
|
|
HKEY hKey = NULL;
|
||
|
|
MULTI_QI mq[1];
|
||
|
|
|
||
|
|
ZeroMemory(&csi, sizeof(csi));
|
||
|
|
csi.pAuthInfo = NULL;
|
||
|
|
csi.pwszName = SysAllocString(servername);
|
||
|
|
|
||
|
|
mq[0].hr = S_OK;
|
||
|
|
mq[0].pIID = &IID_IWiaDevMgr;
|
||
|
|
mq[0].pItf = NULL;
|
||
|
|
|
||
|
|
REQUIRE_S_OK(CoCreateInstanceEx(CLSID_WiaDevMgr, NULL, CLSCTX_REMOTE_SERVER,
|
||
|
|
&csi, 1, mq));
|
||
|
|
|
||
|
|
pDevMgr = (IWiaDevMgr *)mq[0].pItf;
|
||
|
|
|
||
|
|
REQUIRE_SUCCESS(RegCreateKeyEx(
|
||
|
|
HKEY_LOCAL_MACHINE,
|
||
|
|
_T("SYSTEM\\CurrentControlSet\\Control\\StillImage\\DevList"),
|
||
|
|
0,
|
||
|
|
NULL,
|
||
|
|
REG_OPTION_NON_VOLATILE,
|
||
|
|
KEY_ALL_ACCESS,
|
||
|
|
NULL,
|
||
|
|
&hDevList,
|
||
|
|
NULL));
|
||
|
|
|
||
|
|
//
|
||
|
|
// Enumerate all devices scanner
|
||
|
|
//
|
||
|
|
REQUIRE_S_OK(pDevMgr->EnumDeviceInfo(0, &pEnumInfo));
|
||
|
|
|
||
|
|
while(pEnumInfo->Next(1, &pWiaPropStg, NULL) == S_OK)
|
||
|
|
{
|
||
|
|
PROPSPEC ps[NREGVALS];
|
||
|
|
PROPVARIANT pv[NREGVALS];
|
||
|
|
WCHAR keyname[MAX_PATH];
|
||
|
|
WCHAR *p;
|
||
|
|
int i;
|
||
|
|
|
||
|
|
ZeroMemory(pv, sizeof(pv));
|
||
|
|
for(i = 0; i < NREGVALS; i++) {
|
||
|
|
ps[i].ulKind = PRSPEC_PROPID;
|
||
|
|
ps[i].propid = regVals[i].propId;
|
||
|
|
}
|
||
|
|
|
||
|
|
REQUIRE_S_OK(pWiaPropStg->ReadMultiple(
|
||
|
|
sizeof(ps) / sizeof(ps[0]),
|
||
|
|
ps,
|
||
|
|
pv));
|
||
|
|
|
||
|
|
//
|
||
|
|
// Skip any remote devices (anything other than "local"
|
||
|
|
// in server name field
|
||
|
|
//
|
||
|
|
for(i = 0; i < NREGVALS; i++) {
|
||
|
|
if(regVals[i].propId == WIA_DIP_SERVER_NAME) {
|
||
|
|
if(wcscmp(pv[i].bstrVal, L"local")) {
|
||
|
|
goto SkipDevice;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
//
|
||
|
|
// pv[0] consists of device ID and "\NNN", replace "\" with ".", prepend target machine name
|
||
|
|
//
|
||
|
|
p = pv[0].bstrVal;
|
||
|
|
wcscpy(keyname, servername);
|
||
|
|
wcscat(keyname, L".");
|
||
|
|
while(*p && *p != L'\\')
|
||
|
|
{
|
||
|
|
p++;
|
||
|
|
}
|
||
|
|
REQUIRE(*p == L'\\');
|
||
|
|
REQUIRE(*(++p) != L'\0');
|
||
|
|
|
||
|
|
wcscat(keyname, p);
|
||
|
|
|
||
|
|
//
|
||
|
|
// Create device key and populate values
|
||
|
|
//
|
||
|
|
REQUIRE_SUCCESS(RegCreateKeyExW(
|
||
|
|
hDevList,
|
||
|
|
keyname,
|
||
|
|
0,
|
||
|
|
NULL,
|
||
|
|
REG_OPTION_NON_VOLATILE,
|
||
|
|
KEY_ALL_ACCESS,
|
||
|
|
NULL,
|
||
|
|
&hKey,
|
||
|
|
NULL));
|
||
|
|
|
||
|
|
for(i = 0; i < NREGVALS; i++) {
|
||
|
|
|
||
|
|
switch(pv[i].vt) {
|
||
|
|
|
||
|
|
case VT_BSTR:
|
||
|
|
if(regVals[i].propId == WIA_DIP_SERVER_NAME) {
|
||
|
|
//
|
||
|
|
// Server name is reported as "local",
|
||
|
|
// we need to set it to the actual server name
|
||
|
|
//
|
||
|
|
REQUIRE_SUCCESS(RegSetValueExW(
|
||
|
|
hKey,
|
||
|
|
regVals[i].propName,
|
||
|
|
0,
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *)servername,
|
||
|
|
wcslen(servername) * sizeof(WCHAR)));
|
||
|
|
} else {
|
||
|
|
//
|
||
|
|
// All other values we simply copy
|
||
|
|
//
|
||
|
|
REQUIRE_SUCCESS(RegSetValueExW(
|
||
|
|
hKey,
|
||
|
|
regVals[i].propName,
|
||
|
|
0,
|
||
|
|
REG_SZ,
|
||
|
|
(CONST BYTE *)pv[i].bstrVal,
|
||
|
|
wcslen(pv[i].bstrVal) * sizeof(WCHAR)));
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
|
||
|
|
case VT_I4:
|
||
|
|
REQUIRE_SUCCESS(RegSetValueExW(
|
||
|
|
hKey,
|
||
|
|
regVals[i].propName,
|
||
|
|
0,
|
||
|
|
REG_DWORD,
|
||
|
|
(CONST BYTE *)&pv[i].lVal,
|
||
|
|
sizeof(DWORD)));
|
||
|
|
break;
|
||
|
|
|
||
|
|
default:
|
||
|
|
TRACE((_T("Unexpected property type: %d\n"), pv[i].vt));
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
PropVariantClear(pv + i);
|
||
|
|
}
|
||
|
|
|
||
|
|
RegCloseKey(hKey);
|
||
|
|
|
||
|
|
SkipDevice:
|
||
|
|
RELEASE(pWiaPropStg);
|
||
|
|
}
|
||
|
|
|
||
|
|
Cleanup:
|
||
|
|
if(hDevList) RegCloseKey(hDevList);
|
||
|
|
RELEASE(pEnumInfo);
|
||
|
|
RELEASE(pDevMgr);
|
||
|
|
|
||
|
|
if(csi.pwszName) SysFreeString(csi.pwszName);
|
||
|
|
return hr;
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
int __cdecl main(int argc, char **argv)
|
||
|
|
{
|
||
|
|
HRESULT hr;
|
||
|
|
WCHAR servername[MAX_PATH];
|
||
|
|
HKEY hKey;
|
||
|
|
|
||
|
|
|
||
|
|
if(argc < 2) {
|
||
|
|
printf("usage: connect <server name>\n");
|
||
|
|
exit(0);
|
||
|
|
}
|
||
|
|
|
||
|
|
MultiByteToWideChar(CP_ACP, 0, argv[1], -1, servername, sizeof(servername) / sizeof(servername[0]));
|
||
|
|
|
||
|
|
printf("Registering scanners on %s...", argv[1]);
|
||
|
|
|
||
|
|
REQUIRE_S_OK(CoInitializeEx(NULL, COINIT_MULTITHREADED));
|
||
|
|
REQUIRE_S_OK(CoInitializeSecurity(NULL, -1, NULL, NULL,
|
||
|
|
RPC_C_AUTHN_LEVEL_CONNECT,
|
||
|
|
RPC_C_IMP_LEVEL_IMPERSONATE,
|
||
|
|
NULL,
|
||
|
|
0,
|
||
|
|
NULL));
|
||
|
|
REQUIRE_S_OK(RegisterRemoteScanners(servername));
|
||
|
|
|
||
|
|
REQUIRE_SUCCESS(RegOpenKeyEx(HKEY_LOCAL_MACHINE,
|
||
|
|
_T("SYSTEM\\CurrentControlSet\\Control\\StillImage\\DevList"),
|
||
|
|
0,
|
||
|
|
KEY_READ,
|
||
|
|
&hKey));
|
||
|
|
for(int i = 0;;i++) {
|
||
|
|
TCHAR subkeyName[1024];
|
||
|
|
DWORD subkeySize;
|
||
|
|
HKEY hSubKey;
|
||
|
|
TCHAR string[1024];
|
||
|
|
DWORD cbString;
|
||
|
|
DWORD dwType;
|
||
|
|
|
||
|
|
subkeySize = sizeof(subkeyName) / sizeof(subkeyName[0]);
|
||
|
|
if(RegEnumKeyEx(hKey, i, subkeyName, &subkeySize, 0, NULL, NULL, NULL) != ERROR_SUCCESS) {
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
REQUIRE_SUCCESS(RegOpenKeyEx(hKey, subkeyName, 0, KEY_READ, &hSubKey));
|
||
|
|
cbString = sizeof(string);
|
||
|
|
REQUIRE_SUCCESS(RegQueryValueEx(hSubKey, _T("Server"), NULL, &dwType, (BYTE *)string, &cbString));
|
||
|
|
if(lstrcmpi(string, servername))
|
||
|
|
continue;
|
||
|
|
|
||
|
|
cbString = sizeof(string);
|
||
|
|
REQUIRE_SUCCESS(RegQueryValueEx(hSubKey, _T("Name"), NULL, &dwType, (BYTE *)string, &cbString));
|
||
|
|
printf("\nRegistered: %ws", string);
|
||
|
|
RegCloseKey(hSubKey);
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
printf("\ndone.\n");
|
||
|
|
|
||
|
|
Cleanup:
|
||
|
|
CoUninitialize();
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
|