PluginHost.cpp

Go to the documentation of this file.

/**********************************************************************
 
  Audacity: A Digital Audio Editor
 
  @file PluginHost.cpp
 
  @author Vitaly Sverchinsky
 
  Part of lib-module-manager library
 
**********************************************************************/
 
#include "PluginHost.h"
 
#include <wx/log.h>
#include <wx/module.h>
#include <wx/process.h>
 
#include "BasicUI.h"
#include "CommandLineArgs.h"
#include "PathList.h"
#include "FileNames.h"
#include "ModuleManager.h"
#include "IPCClient.h"
#include "PlatformCompatibility.h"
#include "PluginManager.h"
 
namespace
{
   //Attempts to instantiate plugin module and put plugin descriptors into result
   void Discover(detail::PluginValidationResult& result, const wxString& providerId, const wxString& pluginPath)
   {
      try
      {
         if(auto provider = ModuleManager::Get().CreateProviderInstance(providerId, wxEmptyString))
         {
            TranslatableString errorMessage{};
            auto validator = provider->MakeValidator();
            auto numPlugins = provider->DiscoverPluginsAtPath(
               pluginPath, errorMessage, [&](PluginProvider *provider, ComponentInterface *ident)
            {
               //Workaround: use DefaultRegistrationCallback to create all descriptors for us
               //and then put a copy into result
               auto id = PluginManager::DefaultRegistrationCallback(provider, ident);
               if(const auto ptr = PluginManager::Get().GetPlugin(id))
               {
                  auto desc = *ptr;
                  try
                  {
                     if(validator)
                        validator->Validate(*ident);
                  }
                  catch(...)
                  {
                     desc.SetEnabled(false);
                     desc.SetValid(false);
                  }
                  result.Add(std::move(desc));
               }
               return id;
            });
            if(!errorMessage.empty())
               result.SetError(errorMessage.Debug());
            else if(numPlugins == 0)
               result.SetError("no plugins found");
         }
         else
            result.SetError("provider not found");
      }
      catch(...)
      {
         result.SetError("unknown error");
      }
   }
}
 
PluginHost::PluginHost(int connectPort)
{
   FileNames::InitializePathList();
 
   wxFileName configFileName{ FileNames::Configuration() };
   auto pConfig = std::make_unique<FileConfig>(
      AppName, wxEmptyString, configFileName.GetFullPath(),
      wxEmptyString, wxCONFIG_USE_LOCAL_FILE);
   pConfig->Init();
   InitPreferences(std::move(pConfig));
 
   auto& moduleManager = ModuleManager::Get();
   moduleManager.Initialize();
   moduleManager.DiscoverProviders();
 
   mClient = std::make_unique<IPCClient>(connectPort, *this);
}
 
void PluginHost::OnConnect(IPCChannel& channel) noexcept
{
   std::lock_guard lck(mSync);
   mChannel = &channel;
}
 
void PluginHost::OnDisconnect() noexcept
{
   Stop();
}
 
void PluginHost::OnConnectionError() noexcept
{
   Stop();
}
 
void PluginHost::OnDataAvailable(const void* data, size_t size) noexcept
{
   try
   {
      mInputMessageReader.ConsumeBytes(data, size);
      if(mInputMessageReader.CanPop())
      {
         {
            std::lock_guard lck(mSync);
            assert(!mRequest);
            mRequest = mInputMessageReader.Pop();
         }
         mRequestCondition.notify_one();
      }
   }
   catch(...)
   {
      Stop();
   }
}
 
bool PluginHost::Serve()
{
   std::unique_lock lck(mSync);
   mRequestCondition.wait(lck, [this]{ return !mRunning || mRequest.has_value(); });
 
   if(!mRunning)
      return false;
 
   if(mRequest)
   {
      if(mChannel)
         detail::PutMessage(*mChannel, wxEmptyString);
 
      std::optional<wxString> request;
      mRequest.swap(request);
 
      lck.unlock();
      
      wxString providerId;
      wxString pluginPath;
      detail::PluginValidationResult result;
      if(detail::ParseRequestString(*request, providerId, pluginPath))
         Discover(result, providerId, pluginPath);
      else
         result.SetError("malformed request string");
 
      XMLStringWriter xmlWriter;
      result.WriteXML(xmlWriter);
 
      lck.lock();
      if(mChannel)
         detail::PutMessage(*mChannel, xmlWriter);
   }
 
   return true;
}
 
 
void PluginHost::Stop() noexcept
{
   try
   {
      {
         std::lock_guard lck(mSync);//may throw
         mRunning = false;
         mChannel = nullptr;
      }
   }
   catch(...)
   {
      //If something went wrong with mutex locking we'll try to
      //awake main thread if it's blocked on condition variable.
      //Attempt to relock the mutex there should throw as well(?)...
      //which, in turn, will result in std::terminate being called
   }
   mRequestCondition.notify_one();
}
 
bool PluginHost::Start(int connectPort)
{
   const auto cmd = wxString::Format("\"%s\" %s %d",
      PlatformCompatibility::GetExecutablePath(),
      PluginHost::HostArgument,
      connectPort);
 
   auto process = std::make_unique<wxProcess>();
   process->Detach();
   if(wxExecute(cmd, wxEXEC_ASYNC, process.get()) != 0)
   {
      //process will delete itself upon termination
      process.release();
      return true;
   }
   return false;
}
 
bool PluginHost::IsHostProcess()
{
   return CommandLineArgs::argc >= 3 &&
      wxStrcmp(CommandLineArgs::argv[1], HostArgument) == 0;
}
 
class PluginHostModule final :
   public wxModule
{
public:
   DECLARE_DYNAMIC_CLASS(PluginHostModule)
 
   bool OnInit() override
   {
      if(PluginHost::IsHostProcess())
      {
         long connectPort;
         if(!wxString{ CommandLineArgs::argv[2] }.ToLong(&connectPort))
            return false;
 
         //log messages will appear in a separate window
         //redirect to log file later
         wxLog::EnableLogging(false);
 
         //Handle requests...
         PluginHost host(connectPort);
         while(host.Serve()) { }
         //...and terminate app
         return false;
      }
      //do nothing if current process isn't a host process
      return true;
   }
   
   void OnExit() override
   {
 
   }
};
IMPLEMENT_DYNAMIC_CLASS(PluginHostModule, wxModule);