ULONG
WINAPI
CNetConnectionPropertyUi::Release()
{
ULONG refCount = InterlockedDecrement(&ref);
if (!refCount)
{
if (pNCfg)
{
pNCfg->Uninitialize();
pNCfg->Release();
}
if (NCfgLock)
{
NCfgLock->Release();
}
if (pProperties)
{
NcFreeNetconProperties(pProperties);
}
delete this;
}
return refCount;
}
// Install the NDIS protocol driver
bool InstallNdisProtocolDriver(wchar_t *inf_path, wchar_t *id, UINT lock_timeout)
{
bool ret = false;
HRESULT hr;
INetCfg *pNetCfg;
HINSTANCE hSetupApiDll = NULL;
BOOL (WINAPI *_SetupCopyOEMInfW)(PCWSTR, PCWSTR, DWORD, DWORD, PWSTR, DWORD, PDWORD, PWSTR *) = NULL;
BOOL (WINAPI *_SetupUninstallOEMInfW)(PCWSTR, DWORD, PVOID) = NULL;
// Validate arguments
if (inf_path == NULL || id == NULL)
{
return false;
}
hSetupApiDll = LoadLibraryA("setupapi.dll");
if (hSetupApiDll == NULL)
{
WHERE;
goto LABEL_CLEANUP;
}
_SetupCopyOEMInfW =
(UINT (__stdcall *)(PCWSTR,PCWSTR,DWORD,DWORD,PWSTR,DWORD,PDWORD,PWSTR *))
GetProcAddress(hSetupApiDll, "SetupCopyOEMInfW");
_SetupUninstallOEMInfW =
(UINT (__stdcall *)(PCWSTR,DWORD,PVOID))
GetProcAddress(hSetupApiDll, "SetupUninstallOEMInfW");
if (_SetupCopyOEMInfW == NULL || _SetupUninstallOEMInfW == NULL)
{
WHERE;
goto LABEL_CLEANUP;
}
hr = CoCreateInstance(CLSID_CNetCfg, NULL, CLSCTX_INPROC_SERVER, IID_INetCfg, (void **)&pNetCfg);
if (SUCCEEDED(hr))
{
INetCfgLock *pLock;
hr = pNetCfg->QueryInterface(IID_INetCfgLock, (PVOID*)&pLock);
if (SUCCEEDED(hr))
{
LPWSTR locked_by;
hr = pLock->AcquireWriteLock(lock_timeout, L"SoftEther VPN", &locked_by);
if (SUCCEEDED(hr))
{
hr = pNetCfg->Initialize(NULL);
if (SUCCEEDED(hr))
{
wchar_t inf_dir[MAX_PATH];
GetDirNameFromFilePathW(inf_dir, sizeof(inf_dir), inf_path);
if (_SetupCopyOEMInfW(inf_path, inf_dir, SPOST_PATH, 0, NULL, 0, NULL, 0))
{
INetCfgClassSetup *pSetup;
hr = pNetCfg->QueryNetCfgClass(&GUID_DEVCLASS_NETTRANS, IID_INetCfgClassSetup, (void **)&pSetup);
if (SUCCEEDED(hr))
{
OBO_TOKEN token;
INetCfgComponent *pComponent;
Zero(&token, sizeof(token));
token.Type = OBO_USER;
hr = pSetup->Install(id, &token, 0, 0, NULL, NULL, &pComponent);
if (SUCCEEDED(hr))
{
pNetCfg->Apply();
ret = true;
}
else
{
WHERE;
Debug("0x%x\n", hr);
}
pSetup->Release();
}
else
{
WHERE;
}
if (ret == false)
{
wchar_t dst_inf_name[MAX_PATH];
DWORD dst_inf_name_size = MAX_PATH;
if (_SetupCopyOEMInfW(inf_path, inf_dir, SPOST_PATH, SP_COPY_REPLACEONLY,
dst_inf_name, dst_inf_name_size, &dst_inf_name_size, NULL) == false &&
GetLastError() == ERROR_FILE_EXISTS)
{
_SetupUninstallOEMInfW(dst_inf_name, 0, NULL);
}
}
}
else
{
WHERE;
}
}
else
{
WHERE;
}
pLock->ReleaseWriteLock();
}
else
{
WHERE;
}
pLock->Release();
}
pNetCfg->Release();
}
else
{
WHERE;
}
LABEL_CLEANUP:
if (hSetupApiDll != NULL)
{
FreeLibrary(hSetupApiDll);
}
return ret;
}
VOID
CNetConnectionPropertyUi::InitializeLANPropertiesUIDlg(HWND hwndDlg)
{
HRESULT hr;
INetCfg *pNCfg;
INetCfgLock *pNCfgLock;
HWND hDlgCtrl = GetDlgItem(hwndDlg, IDC_COMPONENTSLIST);
LVCOLUMNW lc;
RECT rc;
DWORD dwStyle;
LPWSTR pDisplayName;
LVITEMW li;
SendDlgItemMessageW(hwndDlg, IDC_NETCARDNAME, WM_SETTEXT, 0, (LPARAM)pProperties->pszwDeviceName);
if (pProperties->dwCharacter & NCCF_SHOW_ICON)
{
/* check show item on taskbar*/
SendDlgItemMessageW(hwndDlg, IDC_SHOWTASKBAR, BM_SETCHECK, BST_CHECKED, 0);
}
if (pProperties->dwCharacter & NCCF_NOTIFY_DISCONNECTED)
{
/* check notify item */
SendDlgItemMessageW(hwndDlg, IDC_NOTIFYNOCONNECTION, BM_SETCHECK, BST_CHECKED, 0);
}
memset(&lc, 0, sizeof(LV_COLUMN));
lc.mask = LVCF_TEXT | LVCF_SUBITEM | LVCF_FMT;
lc.fmt = LVCFMT_FIXED_WIDTH;
if (GetClientRect(hDlgCtrl, &rc))
{
lc.mask |= LVCF_WIDTH;
lc.cx = rc.right - rc.left;
}
lc.pszText = (LPWSTR)L"";
(void)SendMessageW(hDlgCtrl, LVM_INSERTCOLUMNW, 0, (LPARAM)&lc);
dwStyle = (DWORD) SendMessage(hDlgCtrl, LVM_GETEXTENDEDLISTVIEWSTYLE, 0, 0);
dwStyle = dwStyle | LVS_EX_FULLROWSELECT | LVS_EX_CHECKBOXES;
SendMessage(hDlgCtrl, LVM_SETEXTENDEDLISTVIEWSTYLE, 0, dwStyle);
hr = CoCreateInstance(CLSID_CNetCfg, NULL, CLSCTX_INPROC_SERVER, IID_INetCfg, (LPVOID*)&pNCfg);
if (FAILED(hr))
return;
hr = pNCfg->QueryInterface(IID_INetCfgLock, (LPVOID*)&pNCfgLock);
hr = pNCfgLock->AcquireWriteLock(100, L"", &pDisplayName);
if (hr == S_FALSE)
{
CoTaskMemFree(pDisplayName);
return;
}
NCfgLock = pNCfgLock;
hr = pNCfg->Initialize(NULL);
if (FAILED(hr))
{
pNCfg->Release();
return;
}
EnumComponents(hDlgCtrl, pNCfg, &GUID_DEVCLASS_NETCLIENT, NET_TYPE_CLIENT);
EnumComponents(hDlgCtrl, pNCfg, &GUID_DEVCLASS_NETSERVICE, NET_TYPE_SERVICE);
EnumComponents(hDlgCtrl, pNCfg, &GUID_DEVCLASS_NETTRANS, NET_TYPE_PROTOCOL);
this->pNCfg = pNCfg;
ZeroMemory(&li, sizeof(li));
li.mask = LVIF_STATE;
li.stateMask = (UINT)-1;
li.state = LVIS_FOCUSED|LVIS_SELECTED;
(void)SendMessageW(hDlgCtrl, LVM_SETITEMW, 0, (LPARAM)&li);
}
VOID
CNetConnectionPropertyUi::EnumComponents(HWND hDlgCtrl, INetCfg *pNCfg, const GUID *CompGuid, UINT Type)
{
HRESULT hr;
IEnumNetCfgComponent * pENetCfg;
INetCfgComponent *pNCfgComp, *pAdapterCfgComp;
INetCfgComponentBindings * pCompBind;
ULONG Num;
DWORD dwCharacteristics;
LPOLESTR pDisplayName, pHelpText;
PNET_ITEM pItem;
BOOL bChecked;
hr = pNCfg->EnumComponents(CompGuid, &pENetCfg);
if (FAILED(hr))
{
pNCfg->Release();
return;
}
while(pENetCfg->Next(1, &pNCfgComp, &Num) == S_OK)
{
hr = pNCfgComp->GetCharacteristics(&dwCharacteristics);
if (SUCCEEDED(hr) && (dwCharacteristics & NCF_HIDDEN))
{
pNCfgComp->Release();
continue;
}
pDisplayName = NULL;
pHelpText = NULL;
hr = pNCfgComp->GetDisplayName(&pDisplayName);
hr = pNCfgComp->GetHelpText(&pHelpText);
bChecked = TRUE; //ReactOS hack
hr = pNCfgComp->QueryInterface(IID_INetCfgComponentBindings, (LPVOID*)&pCompBind);
if (SUCCEEDED(hr))
{
if (GetINetCfgComponent(pNCfg, &pAdapterCfgComp))
{
hr = pCompBind->IsBoundTo(pAdapterCfgComp);
if (hr == S_OK)
bChecked = TRUE;
else
bChecked = FALSE;
pAdapterCfgComp->Release();
pCompBind->Release();
}
}
pItem = (NET_ITEM*)CoTaskMemAlloc(sizeof(NET_ITEM));
if (!pItem)
continue;
pItem->dwCharacteristics = dwCharacteristics;
pItem->szHelp = pHelpText;
pItem->Type = (NET_TYPE)Type;
pItem->pNCfgComp = pNCfgComp;
pItem->NumPropDialogOpen = 0;
AddItemToListView(hDlgCtrl, pItem, pDisplayName, bChecked);
CoTaskMemFree(pDisplayName);
}
pENetCfg->Release();
}
// Install the NDIS protocol driver
bool UninstallNdisProtocolDriver(wchar_t *id, UINT lock_timeout)
{
bool ret = false;
HRESULT hr;
INetCfg *pNetCfg;
// Validate arguments
if (id == NULL)
{
return false;
}
hr = CoCreateInstance(CLSID_CNetCfg, NULL, CLSCTX_INPROC_SERVER, IID_INetCfg, (void **)&pNetCfg);
if (SUCCEEDED(hr))
{
INetCfgLock *pLock;
hr = pNetCfg->QueryInterface(IID_INetCfgLock, (PVOID*)&pLock);
if (SUCCEEDED(hr))
{
LPWSTR locked_by;
hr = pLock->AcquireWriteLock(lock_timeout, L"SoftEther VPN", &locked_by);
if (SUCCEEDED(hr))
{
hr = pNetCfg->Initialize(NULL);
if (SUCCEEDED(hr))
{
INetCfgComponent *pncc = NULL;
hr = pNetCfg->FindComponent(id, &pncc);
if (pncc == NULL || hr == S_FALSE)
{
hr = E_FAIL;
}
if (SUCCEEDED(hr))
{
INetCfgClass *pncClass;
hr = pNetCfg->QueryNetCfgClass(&GUID_DEVCLASS_NETTRANS, IID_INetCfgClass, (void **)&pncClass);
if (SUCCEEDED(hr))
{
INetCfgClassSetup *pncClassSetup;
hr = pncClass->QueryInterface(IID_INetCfgClassSetup, (void **)&pncClassSetup);
if (SUCCEEDED(hr))
{
OBO_TOKEN obo;
wchar_t *c = NULL;
Zero(&obo, sizeof(obo));
obo.Type = OBO_USER;
hr = pncClassSetup->DeInstall(pncc, &obo, &c);
if (SUCCEEDED(hr))
{
hr = pNetCfg->Apply();
if (SUCCEEDED(hr))
{
ret = true;
}
else
{
WHERE;
Debug("0x%x\n", hr);
}
}
else
{
WHERE;
Debug("0x%x\n", hr);
}
pncClassSetup->Release();
}
else
{
WHERE;
}
pncClass->Release();
}
else
{
WHERE;
}
pncc->Release();
}
else
{
WHERE;
}
}
else
{
WHERE;
}
pLock->ReleaseWriteLock();
}
else
{
WHERE;
}
pLock->Release();
}
pNetCfg->Release();
}
else
{
WHERE;
}
return ret;
}
bool UninstallNdisProtocolDriver(wchar_t *id, UINT lock_timeout)
{
bool ret = false;
HRESULT hr;
INetCfg *pNetCfg;
// Validate arguments
if (id == NULL)
{
return false;
}
hr = CoCreateInstance(CLSID_CNetCfg, NULL, CLSCTX_INPROC_SERVER, IID_INetCfg, (void **)&pNetCfg);
if (SUCCEEDED(hr))
{
INetCfgLock *pLock;
hr = pNetCfg->QueryInterface(IID_INetCfgLock, (PVOID*)&pLock);
if (SUCCEEDED(hr))
{
LPWSTR locked_by;
hr = pLock->AcquireWriteLock(lock_timeout, L"SoftEther VPN", &locked_by);
if (SUCCEEDED(hr))
{
hr = pNetCfg->Initialize(NULL);
if (SUCCEEDED(hr))
{
INetCfgComponent *pComponent;
hr = pNetCfg->FindComponent(id, &pComponent);
if (SUCCEEDED(hr))
{
INetCfgClassSetup *pSetup;
hr = pNetCfg->QueryNetCfgClass(&GUID_DEVCLASS_NETTRANS, IID_INetCfgClassSetup, (void **)&pSetup);
if (SUCCEEDED(hr))
{
OBO_TOKEN token;
SeZero(&token, sizeof(token));
token.Type = OBO_USER;
hr = pSetup->DeInstall(pComponent, &token, NULL);
if (SUCCEEDED(hr))
{
pNetCfg->Apply();
ret = true;
}
pSetup->Release();
}
}
}
pLock->ReleaseWriteLock();
}
pLock->Release();
}
pNetCfg->Release();
}
return ret;
}
bool InstallNdisProtocolDriver(char *inf_path, wchar_t *id, UINT lock_timeout)
{
bool ret = false;
HRESULT hr;
INetCfg *pNetCfg;
// Validate arguments
if (inf_path == NULL || id == NULL)
{
return false;
}
hr = CoCreateInstance(CLSID_CNetCfg, NULL, CLSCTX_INPROC_SERVER, IID_INetCfg, (void **)&pNetCfg);
if (SUCCEEDED(hr))
{
INetCfgLock *pLock;
hr = pNetCfg->QueryInterface(IID_INetCfgLock, (PVOID*)&pLock);
if (SUCCEEDED(hr))
{
LPWSTR locked_by;
hr = pLock->AcquireWriteLock(lock_timeout, L"SoftEther VPN", &locked_by);
if (SUCCEEDED(hr))
{
hr = pNetCfg->Initialize(NULL);
if (SUCCEEDED(hr))
{
char inf_dir[MAX_PATH];
SeGetDirNameFromFilePath(inf_dir, sizeof(inf_dir), inf_path);
if (SetupCopyOEMInfA(inf_path, inf_dir, SPOST_PATH, 0, NULL, 0, NULL, 0))
{
INetCfgClassSetup *pSetup;
hr = pNetCfg->QueryNetCfgClass(&GUID_DEVCLASS_NETTRANS, IID_INetCfgClassSetup, (void **)&pSetup);
if (SUCCEEDED(hr))
{
OBO_TOKEN token;
INetCfgComponent *pComponent;
SeZero(&token, sizeof(token));
token.Type = OBO_USER;
hr = pSetup->Install(id, &token, 0, 0, NULL, NULL, &pComponent);
if (SUCCEEDED(hr))
{
pNetCfg->Apply();
ret = true;
}
pSetup->Release();
}
if (ret == false)
{
char dst_inf_name[MAX_PATH];
DWORD dst_inf_name_size = MAX_PATH;
if (SetupCopyOEMInfA(inf_path, inf_dir, SPOST_PATH, SP_COPY_REPLACEONLY,
dst_inf_name, dst_inf_name_size, &dst_inf_name_size, NULL) == false &&
GetLastError() == ERROR_FILE_EXISTS)
{
SetupUninstallOEMInfA(dst_inf_name, 0, NULL);
}
}
}
}
pLock->ReleaseWriteLock();
}
pLock->Release();
}
pNetCfg->Release();
}
return ret;
}
