Chris@2: /* -*- c-basic-offset: 4 indent-tabs-mode: nil -*-  vi:set ts=8 sts=4 sw=4: */
Chris@0: 
Chris@0: /**
Chris@1:  * Plugin Load Checker Helper
Chris@0:  *
Chris@4:  * This program accepts the name of a descriptor symbol as its only
Chris@1:  * command-line argument. It then reads a list of plugin library paths
Chris@1:  * from stdin, one per line. For each path read, it attempts to load
Chris@4:  * that library and retrieve the named descriptor symbol, printing a
Chris@1:  * line to stdout reporting whether this was successful or not and
Chris@1:  * then flushing stdout. The output line format is described
Chris@1:  * below. The program exits with code 0 if all libraries were loaded
Chris@1:  * successfully and non-zero otherwise.
Chris@0:  *
Chris@0:  * Note that library paths must be ready to pass to dlopen() or
Chris@0:  * equivalent; this usually means they should be absolute paths.
Chris@0:  *
Chris@0:  * Output line for successful load of library libname.so:
Chris@0:  * SUCCESS|/path/to/libname.so|
Chris@0:  * 
Chris@0:  * Output line for failed load of library libname.so:
Chris@0:  * FAILURE|/path/to/libname.so|Reason for failure if available
Chris@0:  *
Chris@2:  * Sometimes plugins will crash completely on load, bringing down this
Chris@2:  * program with them. If the program exits before all listed plugins
Chris@2:  * have been checked, this means that the plugin following the last
Chris@2:  * reported one has crashed. Typically the caller may want to run it
Chris@2:  * again, omitting that plugin.
Chris@0:  */
Chris@0: 
Chris@5: /*
Chris@5:     Copyright (c) 2016 Queen Mary, University of London
Chris@5: 
Chris@5:     Permission is hereby granted, free of charge, to any person
Chris@5:     obtaining a copy of this software and associated documentation
Chris@5:     files (the "Software"), to deal in the Software without
Chris@5:     restriction, including without limitation the rights to use, copy,
Chris@5:     modify, merge, publish, distribute, sublicense, and/or sell copies
Chris@5:     of the Software, and to permit persons to whom the Software is
Chris@5:     furnished to do so, subject to the following conditions:
Chris@5: 
Chris@5:     The above copyright notice and this permission notice shall be
Chris@5:     included in all copies or substantial portions of the Software.
Chris@5: 
Chris@5:     THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
Chris@5:     EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
Chris@5:     MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
Chris@5:     NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
Chris@5:     CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
Chris@5:     CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
Chris@5:     WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
Chris@5: 
Chris@5:     Except as contained in this notice, the names of the Centre for
Chris@5:     Digital Music and Queen Mary, University of London shall not be
Chris@5:     used in advertising or otherwise to promote the sale, use or other
Chris@5:     dealings in this Software without prior written authorization.
Chris@5: */
Chris@5: 
Chris@0: #ifdef _WIN32
Chris@0: #include <windows.h>
Chris@0: #include <process.h>
Chris@0: #define DLOPEN(a,b)  LoadLibrary((a).toStdWString().c_str())
Chris@0: #define DLSYM(a,b)   GetProcAddress((HINSTANCE)(a),(b))
Chris@0: #define DLCLOSE(a)   (!FreeLibrary((HINSTANCE)(a)))
Chris@0: #define DLERROR()    ""
Chris@0: #else
Chris@0: #include <dlfcn.h>
Chris@0: #define DLOPEN(a,b)  dlopen((a).c_str(),(b))
Chris@0: #define DLSYM(a,b)   dlsym((a),(b).c_str())
Chris@0: #define DLCLOSE(a)   dlclose((a))
Chris@0: #define DLERROR()    dlerror()
Chris@0: #endif
Chris@0: 
Chris@0: #include <string>
Chris@0: #include <iostream>
Chris@0: 
Chris@6: #include <unistd.h>
Chris@6: 
Chris@0: using namespace std;
Chris@0: 
Chris@0: string error()
Chris@0: {
Chris@0:     string e = dlerror();
Chris@0:     if (e == "") return "(unknown error)";
Chris@0:     else return e;
Chris@0: }
Chris@0: 
Chris@1: string check(string soname, string descriptor)
Chris@0: {
Chris@0:     void *handle = DLOPEN(soname, RTLD_NOW | RTLD_LOCAL);
Chris@0:     if (!handle) {
Chris@0: 	return "Unable to open plugin library: " + error();
Chris@0:     }
Chris@0: 
Chris@0:     void *fn = DLSYM(handle, descriptor);
Chris@0:     if (!fn) {
Chris@0: 	return "Failed to find plugin descriptor " + descriptor +
Chris@0: 	    " in library: " + error();
Chris@0:     }
Chris@0: 
Chris@0:     return "";
Chris@0: }
Chris@0: 
Chris@0: int main(int argc, char **argv)
Chris@0: {
Chris@0:     bool allGood = true;
Chris@0:     string soname;
Chris@0: 
Chris@1:     if (argc != 2) {
Chris@1: 	cerr << "\nUsage:\n    " << argv[0] << " descriptorname\n"
Chris@4: 	    "\nwhere descriptorname is the name of a plugin descriptor symbol to be sought\n"
Chris@1: 	    "in each library (e.g. vampGetPluginDescriptor for Vamp plugins). The list of\n"
Chris@1: 	    "candidate plugin library filenames is read from stdin.\n" << endl;
Chris@1: 	return 2;
Chris@1:     }
Chris@1: 
Chris@1:     string descriptor = argv[1];
Chris@1:     
Chris@0:     while (getline(cin, soname)) {
Chris@1: 	string report = check(soname, descriptor);
Chris@0: 	if (report != "") {
Chris@0: 	    cout << "FAILURE|" << soname << "|" << report << endl;
Chris@0: 	    allGood = false;
Chris@0: 	} else {
Chris@0: 	    cout << "SUCCESS|" << soname << "|" << endl;
Chris@0: 	}
Chris@0:     }
Chris@0: 
Chris@0:     return allGood ? 0 : 1;
Chris@0: }