/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "nsNPAPIPlugin.h"
#include "nsNPAPIPluginInstance.h"
#include "nsPluginsDir.h"
#include "nsPluginsDirUtils.h"
#include "prenv.h"
#include "prerror.h"
#include "prio.h"
#include <sys/stat.h>
#include "nsString.h"
#include "nsIFile.h"

#define LOCAL_PLUGIN_DLL_SUFFIX ".so"
#if defined(__hpux)
#  define DEFAULT_X11_PATH "/usr/lib/X11R6/"
#  undef LOCAL_PLUGIN_DLL_SUFFIX
#  define LOCAL_PLUGIN_DLL_SUFFIX ".sl"
#  define LOCAL_PLUGIN_DLL_ALT_SUFFIX ".so"
#elif defined(_AIX)
#  define DEFAULT_X11_PATH "/usr/lib"
#  define LOCAL_PLUGIN_DLL_ALT_SUFFIX ".a"
#elif defined(SOLARIS)
#  define DEFAULT_X11_PATH "/usr/openwin/lib/"
#elif defined(LINUX)
#  define DEFAULT_X11_PATH "/usr/X11R6/lib/"
#elif defined(__APPLE__)
#  define DEFAULT_X11_PATH "/usr/X11R6/lib"
#  undef LOCAL_PLUGIN_DLL_SUFFIX
#  define LOCAL_PLUGIN_DLL_SUFFIX ".dylib"
#  define LOCAL_PLUGIN_DLL_ALT_SUFFIX ".so"
#else
#  define DEFAULT_X11_PATH ""
#endif

/* nsPluginsDir implementation */

bool nsPluginsDir::IsPluginFile(nsIFile* file) {
  nsAutoCString filename;
  if (NS_FAILED(file->GetNativeLeafName(filename))) return false;

  constexpr auto dllSuffix = nsLiteralCString{LOCAL_PLUGIN_DLL_SUFFIX};
  if (filename.Length() > dllSuffix.Length() &&
      StringEndsWith(filename, dllSuffix))
    return true;

#ifdef LOCAL_PLUGIN_DLL_ALT_SUFFIX
  constexpr auto dllAltSuffix = nsLiteralCString{LOCAL_PLUGIN_DLL_ALT_SUFFIX};
  if (filename.Length() > dllAltSuffix.Length() &&
      StringEndsWith(filename, dllAltSuffix))
    return true;
#endif
  return false;
}

/* nsPluginFile implementation */

nsPluginFile::nsPluginFile(nsIFile* file) : mPlugin(file) {}

nsPluginFile::~nsPluginFile() = default;

nsresult nsPluginFile::LoadPlugin(PRLibrary** outLibrary) {
  PRLibSpec libSpec;
  libSpec.type = PR_LibSpec_Pathname;
  bool exists = false;
  mPlugin->Exists(&exists);
  if (!exists) return NS_ERROR_FILE_NOT_FOUND;

  nsresult rv;
  nsAutoCString path;
  rv = mPlugin->GetNativePath(path);
  if (NS_FAILED(rv)) return rv;

  libSpec.value.pathname = path.get();

  *outLibrary = PR_LoadLibraryWithFlags(libSpec, 0);
  pLibrary = *outLibrary;

#ifdef DEBUG
  printf("LoadPlugin() %s returned %lx\n", libSpec.value.pathname,
         (unsigned long)pLibrary);
#endif

  if (!pLibrary) {
    return NS_ERROR_FAILURE;
  }

  return NS_OK;
}

nsresult nsPluginFile::GetPluginInfo(nsPluginInfo& info,
                                     PRLibrary** outLibrary) {
  *outLibrary = nullptr;

  info.fVersion = nullptr;

  // Sadly we have to load the library for this to work.
  nsresult rv = LoadPlugin(outLibrary);
  if (NS_FAILED(rv)) return rv;

  const char* (*npGetPluginVersion)() =
      (const char* (*)())PR_FindFunctionSymbol(pLibrary, "NP_GetPluginVersion");
  if (npGetPluginVersion) {
    info.fVersion = PL_strdup(npGetPluginVersion());
  }

  const char* (*npGetMIMEDescription)() =
      (const char* (*)())PR_FindFunctionSymbol(pLibrary,
                                               "NP_GetMIMEDescription");
  if (!npGetMIMEDescription) {
    return NS_ERROR_FAILURE;
  }

  const char* mimedescr = npGetMIMEDescription();
  if (!mimedescr) {
    return NS_ERROR_FAILURE;
  }

  rv = ParsePluginMimeDescription(mimedescr, info);
  if (NS_FAILED(rv)) {
    return rv;
  }

  nsAutoCString path;
  if (NS_FAILED(rv = mPlugin->GetNativePath(path))) return rv;
  info.fFullPath = PL_strdup(path.get());

  nsAutoCString fileName;
  if (NS_FAILED(rv = mPlugin->GetNativeLeafName(fileName))) return rv;
  info.fFileName = PL_strdup(fileName.get());

  NP_GetValueFunc npGetValue =
      (NP_GetValueFunc)PR_FindFunctionSymbol(pLibrary, "NP_GetValue");
  if (!npGetValue) {
    return NS_ERROR_FAILURE;
  }

  const char* name = nullptr;
  npGetValue(nullptr, NPPVpluginNameString, &name);
  if (name) {
    info.fName = PL_strdup(name);
  } else {
    info.fName = PL_strdup(fileName.get());
  }

  const char* description = nullptr;
  npGetValue(nullptr, NPPVpluginDescriptionString, &description);
  if (description) {
    info.fDescription = PL_strdup(description);
  } else {
    info.fDescription = PL_strdup("");
  }

  return NS_OK;
}

nsresult nsPluginFile::FreePluginInfo(nsPluginInfo& info) {
  if (info.fName != nullptr) PL_strfree(info.fName);

  if (info.fDescription != nullptr) PL_strfree(info.fDescription);

  for (uint32_t i = 0; i < info.fVariantCount; i++) {
    if (info.fMimeTypeArray[i] != nullptr) PL_strfree(info.fMimeTypeArray[i]);

    if (info.fMimeDescriptionArray[i] != nullptr)
      PL_strfree(info.fMimeDescriptionArray[i]);

    if (info.fExtensionArray[i] != nullptr) PL_strfree(info.fExtensionArray[i]);
  }

  free(info.fMimeTypeArray);
  info.fMimeTypeArray = nullptr;
  free(info.fMimeDescriptionArray);
  info.fMimeDescriptionArray = nullptr;
  free(info.fExtensionArray);
  info.fExtensionArray = nullptr;

  if (info.fFullPath != nullptr) PL_strfree(info.fFullPath);

  if (info.fFileName != nullptr) PL_strfree(info.fFileName);

  if (info.fVersion != nullptr) PL_strfree(info.fVersion);

  return NS_OK;
}