mpr: Enumerate connections capability before looking for functions entry points.

Signed-off-by: Pierre Schweitzer <pierre@reactos.org> Signed-off-by: Alexandre Julliard <julliard@winehq.org>

mpr: Enumerate connections capability before looking for functions entry points.
96a93ee6 · Pierre Schweitzer · Alexandre Julliard · d961f5ac · 96a93ee6
Commit 96a93ee6 authored Dec 30, 2016 by Pierre Schweitzer Committed by Alexandre Julliard Jan 02, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 8 additions and 3 deletions

wnet.c dlls/mpr/wnet.c +8 -3

No files found.
--- a/dlls/mpr/wnet.c
+++ b/dlls/mpr/wnet.c
@@ -178,6 +178,7 @@ static void _tryLoadProvider(PCWSTR provider)
                    TRACE("loaded lib %p\n", hLib);
                    if (getCaps)
                    {
+                        DWORD connectCap;
                        PWNetProvider provider =
                         &providerTable->table[providerTable->numProviders];

@@ -211,9 +212,13 @@ static void _tryLoadProvider(PCWSTR provider)
                                WARN("Couldn't load enumeration functions\n");
                            }
                        }
-                        provider->addConnection = MPR_GETPROC(NPAddConnection);
-                        provider->addConnection3 = MPR_GETPROC(NPAddConnection3);
-                        provider->cancelConnection = MPR_GETPROC(NPCancelConnection);
+                        connectCap = getCaps(WNNC_CONNECTION);
+                        if (connectCap & WNNC_CON_ADDCONNECTION)
+                            provider->addConnection = MPR_GETPROC(NPAddConnection);
+                        if (connectCap & WNNC_CON_ADDCONNECTION3)
+                            provider->addConnection3 = MPR_GETPROC(NPAddConnection3);
+                        if (connectCap & WNNC_CON_CANCELCONNECTION)
+                            provider->cancelConnection = MPR_GETPROC(NPCancelConnection);
                        TRACE("NPAddConnection %p\n", provider->addConnection);
                        TRACE("NPAddConnection3 %p\n", provider->addConnection3);
                        TRACE("NPCancelConnection %p\n", provider->cancelConnection);