Skip to content

Commit

Permalink
* Fix crashing bug in event vector cleanup
Browse files Browse the repository at this point in the history
* Fix warnings in settings.cpp
* Add settings to ui of powertoys
  • Loading branch information
Chris Davis committed Nov 12, 2019
1 parent 1e89054 commit e328c5d
Show file tree
Hide file tree
Showing 8 changed files with 476 additions and 418 deletions.
379 changes: 190 additions & 189 deletions src/modules/powerrename/dll/PowerRenameExt.vcxproj

Large diffs are not rendered by default.

49 changes: 49 additions & 0 deletions src/modules/powerrename/dll/dllmain.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -207,13 +207,62 @@ class PowerRenameModule : public PowertoyModuleIface
// Link to the GitHub PowerRename sub-page
settings.set_overview_link(L"https://github.com/microsoft/PowerToys/tree/master/src/modules/powerrename");

settings.add_bool_toogle(
L"bool_persist_input",
L"Restore search, replace and flags values on launch from previous run.",
CSettings::GetPersistState()
);

settings.add_bool_toogle(
L"bool_mru_enabled",
L"Enable autocomplete and autosuggest of recently used inputs for search and replace values.",
CSettings::GetMRUEnabled()
);

settings.add_int_spinner(
L"int_max_mru_size",
L"Maximum number of items to show in recently used list for autocomplete dropdown.",
CSettings::GetMaxMRUSize(),
0,
20,
1
);

settings.add_bool_toogle(
L"bool_show_icon_on_menu",
L"Show icon on context menu.",
CSettings::GetShowIconOnMenu()
);

settings.add_bool_toogle(
L"bool_show_extended_menu",
L"Only show the PowerRename menu item on the extended context menu (SHIFT + Right-click).",
CSettings::GetExtendedContextMenuOnly()
);

return settings.serialize_to_buffer(buffer, buffer_size);
}

// Passes JSON with the configuration settings for the powertoy.
// This is called when the user hits Save on the settings page.
virtual void set_config(PCWSTR config) override
{
try {
// Parse the input JSON string.
PowerToysSettings::PowerToyValues values =
PowerToysSettings::PowerToyValues::from_json_string(config);

CSettings::SetPersistState(values.get_bool_value(L"bool_persist_input"));
CSettings::SetMRUEnabled(values.get_bool_value(L"bool_mru_enabled"));
CSettings::SetMaxMRUSize(values.get_int_value(L"int_max_mru_size"));
CSettings::SetShowIconOnMenu(values.get_bool_value(L"bool_show_icon_on_menu"));
CSettings::SetExtendedContextMenuOnly(values.get_int_value(L"bool_show_extended_menu"));

values.save_to_settings_file();
}
catch (std::exception & ex) {
// Improper JSON.
}
}

// Signal from the Settings editor to call a custom action.
Expand Down
52 changes: 28 additions & 24 deletions src/modules/powerrename/lib/PowerRenameManager.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,7 @@ IFACEMETHODIMP CPowerRenameManager::Advise(_In_ IPowerRenameManagerEvents* renam
srme.cookie = m_cookie;
srme.pEvents = renameOpEvents;
renameOpEvents->AddRef();
m_PowerRenameManagerEvents.push_back(srme);
m_powerRenameManagerEvents.push_back(srme);

*cookie = m_cookie;

Expand All @@ -60,16 +60,16 @@ IFACEMETHODIMP CPowerRenameManager::UnAdvise(_In_ DWORD cookie)
HRESULT hr = E_FAIL;
CSRWExclusiveAutoLock lock(&m_lockEvents);

for (auto it : m_PowerRenameManagerEvents)
for (std::vector<RENAME_MGR_EVENT>::iterator it = m_powerRenameManagerEvents.begin(); it != m_powerRenameManagerEvents.end(); ++it)
{
if (it.cookie == cookie)
if (it->cookie == cookie)
{
hr = S_OK;
it.cookie = 0;
if (it.pEvents)
it->cookie = 0;
if (it->pEvents)
{
it.pEvents->Release();
it.pEvents = nullptr;
it->pEvents->Release();
it->pEvents = nullptr;
}
break;
}
Expand Down Expand Up @@ -921,7 +921,7 @@ void CPowerRenameManager::_OnItemAdded(_In_ IPowerRenameItem* renameItem)
{
CSRWSharedAutoLock lock(&m_lockEvents);

for (auto it : m_PowerRenameManagerEvents)
for (auto it : m_powerRenameManagerEvents)
{
if (it.pEvents)
{
Expand All @@ -934,7 +934,7 @@ void CPowerRenameManager::_OnUpdate(_In_ IPowerRenameItem* renameItem)
{
CSRWSharedAutoLock lock(&m_lockEvents);

for (auto it : m_PowerRenameManagerEvents)
for (auto it : m_powerRenameManagerEvents)
{
if (it.pEvents)
{
Expand All @@ -947,7 +947,7 @@ void CPowerRenameManager::_OnError(_In_ IPowerRenameItem* renameItem)
{
CSRWSharedAutoLock lock(&m_lockEvents);

for (auto it : m_PowerRenameManagerEvents)
for (auto it : m_powerRenameManagerEvents)
{
if (it.pEvents)
{
Expand All @@ -960,7 +960,7 @@ void CPowerRenameManager::_OnRegExStarted(_In_ DWORD threadId)
{
CSRWSharedAutoLock lock(&m_lockEvents);

for (auto it : m_PowerRenameManagerEvents)
for (auto it : m_powerRenameManagerEvents)
{
if (it.pEvents)
{
Expand All @@ -973,7 +973,7 @@ void CPowerRenameManager::_OnRegExCanceled(_In_ DWORD threadId)
{
CSRWSharedAutoLock lock(&m_lockEvents);

for (auto it : m_PowerRenameManagerEvents)
for (auto it : m_powerRenameManagerEvents)
{
if (it.pEvents)
{
Expand All @@ -986,7 +986,7 @@ void CPowerRenameManager::_OnRegExCompleted(_In_ DWORD threadId)
{
CSRWSharedAutoLock lock(&m_lockEvents);

for (auto it : m_PowerRenameManagerEvents)
for (auto it : m_powerRenameManagerEvents)
{
if (it.pEvents)
{
Expand All @@ -999,7 +999,7 @@ void CPowerRenameManager::_OnRenameStarted()
{
CSRWSharedAutoLock lock(&m_lockEvents);

for (auto it : m_PowerRenameManagerEvents)
for (auto it : m_powerRenameManagerEvents)
{
if (it.pEvents)
{
Expand All @@ -1012,7 +1012,7 @@ void CPowerRenameManager::_OnRenameCompleted()
{
CSRWSharedAutoLock lock(&m_lockEvents);

for (auto it : m_PowerRenameManagerEvents)
for (auto it : m_powerRenameManagerEvents)
{
if (it.pEvents)
{
Expand All @@ -1026,28 +1026,32 @@ void CPowerRenameManager::_ClearEventHandlers()
CSRWExclusiveAutoLock lock(&m_lockEvents);

// Cleanup event handlers
for (auto it : m_PowerRenameManagerEvents)
for (std::vector<RENAME_MGR_EVENT>::iterator it = m_powerRenameManagerEvents.begin(); it != m_powerRenameManagerEvents.end(); ++it)
{
it.cookie = 0;
if (it.pEvents)
it->cookie = 0;
if (it->pEvents)
{
it.pEvents->Release();
it.pEvents = nullptr;
it->pEvents->Release();
it->pEvents = nullptr;
}
}

m_PowerRenameManagerEvents.clear();
m_powerRenameManagerEvents.clear();
}

void CPowerRenameManager::_ClearPowerRenameItems()
{
CSRWExclusiveAutoLock lock(&m_lockItems);

// Cleanup rename items
for (auto it : m_renameItems)
for (std::map<int, IPowerRenameItem*>::iterator it = m_renameItems.begin(); it != m_renameItems.end(); ++it)
{
IPowerRenameItem* pItem = it.second;
pItem->Release();
IPowerRenameItem* pItem = it->second;
if (pItem)
{
pItem->Release();
it->second = nullptr;
}
}

m_renameItems.clear();
Expand Down
2 changes: 1 addition & 1 deletion src/modules/powerrename/lib/PowerRenameManager.h
Original file line number Diff line number Diff line change
Expand Up @@ -108,7 +108,7 @@ class CPowerRenameManager :
CComPtr<IPowerRenameItemFactory> m_spItemFactory;
CComPtr<IPowerRenameRegEx> m_spRegEx;

_Guarded_by_(m_lockEvents) std::vector<RENAME_MGR_EVENT> m_PowerRenameManagerEvents;
_Guarded_by_(m_lockEvents) std::vector<RENAME_MGR_EVENT> m_powerRenameManagerEvents;
_Guarded_by_(m_lockItems) std::map<int, IPowerRenameItem*> m_renameItems;

// Parent HWND used by IFileOperation
Expand Down
12 changes: 6 additions & 6 deletions src/modules/powerrename/lib/PowerRenameRegEx.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -53,16 +53,16 @@ IFACEMETHODIMP CPowerRenameRegEx::UnAdvise(_In_ DWORD cookie)
HRESULT hr = E_FAIL;
CSRWExclusiveAutoLock lock(&m_lockEvents);

for (auto it : m_renameRegExEvents)
for (std::vector<RENAME_REGEX_EVENT>::iterator it = m_renameRegExEvents.begin(); it != m_renameRegExEvents.end(); ++it)
{
if (it.cookie == cookie)
if (it->cookie == cookie)
{
hr = S_OK;
it.cookie = 0;
if (it.pEvents)
it->cookie = 0;
if (it->pEvents)
{
it.pEvents->Release();
it.pEvents = nullptr;
it->pEvents->Release();
it->pEvents = nullptr;
}
break;
}
Expand Down
4 changes: 1 addition & 3 deletions src/modules/powerrename/lib/Settings.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ class CRenameMRU :

// IEnumString
IFACEMETHODIMP Next(__in ULONG celt, __out_ecount_part(celt, *pceltFetched) LPOLESTR* rgelt, __out_opt ULONG* pceltFetched);
IFACEMETHODIMP Skip(__in ULONG celt) { return E_NOTIMPL; }
IFACEMETHODIMP Skip(__in ULONG) { return E_NOTIMPL; }
IFACEMETHODIMP Reset();
IFACEMETHODIMP Clone(__deref_out IEnumString** ppenum) { *ppenum = nullptr; return E_NOTIMPL; }

Expand Down Expand Up @@ -380,8 +380,6 @@ IFACEMETHODIMP CRenameMRU::AddMRUString(_In_ PCWSTR entry)

HRESULT CRenameMRU::_CreateMRUList(_In_ MRUINFO* pmi)
{
HRESULT hr = E_FAIL;

if (m_mruHandle != NULL)
{
_FreeMRUList();
Expand Down

0 comments on commit e328c5d

Please sign in to comment.