Bug 661991 - dynamically load entry point to SHCreateItemFromParsingName and share this with jump list code. r=ehsan.

widget/src/windows/JumpListItem.cpp

@@ -55,16 +55,12 @@
 #include "mozIAsyncFavicons.h"
 #include "mozilla/Preferences.h"
 #include "JumpListBuilder.h"
+#include "nsToolkit.h"
 
 namespace mozilla {
 namespace widget {
 
-// SHCreateItemFromParsingName is only available on vista and up. We only load this if we
-// need to call it on win7+.
-JumpListLink::SHCreateItemFromParsingNamePtr JumpListLink::createItemFromParsingName = nsnull;
-const PRUnichar JumpListLink::kSehllLibraryName[] =  L"shell32.dll";
 const char JumpListItem::kJumpListCacheDir[] = "jumpListCache";
-HMODULE JumpListLink::sShellDll = nsnull;
 
 // ISUPPORTS Impl's
 NS_IMPL_ISUPPORTS1(JumpListItem,
@@ -742,18 +738,11 @@ nsresult JumpListLink::GetShellItem(nsCOMPtr<nsIJumpListItem>& item, nsRefPtr<IS
   NS_ENSURE_SUCCESS(rv, rv);
 
   // Load vista+ SHCreateItemFromParsingName
-  if (createItemFromParsingName == nsnull) {
-    if (sShellDll)
-      return NS_ERROR_UNEXPECTED;
-    sShellDll = ::LoadLibraryW(kSehllLibraryName);
-    if (sShellDll)
-      createItemFromParsingName = (SHCreateItemFromParsingNamePtr)GetProcAddress(sShellDll, "SHCreateItemFromParsingName");
-    if (createItemFromParsingName == nsnull)
-      return NS_ERROR_UNEXPECTED;
-  }
+  if (!nsToolkit::VistaCreateItemFromParsingNameInit())
+    return NS_ERROR_UNEXPECTED;
 
   // Create the IShellItem
-  if (FAILED(createItemFromParsingName(NS_ConvertASCIItoUTF16(spec).get(),
+  if (FAILED(nsToolkit::createItemFromParsingName(NS_ConvertASCIItoUTF16(spec).get(),
              NULL, IID_PPV_ARGS(&psi))))
     return NS_ERROR_INVALID_ARG;
 

widget/src/windows/JumpListItem.h

@@ -118,14 +118,9 @@ public:
   static nsresult GetJumpListLink(IShellItem *pItem, nsCOMPtr<nsIJumpListLink>& aLink);
 
 protected:
-  typedef HRESULT (WINAPI * SHCreateItemFromParsingNamePtr)(PCWSTR pszPath, IBindCtx *pbc, REFIID riid, void **ppv);
-
   nsString mUriTitle;
   nsCOMPtr<nsIURI> mURI;
   nsCOMPtr<nsICryptoHash> mCryptoHash;
-  static const PRUnichar kSehllLibraryName[];
-  static HMODULE sShellDll;
-  static SHCreateItemFromParsingNamePtr createItemFromParsingName;
 };
 
 class JumpListShortcut : public JumpListItem, public nsIJumpListShortcut

widget/src/windows/nsFilePicker.cpp

@@ -68,6 +68,7 @@ static const DWORD kDialogTimerID = 9999;
 static const unsigned long kDialogTimerTimeout = 300;
 
 #define MAX_EXTENSION_LENGTH 10
+#define FILE_BUFFER_SIZE     4096 
 
 ///////////////////////////////////////////////////////////////////////////////
 // Helper classes
@@ -612,11 +613,12 @@ nsFilePicker::ShowFolderPicker(const nsString& aInitialDir)
  
   // initial strings
   dialog->SetTitle(mTitle.get());
-  if (!aInitialDir.IsEmpty()) {
+  if (!aInitialDir.IsEmpty() &&
+      nsToolkit::VistaCreateItemFromParsingNameInit()) {
     nsRefPtr<IShellItem> folder;
-    if (SUCCEEDED(SHCreateItemFromParsingName(aInitialDir.get(), NULL,
-                                              IID_IShellItem,
-                                              getter_AddRefs(folder)))) {
+    if (SUCCEEDED(nsToolkit::createItemFromParsingName(aInitialDir.get(), NULL,
+                                                       IID_IShellItem,
+                                                       getter_AddRefs(folder)))) {
       dialog->SetFolder(folder);
     }
   }
@@ -938,11 +940,12 @@ nsFilePicker::ShowFilePicker(const nsString& aInitialDir)
   }
 
   // initial location
-  if (!aInitialDir.IsEmpty()) {
+  if (!aInitialDir.IsEmpty() &&
+      nsToolkit::VistaCreateItemFromParsingNameInit()) {
     nsRefPtr<IShellItem> folder;
-    if (SUCCEEDED(SHCreateItemFromParsingName(aInitialDir.get(), NULL,
-                                              IID_IShellItem,
-                                              getter_AddRefs(folder)))) {
+    if (SUCCEEDED(nsToolkit::createItemFromParsingName(aInitialDir.get(), NULL,
+                                                       IID_IShellItem,
+                                                       getter_AddRefs(folder)))) {
       dialog->SetFolder(folder);
     }
   }

widget/src/windows/nsToolkit.cpp

@@ -54,11 +54,14 @@
 #include <unknwn.h>
 
 nsToolkit* nsToolkit::gToolkit = nsnull;
-
 HINSTANCE nsToolkit::mDllInstance = 0;
-
 static const unsigned long kD3DUsageDelay = 5000;
 
+// SHCreateItemFromParsingName is only available on vista and up.
+nsToolkit::SHCreateItemFromParsingNamePtr nsToolkit::createItemFromParsingName = nsnull;
+const PRUnichar nsToolkit::kSehllLibraryName[] =  L"shell32.dll";
+HMODULE nsToolkit::sShellDll = nsnull;
+
 static void
 StartAllowingD3D9(nsITimer *aTimer, void *aClosure)
 {
@@ -122,6 +125,24 @@ nsToolkit::StartAllowingD3D9()
   nsWindow::StartAllowingD3D9(false);
 }
 
+// Load and store Vista+ SHCreateItemFromParsingName
+bool
+nsToolkit::VistaCreateItemFromParsingNameInit()
+{
+  if (createItemFromParsingName)
+    return true;
+  if (sShellDll)
+    return false;
+  sShellDll = LoadLibraryW(kSehllLibraryName);
+  if (!sShellDll)
+    return false;
+  createItemFromParsingName = (SHCreateItemFromParsingNamePtr)
+    GetProcAddress(sShellDll, "SHCreateItemFromParsingName");
+  if (createItemFromParsingName == nsnull)
+    return false;
+  return true;
+}
+
 //-------------------------------------------------------------------------
 //
 // Return the nsToolkit for the current thread.  If a toolkit does not

widget/src/windows/nsToolkit.h

@@ -43,7 +43,8 @@
 
 #include "nsITimer.h"
 #include "nsCOMPtr.h"
-
+#include <windows.h>
+#include <shobjidl.h>
 #include <imm.h>
 
 // Avoid including windowsx.h to prevent macro pollution
@@ -54,15 +55,6 @@
 #define GET_Y_LPARAM(pt) (short(HIWORD(pt)))
 #endif
 
-// we used to use MAX_PATH
-// which works great for one file
-// but for multiple files, the format is
-// dirpath\0\file1\0file2\0...filen\0\0
-// and that can quickly be more than MAX_PATH (260)
-// see bug #172001 for more details
-#define FILE_BUFFER_SIZE 4096 
-
-
 /**
  * Makes sure exit/enter mouse messages are always dispatched.
  * In the case where the mouse has exited the outer most window the
@@ -121,12 +113,18 @@ public:
     static void Startup(HMODULE hModule);
     static void Shutdown();
     static void StartAllowingD3D9();
+    static bool VistaCreateItemFromParsingNameInit();
+
+    typedef HRESULT (WINAPI * SHCreateItemFromParsingNamePtr)(PCWSTR pszPath, IBindCtx *pbc, REFIID riid, void **ppv);
+    static SHCreateItemFromParsingNamePtr createItemFromParsingName;
 
 protected:
     static nsToolkit* gToolkit;
 
     nsCOMPtr<nsITimer> mD3D9Timer;
     MouseTrailer mMouseTrailer;
+    static const PRUnichar kSehllLibraryName[];
+    static HMODULE sShellDll;
 };
 
 #endif  // TOOLKIT_H