mirror of
https://github.com/zodiacon/WFPExplorer
synced 2026-06-08 18:39:31 +00:00
Pavel Yosifovich
127 lines
3.2 KiB
C++
127 lines
3.2 KiB
C++
#include "pch.h"
|
|
#include "resource.h"
|
|
#include "NewSubLayerDlg.h"
|
|
#include "WFPHelper.h"
|
|
#include "StringHelper.h"
|
|
#include <WFPEngine.h>
|
|
#include <WFPEnumerators.h>
|
|
|
|
CNewSubLayerDlg::CNewSubLayerDlg(FWPM_SUBLAYER* sublayer) : m_SubLayer(sublayer) {
|
|
}
|
|
|
|
bool CNewSubLayerDlg::IsNewSubLayer() const {
|
|
return m_SubLayer->subLayerKey == GUID_NULL;
|
|
}
|
|
|
|
bool CNewSubLayerDlg::UpdateSubLayer() {
|
|
GetDlgItemText(IDC_NAME, m_Name);
|
|
if (m_Name.IsEmpty()) {
|
|
AtlMessageBox(m_hWnd, L"Sublayer name cannot be empty.", L"New Sublayer", MB_ICONWARNING);
|
|
return false;
|
|
}
|
|
|
|
CString key;
|
|
GetDlgItemText(IDC_KEY, key);
|
|
GetDlgItemText(IDC_DESC, m_Desc);
|
|
m_SubLayer->displayData.name = m_Name.GetBuffer();
|
|
m_SubLayer->displayData.description = m_Desc.GetBuffer();
|
|
::CLSIDFromString(key, &m_SubLayer->subLayerKey);
|
|
|
|
GetDlgItemText(IDC_WEIGHT, key);
|
|
m_SubLayer->weight = (UINT16)wcstol(key, nullptr, 0);
|
|
|
|
//
|
|
// get provider
|
|
//
|
|
auto cb = (CComboBox)GetDlgItem(IDC_PROVIDER);
|
|
if (cb.GetCurSel()) {
|
|
CString text;
|
|
cb.GetLBText(cb.GetCurSel(), text);
|
|
m_ProviderKey = StringHelper::ExtractGuid(text);
|
|
ATLASSERT(m_ProviderKey != GUID_NULL);
|
|
m_SubLayer->providerKey = &m_ProviderKey;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
LRESULT CNewSubLayerDlg::OnInitDialog(UINT, WPARAM, LPARAM, BOOL&) {
|
|
using namespace std;
|
|
using namespace std::views;
|
|
using namespace std::ranges;
|
|
|
|
SetDialogIcon(IDI_SUBLAYER);
|
|
WFPEngine engine;
|
|
if (engine.Open()) {
|
|
//
|
|
// add providers to combo
|
|
//
|
|
WFPProviderEnumerator en(engine.Handle());
|
|
auto providers = en.Next(256);
|
|
struct Data {
|
|
CString text;
|
|
int type;
|
|
};
|
|
int selected = -1, i = 0;
|
|
auto data = providers | std::views::transform([&](auto& p) {
|
|
if (selected < 0 && m_SubLayer->providerKey && IsNewSubLayer() && p->providerKey == *m_SubLayer->providerKey)
|
|
selected = i;
|
|
i++;
|
|
|
|
return Data{ p->displayData.name + CString(" (") + StringHelper::GuidToString(p->providerKey) + L")", i - 1 };
|
|
}) | to<vector>();
|
|
|
|
data.insert(data.begin(), { L"(None)", -1 });
|
|
InitCombo(IDC_PROVIDER, data.data(), (int)data.size(), selected);
|
|
}
|
|
|
|
if (IsNewSubLayer()) {
|
|
SetDlgItemInt(IDC_WEIGHT, 0);
|
|
}
|
|
else {
|
|
SetDlgItemText(IDC_NAME, m_SubLayer->displayData.name);
|
|
SetDlgItemText(IDC_DESC, m_SubLayer->displayData.description);
|
|
SetDlgItemText(IDC_KEY, StringHelper::GuidToString(m_SubLayer->subLayerKey));
|
|
if (m_SubLayer->flags & FWPM_SUBLAYER_FLAG_PERSISTENT)
|
|
CheckDlgButton(IDC_PERSISTENT, BST_CHECKED);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
LRESULT CNewSubLayerDlg::OnCloseCmd(WORD, WORD wID, HWND, BOOL&) {
|
|
if (wID == IDOK) {
|
|
if (!UpdateSubLayer())
|
|
return 0;
|
|
}
|
|
EndDialog(wID);
|
|
return 0;
|
|
}
|
|
|
|
LRESULT CNewSubLayerDlg::OnGenerateKey(WORD, WORD wID, HWND, BOOL&) {
|
|
GUID guid;
|
|
::CoCreateGuid(&guid);
|
|
WCHAR sguid[64];
|
|
::StringFromGUID2(guid, sguid, _countof(sguid));
|
|
SetDlgItemText(IDC_KEY, sguid);
|
|
|
|
return 0;
|
|
}
|
|
|
|
LRESULT CNewSubLayerDlg::OnProviderProperties(WORD, WORD wID, HWND, BOOL&) {
|
|
auto cb = (CComboBox)GetDlgItem(IDC_PROVIDER);
|
|
auto selected = cb.GetCurSel();
|
|
if (selected == 0)
|
|
return 0;
|
|
|
|
CString text;
|
|
cb.GetLBText(selected, text);
|
|
auto guid = StringHelper::ExtractGuid(text);
|
|
|
|
WFPEngine engine;
|
|
engine.Open();
|
|
WFPHelper::ShowProviderProperties(engine, *engine.GetProviderByKey(guid));
|
|
|
|
return 0;
|
|
}
|