Issue 1109 - Support notifications

src/shared/Utils.cpp

 /*
  * This file is part of Adblock Plus <https://adblockplus.org/>,
  * Copyright (C) 2006-2015 Eyeo GmbH
  *
  * Adblock Plus is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License version 3 as
  * published by the Free Software Foundation.
  *
  * Adblock Plus is distributed in the hope that it will be useful,
  * but WITHOUT ANY WARRANTY; without even the implied warranty of
@@ skipping 73 matching lines @@
 }
 
 std::vector<std::wstring> ToUtf16Strings(const std::vector<std::string>& values)
 {
   std::vector<std::wstring> result;
   result.reserve(values.size());
   transform(values.begin(), values.end(), back_inserter(result), ToUtf16String);
   return result;
 }
 
+namespace
+{
+  std::wstring GetModulePath(HINSTANCE hInstance)
+  {
+    std::vector<WCHAR> path(MAX_PATH);
+    int length = GetModuleFileNameW(hInstance, &path[0], static_cast<DWORD>(path.size()));
+
+    while (length == path.size())
+    {
+      // Buffer too small, double buffer size
+      path.resize(path.size() * 2);
+      length = GetModuleFileNameW(hInstance, &path[0], static_cast<DWORD>(path.size()));
+    }
+
+    try
+    {
+      if (length == 0)
+        throw std::runtime_error("Failed determining module path");
+
+      std::vector<WCHAR>::reverse_iterator it = std::find(path.rbegin(), path.rend(), L'\\');
+      if (it == path.rend())
+        throw std::runtime_error("Unexpected plugin path, no backslash found");
+
+      return std::wstring(path.begin(), it.base());
+    }
+    catch (const std::exception&)
+    {
+      return std::wstring();
+    }
+  }
+}
+
 std::wstring GetDllDir()
 {
-  std::vector<WCHAR> path(MAX_PATH);
-  int length = GetModuleFileNameW((HINSTANCE)&__ImageBase, &path[0], static_cast<DWORD>(path.size()));
+  return GetModulePath((HINSTANCE)&__ImageBase);
+}
 
-  while (length == path.size())
-  {
-    // Buffer too small, double buffer size
-    path.resize(path.size() * 2);
-    length = GetModuleFileNameW((HINSTANCE)&__ImageBase, &path[0], static_cast<DWORD>(path.size()));
-  }
-
-  try
-  {
-    if (length == 0)
-      throw std::runtime_error("Failed determining module path");
-
-    std::vector<WCHAR>::reverse_iterator it = std::find(path.rbegin(), path.rend(), L'\\');
-    if (it == path.rend())
-      throw std::runtime_error("Unexpected plugin path, no backslash found");
-
-    return std::wstring(path.begin(), it.base());
-  }
-  catch (const std::exception&)
-  {
-    return std::wstring();
-  }
+std::wstring GetExeDir()
+{
+  return GetModulePath(nullptr);
 }
 
 std::wstring GetAppDataPath()
 {
   if (appDataPath.empty())
   {
     if (IsWindowsVistaOrLater())
     {
       WCHAR* pathBuffer;
       if (FAILED(SHGetKnownFolderPath(FOLDERID_LocalAppDataLow, 0, 0, &pathBuffer)))
         throw std::runtime_error("Unable to find app data directory");
       appDataPath.assign(pathBuffer);
       CoTaskMemFree(pathBuffer);
     }
     else
     {
       std::auto_ptr<wchar_t> pathBuffer(new wchar_t[MAX_PATH]);
       if (!SHGetSpecialFolderPathW(0, pathBuffer.get(), CSIDL_LOCAL_APPDATA, true))
         throw std::runtime_error("Unable to find app data directory");
       appDataPath.assign(pathBuffer.get());
     }
     appDataPath += L"\\Adblock Plus for IE";
 
     // Ignore errors here, this isn't a critical operation
     ::CreateDirectoryW(appDataPath.c_str(), NULL);
   }
   return appDataPath;
 }
 
-void ReplaceString(std::wstring& input, const std::wstring placeholder, const std::wstring replacement)
+void ReplaceString(std::wstring& input, const std::wstring& placeholder, const std::wstring& replacement)
 {
   size_t replaceStart = input.find(placeholder);
   if (replaceStart != std::string::npos)
   {
     input.replace(replaceStart, placeholder.length(), replacement);
   }
 }
 
 std::wstring GetSchemeAndHierarchicalPart(const std::wstring& url)
 {
@@ skipping 13 matching lines @@
     return L"";
   }
   auto queryStringBeginsAt = questionSignPos + 1;
   auto endQueryStringPos = url.find(L'#', queryStringBeginsAt);
   if (endQueryStringPos == std::wstring::npos)
   {
     endQueryStringPos = url.length();
   }
   return url.substr(queryStringBeginsAt, endQueryStringPos - queryStringBeginsAt);
 }