From e119672f1ca372e6ef6c91d5526d93398e29b6a5 Mon Sep 17 00:00:00 2001 From: spx Date: Mon, 14 Nov 2011 14:32:11 +0000 Subject: [PATCH] Minor tweaks. --- Library/ConfigParser.cpp | 8 +----- Library/ConfigParser.h | 2 +- Library/Litestep.cpp | 2 +- Library/System.cpp | 58 +++++++++++++++++++++++++--------------- 4 files changed, 39 insertions(+), 31 deletions(-) diff --git a/Library/ConfigParser.cpp b/Library/ConfigParser.cpp index 3d2de5dc..708d7bcd 100644 --- a/Library/ConfigParser.cpp +++ b/Library/ConfigParser.cpp @@ -696,14 +696,8 @@ std::vector CConfigParser::ReadFloats(LPCTSTR section, LPCTSTR ke const std::wstring& string = ReadString(section, key, L""); if (!string.empty()) { - std::wstring tmp = string; - if (tmp[tmp.length() - 1] != L';') - { - tmp += L";"; - } - // Tokenize and parse the floats - std::vector tokens = Tokenize(tmp, L";"); + std::vector tokens = Tokenize(string, L";"); std::vector::const_iterator iter = tokens.begin(); for ( ; iter != tokens.end(); ++iter) { diff --git a/Library/ConfigParser.h b/Library/ConfigParser.h index a36db4d1..109ef68c 100644 --- a/Library/ConfigParser.h +++ b/Library/ConfigParser.h @@ -51,7 +51,7 @@ public: void SetValue(const std::wstring& strSection, const std::wstring& strKey, const std::wstring& strValue); void DeleteValue(const std::wstring& strSection, const std::wstring& strKey); - void SetStyleTemplate(const std::wstring& strStyle) { m_StyleTemplate = Tokenize(strStyle, L"|"); Shrink(m_StyleTemplate); } + void SetStyleTemplate(const std::wstring& strStyle) { Tokenize(strStyle, L"|").swap(m_StyleTemplate); Shrink(m_StyleTemplate); } void ClearStyleTemplate() { m_StyleTemplate.clear(); } const std::wstring& GetLastUsedStyle() { return m_LastUsedStyle; } diff --git a/Library/Litestep.cpp b/Library/Litestep.cpp index 154ea11d..a54a1c4c 100644 --- a/Library/Litestep.cpp +++ b/Library/Litestep.cpp @@ -323,7 +323,7 @@ BOOL LogInternal(int nLevel, ULONGLONG elapsed, LPCTSTR pszMessage) // The stub implementation if (Rainmeter->GetLogging()) { - std::wstring logfile = Rainmeter->GetLogFile(); + const std::wstring& logfile = Rainmeter->GetLogFile(); if (logFound == 0) { // Check if the file exists diff --git a/Library/System.cpp b/Library/System.cpp index f0b96036..cf929aae 100644 --- a/Library/System.cpp +++ b/Library/System.cpp @@ -1180,11 +1180,14 @@ bool CSystem::CopyFiles(const std::wstring& strFrom, const std::wstring& strTo, tmpFrom.append(1, L'\0'); tmpTo.append(1, L'\0'); - SHFILEOPSTRUCT fo = {0}; - fo.wFunc = bMove ? FO_MOVE : FO_COPY; - fo.pFrom = tmpFrom.c_str(); - fo.pTo = tmpTo.c_str(); - fo.fFlags = FOF_NO_UI | FOF_NOCONFIRMATION | FOF_ALLOWUNDO; + SHFILEOPSTRUCT fo = + { + NULL, + bMove ? FO_MOVE : FO_COPY, + tmpFrom.c_str(), + tmpTo.c_str(), + FOF_NO_UI | FOF_NOCONFIRMATION | FOF_ALLOWUNDO + }; int result = SHFileOperation(&fo); if (result != 0) @@ -1226,10 +1229,14 @@ bool CSystem::RemoveFolder(const std::wstring& strFolder) // The strings must end with double nul tmpFolder.append(1, L'\0'); - SHFILEOPSTRUCT fo = {0}; - fo.wFunc = FO_DELETE; - fo.pFrom = tmpFolder.c_str(); - fo.fFlags = FOF_NO_UI | FOF_NOCONFIRMATION | FOF_ALLOWUNDO; + SHFILEOPSTRUCT fo = + { + NULL, + FO_DELETE, + tmpFolder.c_str(), + NULL, + FOF_NO_UI | FOF_NOCONFIRMATION | FOF_ALLOWUNDO + }; int result = SHFileOperation(&fo); if (result != 0) @@ -1287,7 +1294,7 @@ void CSystem::UpdateIniFileMappingList() c_IniFileMappings.clear(); } - WCHAR buffer[MAX_PATH]; + WCHAR* buffer = new WCHAR[MAX_PATH]; DWORD index = 0, cch = MAX_PATH; while ((ret = RegEnumKeyEx(hKey, index++, buffer, &cch, NULL, NULL, NULL, NULL)) != ERROR_NO_MORE_ITEMS) @@ -1298,6 +1305,8 @@ void CSystem::UpdateIniFileMappingList() } cch = MAX_PATH; } + + delete [] buffer; } RegCloseKey(hKey); @@ -1326,31 +1335,36 @@ std::wstring CSystem::GetTemporaryFile(const std::wstring& iniFile) { if (_wcsicmp((*iter).c_str(), filename) == 0) { - WCHAR buffer[MAX_PATH]; + WCHAR* buffer = new WCHAR[MAX_PATH]; - GetTempPath(MAX_PATH, buffer); - temporary = buffer; - if (GetTempFileName(temporary.c_str(), L"cfg", 0, buffer) != 0) + if (GetTempPath(MAX_PATH, buffer) != 0 && + GetTempFileName(buffer, L"cfg", 0, buffer) != 0) { temporary = buffer; std::wstring tmp = GetTemporaryFile(temporary); - if (tmp.empty() && CopyFiles(iniFile, temporary)) - { - return temporary; - } - else // temporary is reserved or failed + if (!tmp.empty() || !CopyFiles(iniFile, temporary)) // temporary is reserved or failed { RemoveFile(temporary); - if (tmp.empty()) tmp = L"?"; - return tmp; + + if (tmp.empty()) + { + temporary = L"?"; + } + else + { + temporary.swap(tmp); + } } } else // failed { LogWithArgs(LOG_ERROR, L"Unable to create temporary file to: %s", temporary.c_str()); - return L"?"; + temporary = L"?"; } + + delete [] buffer; + break; } } }