remotevstclient.cpp

// -*- c-basic-offset: 4 -*-

/*
  dssi-vst: a DSSI plugin wrapper for VST effects and instruments
  Copyright 2004-2010 Chris Cannam
*/

#include "remotevstclient.h"

#include <sys/un.h>
#include <sys/socket.h>
#include <sys/stat.h>
#include <sys/poll.h>
#include <fcntl.h>
#include <iostream>
#include <dirent.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <errno.h>
#include <cstdio>
#include <stdlib.h>

#include "rdwrops.h"
#include "paths.h"

RemoteVSTClient::RemoteVSTClient(std::string dllName, bool showGUI) :
    RemotePluginClient()
{
    pid_t child;
    std::string arg = dllName + "," + getFileIdentifiers();
    if (showGUI) arg = "-g " + arg;

    const char *argStr = arg.c_str();

    // We want to run the dssi-vst-server script, which runs wine
    // dssi-vst-server.exe.so.  We expect to find this script in the
    // same subdirectory of a directory in the DSSI_PATH as a host
    // would look for the GUI for this plugin: one called dssi-vst.
    // See also RemoteVSTClient::queryPlugins below.

    std::vector<std::string> dssiPath = Paths::getPath
	("DSSI_PATH", "/usr/local/lib/dssi:/usr/lib/dssi", "/.dssi");

    bool found = false;

    for (size_t i = 0; i < dssiPath.size(); ++i) {

	std::string subDir = dssiPath[i] + "/dssi-vst";

	DIR *directory = opendir(subDir.c_str());
	if (!directory) {
	    continue;
	}
	closedir(directory);

	struct stat st;
	std::string fileName = subDir + "/dssi-vst-server.exe";

	if (stat(fileName.c_str(), &st)) {
	    continue;
	}

	if (!(S_ISREG(st.st_mode) || S_ISLNK(st.st_mode)) ||
	    !(st.st_mode & (S_IXUSR | S_IXGRP | S_IXOTH))) {

	    std::cerr << "RemoteVSTClient: file " << fileName
		      << " exists but can't be executed" << std::endl;
	    continue;
	}

	found = true;

	std::cerr << "RemoteVSTClient: executing "
		  << fileName << " " << argStr << std::endl;

        const char* fileNameStr = fileName.c_str();
        if ((child = vfork()) < 0) {
	    cleanup();
	    throw((std::string)"Fork failed");
	} else if (child == 0) { // child process
	    if (execlp(fileNameStr, fileNameStr, argStr, NULL)) {
                // vfork() docs say you shouldn't call a function here,
                // but it seems to work for me.
		perror("Exec failed");
		_exit(1);
	    }
	}

	break;
    }

    if (!found) {
	cleanup();
	throw((std::string)"Failed to find dssi-vst-server.exe");
    } else {
	syncStartup();
    }
}

RemoteVSTClient::~RemoteVSTClient()
{
    for (int i = 0; i < 3; ++i) {
	if (waitpid(-1, NULL, WNOHANG)) break;
	sleep(1);
    }
}

bool
RemoteVSTClient::addFromFd(int fd, PluginRecord &rec)
{
    char buffer[64];

    try {
	tryRead(fd, buffer, 64);
	rec.dllName = buffer;
    } catch (RemotePluginClosedException) {
	// This is acceptable here; it probably just means we're done
	return false;
    }
    
    tryRead(fd, buffer, 64);
    rec.pluginName = buffer;
    
//	    std::cerr << "Plugin " << rec.pluginName << std::endl;
    
    tryRead(fd, buffer, 64);
    rec.vendorName = buffer;
    
    tryRead(fd, &rec.isSynth, sizeof(bool));
    tryRead(fd, &rec.hasGUI, sizeof(bool));
    tryRead(fd, &rec.inputs, sizeof(int));
    tryRead(fd, &rec.outputs, sizeof(int));
    tryRead(fd, &rec.parameters, sizeof(int));
    
//	    std::cerr << rec.parameters << " parameters" << std::endl;
    
    for (int i = 0; i < rec.parameters; ++i) {
        try {
            tryRead(fd, buffer, 64);
        } catch (RemotePluginClosedException) {
            return false; // plugin check failed
        }
	rec.parameterNames.push_back(std::string(buffer));
	float f;
	tryRead(fd, &f, sizeof(float));
	rec.parameterDefaults.push_back(f);
//		std::cerr << "Parameter " << i << ": name " << buffer << ", default " << f << std::endl;
    }
    
    tryRead(fd, &rec.programs, sizeof(int));
    
//	    std::cerr << rec.programs << " programs" << std::endl;
    
    for (int i = 0; i < rec.programs; ++i) {
	tryRead(fd, buffer, 64);
	rec.programNames.push_back(std::string(buffer));
//		std::cerr << "Program " << i << ": name " << buffer << std::endl;
    }

    return true;
}
    
void
RemoteVSTClient::queryPlugins(std::vector<PluginRecord> &plugins)
{
    // First check whether there are any DLLs in the same VST path as
    // the scanner uses.  If not, we know immediately there are no
    // plugins and we don't need to run the (Wine-based) scanner.
    // If there are, but they all have up-to-date cache files, then
    // we can just read those and again not have to run the scanner.
    
    std::vector<std::string> vstPath = Paths::getPath
	("VST_PATH", "/usr/local/lib/vst:/usr/lib/vst", "/vst");

    bool haveDll = false;
    bool haveAllCaches = false;
    bool haveCacheDir = false;
    int version = int(RemotePluginVersion * 1000);

    char *home = getenv("HOME");
    std::string cacheDir = std::string(home) + "/.dssi-vst";
    DIR *test = opendir(cacheDir.c_str());
    if (test) {
	haveCacheDir = true;
	haveAllCaches = true; // until proven otherwise
	closedir(test);
    }

    for (size_t i = 0; i < vstPath.size(); ++i) {
	
	std::string vstDir = vstPath[i];
	DIR *directory = opendir(vstDir.c_str());
	if (!directory) continue;
	struct dirent *entry;

	while ((entry = readdir(directory))) {
	    
	    std::string libname = entry->d_name;

	    if (libname[0] != '.' &&
		libname.length() >= 5 &&
		(libname.substr(libname.length() - 4) == ".dll" ||
		 libname.substr(libname.length() - 4) == ".DLL")) {

		haveDll = true;
		if (!haveCacheDir) break;

		if (haveAllCaches) {

		    std::string cacheFileName = cacheDir + "/" + libname + ".cache";
		    struct stat st;

		    if (stat(cacheFileName.c_str(), &st)) {
			haveAllCaches = false;
		    } else {
			int testfd = open(cacheFileName.c_str(), O_RDONLY);
			int testVersion = 0;
			if (testfd < 0 || 
			    read(testfd, &testVersion, sizeof(int)) != sizeof(int) ||
			    testVersion != version) {
			    haveAllCaches = false;
			}
			if (testfd >= 0) close(testfd);
		    }
		}
	    }
	}

	closedir(directory);
	if (haveDll && !haveAllCaches) break;
    }

    if (!haveDll) return;

    if (haveCacheDir && haveAllCaches) {
	
	std::cerr << "RemoteVSTClient: all cache files are up-to-date, "
		  << "not running scanner" << std::endl;

	for (size_t i = 0; i < vstPath.size(); ++i) {
	
	    std::string vstDir = vstPath[i];
	    DIR *directory = opendir(vstDir.c_str());
	    if (!directory) continue;
	    struct dirent *entry;

	    while ((entry = readdir(directory))) {
	    
		std::string libname = entry->d_name;

		if (libname[0] != '.' &&
		    libname.length() >= 5 &&
		    (libname.substr(libname.length() - 4) == ".dll" ||
		     libname.substr(libname.length() - 4) == ".DLL")) {

		    std::string cacheFileName = cacheDir + "/" + libname + ".cache";
		    int fd = -1;
		    int testVersion = 0;

		    if ((fd = open(cacheFileName.c_str(), O_RDONLY)) < 0) continue;
		    if (read(fd, &testVersion, sizeof(int)) != sizeof(int) ||
			testVersion != version) {
			close(fd);
			continue;
		    }

		    PluginRecord rec;
		    if (addFromFd(fd, rec)) {
			plugins.push_back(rec);
		    }

		    close(fd);
		}
	    }
	}

	return;
    }

    char fifoFile[60];

    sprintf(fifoFile, "/tmp/rplugin_qry_XXXXXX");
    if (mkstemp(fifoFile) < 0) {
	throw((std::string)"Failed to obtain temporary filename");
    }

    unlink(fifoFile);
    if (mkfifo(fifoFile, 0666)) { //!!! what permission is correct here?
	perror(fifoFile);
	throw((std::string)"Failed to create FIFO");
    }

    // We open the fd nonblocking, then start the scanner, then wait
    // to see whether the scanner starts sending anything on it.  If
    // no input is available after a certain time, give up.

    int fd = -1;

    if ((fd = open(fifoFile, O_RDONLY | O_NONBLOCK)) < 0) {
	unlink(fifoFile);
	throw((std::string)"Failed to open FIFO");
    }

    // We want to run the dssi-vst-scanner script, which runs wine
    // dssi-vst-scanner.exe.so.  We expect to find this script in the
    // same subdirectory of a directory in the DSSI_PATH as a host
    // would look for the GUI for this plugin: one called dssi-vst.
    // See also the RemoteVSTClient constructor above.

    std::vector<std::string> dssiPath = Paths::getPath
	("DSSI_PATH", "/usr/local/lib/dssi:/usr/lib/dssi", "/.dssi");

    bool found = false;
    pid_t child;

    for (size_t i = 0; i < dssiPath.size(); ++i) {

	std::string subDir = dssiPath[i] + "/dssi-vst";

	DIR *directory = opendir(subDir.c_str());
	if (!directory) {
	    continue;
	}
	closedir(directory);

	struct stat st;
	std::string fileName = subDir + "/dssi-vst-scanner.exe";

	if (stat(fileName.c_str(), &st)) {
	    continue;
	}

	if (!(S_ISREG(st.st_mode) || S_ISLNK(st.st_mode)) ||
	    !(st.st_mode & (S_IXUSR | S_IXGRP | S_IXOTH))) {

	    std::cerr << "RemoteVSTClient: file " << fileName
		      << " exists but can't be executed" << std::endl;
	    continue;
	}

	found = true;

	std::cerr << "RemoteVSTClient: executing "
		  << fileName << " " << fifoFile << std::endl;

	if ((child = fork()) < 0) {
	    unlink(fifoFile);
	    throw((std::string)"Fork failed");
	} else if (child == 0) { // child process
	    if (execlp(fileName.c_str(), fileName.c_str(), fifoFile, NULL)) {
		perror("Exec failed");
		unlink(fifoFile);
		exit(1);
	    }
	}

	break;
    }

    if (!found) {
	unlink(fifoFile);
	throw((std::string)"Failed to find dssi-vst-scanner.exe");
    }

    struct pollfd pfd;
    pfd.fd = fd;
    pfd.events = POLLIN;
    int sec;
    int timeout = 40;

    for (sec = 0; sec < timeout; ++sec) {

	int rv = poll(&pfd, 1, 1000);

	if (rv < 0) {
	    if (errno == EINTR || errno == EAGAIN) {
		// try again
		sleep(1);
		continue;
	    } else {
		close(fd);
		unlink(fifoFile);
		throw ((std::string)"Plugin scanner startup failed.");
	    }
	} else if (rv > 0) {
	    break;
	}
    }

    if (sec >= timeout) {
	close(fd);
	unlink(fifoFile);
	throw ((std::string)"Plugin scanner timed out on startup.");
    }

    try {
	int version = 0;

	tryRead(fd, &version, sizeof(int));
	if (version != int(RemotePluginVersion * 1000)) {
	    throw ((std::string)"Plugin scanner version mismatch");
	}

	while (1) {
	    PluginRecord rec;
	    if (!addFromFd(fd, rec)) break; // done
	    plugins.push_back(rec);
	}

    } catch (std::string s) {
	std::cerr << s << std::endl;
    } catch (RemotePluginClosedException) {
	std::cerr << "Plugin scanner exited unexpectedly." << std::endl;
    }

    close(fd);
    unlink(fifoFile);

    for (int i = 0; i < 3; ++i) {
	if (waitpid(-1, NULL, WNOHANG)) break;
	sleep(1);
    }
}