comparison src/helper.cpp @ 38:a43d7a2867d2

Tidy up error handling, especially on Windows

 static const char programName[] = "vamp-plugin-load-checker";
 
 #ifdef _WIN32
 #include <windows.h>
 #include <process.h>
+#endif
+
 #include <string>
-#ifdef UNICODE
+#include <iostream>
+
+#ifdef _WIN32
+#ifndef UNICODE
+#error "This must be compiled with UNICODE defined"
+#endif
 static std::string lastLibraryName = "";
 static HMODULE LoadLibraryUTF8(std::string name) {
     lastLibraryName = name;
     int n = name.size();
     int wn = MultiByteToWideChar(CP_UTF8, 0, name.c_str(), n, 0, 0);

     HMODULE h = LoadLibraryW(wname);
     delete[] wname;
     return h;
 }
 static std::string GetErrorText() {
+    DWORD err = GetLastError();
     wchar_t *buffer;
-    DWORD err = GetLastError();
-    FormatMessage(
+    FormatMessageW(
         FORMAT_MESSAGE_ALLOCATE_BUFFER |
         FORMAT_MESSAGE_FROM_SYSTEM |
         FORMAT_MESSAGE_IGNORE_INSERTS,
         NULL,
         err,
         MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT),
-        (LPTSTR) &buffer,
+        (LPWSTR) &buffer,
         0, NULL );
     int wn = wcslen(buffer);
     int n = WideCharToMultiByte(CP_UTF8, 0, buffer, wn, 0, 0, 0, 0);
     if (n < 0) {
         LocalFree(&buffer);

         s.replace(pos, 2, lastLibraryName);
     }
     return s;
 }
 #define DLOPEN(a,b)  LoadLibraryUTF8(a)
-#else
-#define DLOPEN(a,b)  LoadLibrary((a).c_str())
-#define GetErrorText() ""
-#endif
 #define DLSYM(a,b)   (void *)GetProcAddress((HINSTANCE)(a),(b).c_str())
 #define DLCLOSE(a)   (!FreeLibrary((HINSTANCE)(a)))
 #define DLERROR()    (GetErrorText())
 #else
 #include <dlfcn.h>

 #define DLSYM(a,b)   dlsym((a),(b).c_str())
 #define DLCLOSE(a)   dlclose((a))
 #define DLERROR()    dlerror()
 #endif
 
-#include <string>
-#include <iostream>
-
 //#include <unistd.h>
 
 using namespace std;
 
 string error()

     typedef const void *(*DFn)(unsigned long);
     DFn fn = DFn(f);
     unsigned long index = 0;
     while (fn(index)) ++index;
     if (index == 0) return "Library contains no plugins";
-//    else cerr << "Library contains " << index << " plugin(s)" << endl;
     return "";
 }
 
 string checkVampDescriptorFn(void *f)
 {
     typedef const void *(*DFn)(unsigned int, unsigned int);
     DFn fn = DFn(f);
     unsigned int index = 0;
     while (fn(2, index)) ++index;
     if (index == 0) return "Library contains no plugins";
-//    else cerr << "Library contains " << index << " plugin(s)" << endl;
     return "";
 }
 
 string check(string soname, string descriptor)
 {
     void *handle = DLOPEN(soname, RTLD_NOW | RTLD_LOCAL);
     if (!handle) {
         return "Unable to open plugin library: " + error();
     }
 
+    string msg = "";
+    
     void *fn = DLSYM(handle, descriptor);
     if (!fn) {
-        return "Failed to find plugin descriptor " + descriptor +
+        msg = "Failed to find plugin descriptor " + descriptor +
             " in library: " + error();
-    }
-
-    if (descriptor == "ladspa_descriptor") {
-        return checkLADSPAStyleDescriptorFn(fn);
+    } else if (descriptor == "ladspa_descriptor") {
+        msg = checkLADSPAStyleDescriptorFn(fn);
     } else if (descriptor == "dssi_descriptor") {
-        return checkLADSPAStyleDescriptorFn(fn);
+        msg = checkLADSPAStyleDescriptorFn(fn);
     } else if (descriptor == "vampGetPluginDescriptor") {
-        return checkVampDescriptorFn(fn);
+        msg = checkVampDescriptorFn(fn);
     } else {
         cerr << "Note: no descriptor logic known for descriptor function \""
              << descriptor << "\"; not actually calling it" << endl;
     }
-    
-    return "";
+
+    DLCLOSE(handle);
+    
+    return msg;
 }
 
 int main(int argc, char **argv)
 {
     bool allGood = true;