gecko-dev/xpcom/components/xcDll.cpp

450 строки
12 KiB
C++
Исходник Обычный вид История

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/* ***** BEGIN LICENSE BLOCK *****
* Version: NPL 1.1/GPL 2.0/LGPL 2.1
1999-02-02 01:40:48 +03:00
*
* The contents of this file are subject to the Netscape Public License
* Version 1.1 (the "License"); you may not use this file except in
* compliance with the License. You may obtain a copy of the License at
* http://www.mozilla.org/NPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
1999-02-02 01:40:48 +03:00
*
* The Original Code is mozilla.org code.
*
* The Initial Developer of the Original Code is
* Netscape Communications Corporation.
* Portions created by the Initial Developer are Copyright (C) 1998
* the Initial Developer. All Rights Reserved.
*
* Contributor(s):
*
* Alternatively, the contents of this file may be used under the terms of
* either the GNU General Public License Version 2 or later (the "GPL"), or
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
* in which case the provisions of the GPL or the LGPL are applicable instead
* of those above. If you wish to allow use of your version of this file only
* under the terms of either the GPL or the LGPL, and not to allow others to
* use your version of this file under the terms of the NPL, indicate your
* decision by deleting the provisions above and replace them with the notice
* and other provisions required by the GPL or the LGPL. If you do not delete
* the provisions above, a recipient may use your version of this file under
* the terms of any one of the NPL, the GPL or the LGPL.
*
* ***** END LICENSE BLOCK ***** */
1999-02-02 01:40:48 +03:00
/* nsDll
1999-02-02 01:40:48 +03:00
*
* Abstraction of a Dll. Stores modifiedTime and size for easy detection of
* change in dll.
*
* dp Suresh <dp@netscape.com>
*/
#include "xcDll.h"
#include "nsDebug.h"
1999-06-22 18:02:58 +04:00
#include "nsIComponentManager.h"
#include "nsIComponentLoaderManager.h"
#include "nsIModule.h"
2000-01-25 00:28:28 +03:00
#include "nsILocalFile.h"
#include "nsCOMPtr.h"
2000-01-25 00:28:28 +03:00
#include "nsCRT.h"
#include "nsString.h"
#include "nsITimelineService.h"
#ifdef DEBUG
#if defined(VMS)
#include <lib$routines.h>
#include <ssdef.h>
#elif defined(XP_MACOSX)
#include <signal.h>
#endif
#endif /* defined(DEBUG) */
#include "nsNativeComponentLoader.h"
1999-06-22 18:02:58 +04:00
nsDll::nsDll(const char *codeDllName, int type)
2000-01-25 00:28:28 +03:00
: m_dllName(NULL),
m_instance(NULL), m_status(DLL_OK), m_moduleObject(NULL),
1999-09-14 02:34:42 +04:00
m_persistentDescriptor(NULL), m_nativePath(NULL),
m_markForUnload(PR_FALSE), m_registryLocation(0)
1999-02-02 01:40:48 +03:00
{
1999-06-22 18:02:58 +04:00
if (!codeDllName || !*codeDllName)
{
m_status = DLL_INVALID_PARAM;
return;
}
1999-07-31 04:28:51 +04:00
m_dllName = nsCRT::strdup(codeDllName);
1999-06-22 18:02:58 +04:00
if (!m_dllName)
{
m_status = DLL_NO_MEM;
return;
}
}
1999-02-03 21:49:04 +03:00
2000-01-25 00:28:28 +03:00
nsDll::nsDll(nsIFile *dllSpec, const char *registryLocation)
: m_dllName(NULL),
m_instance(NULL), m_status(DLL_OK), m_moduleObject(NULL),
m_persistentDescriptor(NULL), m_nativePath(NULL), m_markForUnload(PR_FALSE)
1999-06-22 18:02:58 +04:00
{
m_dllSpec = dllSpec;
2000-01-25 00:28:28 +03:00
m_registryLocation = nsCRT::strdup(registryLocation);
1999-06-22 18:02:58 +04:00
Init(dllSpec);
1999-02-02 01:40:48 +03:00
}
1999-06-22 18:02:58 +04:00
nsDll::nsDll(const char *libPersistentDescriptor)
2000-01-25 00:28:28 +03:00
: m_dllName(NULL),
m_instance(NULL), m_status(DLL_OK), m_moduleObject(NULL),
1999-09-14 02:34:42 +04:00
m_persistentDescriptor(NULL), m_nativePath(NULL),
m_markForUnload(PR_FALSE), m_registryLocation(0)
1999-06-22 18:02:58 +04:00
{
Init(libPersistentDescriptor);
}
1999-06-22 18:02:58 +04:00
void
2000-01-25 00:28:28 +03:00
nsDll::Init(nsIFile *dllSpec)
1999-06-22 18:02:58 +04:00
{
// Load will fail anyway. So dont bother to stat the file
1999-06-22 18:02:58 +04:00
m_dllSpec = do_QueryInterface(dllSpec);
1999-06-22 18:02:58 +04:00
m_status = DLL_OK;
}
void
nsDll::Init(const char *libPersistentDescriptor)
{
nsresult rv;
1999-06-22 18:02:58 +04:00
if (libPersistentDescriptor == NULL)
{
m_status = DLL_INVALID_PARAM;
return;
}
1999-06-22 18:02:58 +04:00
// Create a FileSpec from the persistentDescriptor
2000-01-25 00:28:28 +03:00
nsCOMPtr<nsILocalFile> dllSpec;
nsCID clsid;
nsComponentManager::ContractIDToClassID(NS_LOCAL_FILE_CONTRACTID, &clsid);
2000-01-25 00:28:28 +03:00
rv = nsComponentManager::CreateInstance(clsid, nsnull,
NS_GET_IID(nsILocalFile),
(void**)getter_AddRefs(dllSpec));
1999-06-22 18:02:58 +04:00
if (NS_FAILED(rv))
{
m_status = DLL_INVALID_PARAM;
return;
}
rv = dllSpec->InitWithNativePath(nsDependentCString(libPersistentDescriptor));
1999-06-22 18:02:58 +04:00
if (NS_FAILED(rv))
{
m_status = DLL_INVALID_PARAM;
return;
}
}
1999-06-22 18:02:58 +04:00
nsDll::~nsDll(void)
1999-02-02 01:40:48 +03:00
{
#if DEBUG_dougt
// The dll gets deleted when the dllStore is destroyed. This happens on
// app shutdown. At that point, unloading dlls can cause crashes if we have
// - dll dependencies
// - callbacks
// - static dtors
// Hence turn it back on after all the above have been removed.
1999-07-31 04:28:51 +04:00
Unload();
#endif
1999-06-22 18:02:58 +04:00
if (m_dllName)
1999-07-31 04:28:51 +04:00
nsCRT::free(m_dllName);
if (m_registryLocation)
nsCRT::free(m_registryLocation);
1999-02-02 01:40:48 +03:00
}
2000-01-25 00:28:28 +03:00
1999-06-22 18:02:58 +04:00
const char *
2000-01-25 00:28:28 +03:00
nsDll::GetDisplayPath()
1999-06-22 18:02:58 +04:00
{
if (m_dllName)
return m_dllName;
if (!m_nativePath.IsEmpty())
return m_nativePath.get();
m_dllSpec->GetNativePath(m_nativePath);
return m_nativePath.get();
1999-06-22 18:02:58 +04:00
}
const char *
nsDll::GetPersistentDescriptorString()
{
if (m_dllName)
return m_dllName;
if (!m_persistentDescriptor.IsEmpty())
return m_persistentDescriptor.get();
m_dllSpec->GetNativePath(m_persistentDescriptor);
return m_persistentDescriptor.get();
1999-06-22 18:02:58 +04:00
}
PRBool
nsDll::HasChanged()
{
if (m_dllName)
return PR_FALSE;
extern nsresult NS_GetComponentLoaderManager(nsIComponentLoaderManager* *result);
nsCOMPtr<nsIComponentLoaderManager> manager;
NS_GetComponentLoaderManager(getter_AddRefs(manager));
if (!manager)
return PR_TRUE;
1999-06-22 18:02:58 +04:00
// If mod date has changed, then dll has changed
2000-01-25 00:28:28 +03:00
PRInt64 currentDate;
nsresult rv = m_dllSpec->GetLastModifiedTime(&currentDate);
PRBool changed = PR_TRUE;
manager->HasFileChanged(m_dllSpec, nsnull, currentDate, &changed);
return changed;
1999-06-22 18:02:58 +04:00
}
PRBool nsDll::Load(void)
1999-02-02 01:40:48 +03:00
{
if (m_status != DLL_OK)
{
return (PR_FALSE);
}
if (m_instance != NULL)
{
// Already loaded
return (PR_TRUE);
}
1999-06-22 18:02:58 +04:00
2000-01-25 00:28:28 +03:00
if (m_dllSpec)
1999-06-22 18:02:58 +04:00
{
#ifdef NS_BUILD_REFCNT_LOGGING
nsTraceRefcnt::SetActivityIsLegal(PR_FALSE);
#endif
nsCOMPtr<nsILocalFile> lf(do_QueryInterface(m_dllSpec));
NS_ASSERTION(lf, "nsIFile here must implement a nsILocalFile");
lf->Load(&m_instance);
2000-01-25 00:28:28 +03:00
#ifdef NS_BUILD_REFCNT_LOGGING
nsTraceRefcnt::SetActivityIsLegal(PR_TRUE);
2000-01-25 00:28:28 +03:00
if (m_instance) {
// Inform refcnt tracer of new library so that calls through the
// new library can be traced.
nsCAutoString displayPath;
m_dllSpec->GetNativePath(displayPath);
nsTraceRefcnt::LoadLibrarySymbols(displayPath.get(), m_instance);
2000-01-25 00:28:28 +03:00
}
#endif
1999-06-22 18:02:58 +04:00
}
2000-01-25 00:28:28 +03:00
else if (m_dllName)
1999-06-22 18:02:58 +04:00
{
2000-01-25 00:28:28 +03:00
// if there is not an nsIFile, but there is a dll name, just try to load that..
#ifdef NS_BUILD_REFCNT_LOGGING
nsTraceRefcnt::SetActivityIsLegal(PR_FALSE);
#endif
NS_TIMELINE_START_TIMER("PR_LoadLibrary");
2000-01-25 00:28:28 +03:00
m_instance = PR_LoadLibrary(m_dllName);
NS_TIMELINE_STOP_TIMER("PR_LoadLibrary");
NS_TIMELINE_MARK_TIMER("PR_LoadLibrary");
2000-01-25 00:28:28 +03:00
#ifdef NS_BUILD_REFCNT_LOGGING
nsTraceRefcnt::SetActivityIsLegal(PR_TRUE);
2000-01-25 00:28:28 +03:00
if (m_instance) {
// Inform refcnt tracer of new library so that calls through the
// new library can be traced.
nsTraceRefcnt::LoadLibrarySymbols(m_dllName, m_instance);
}
#endif
}
2000-01-25 00:28:28 +03:00
#if defined(DEBUG) && defined(XP_UNIX)
// Debugging help for components. Component dlls need to have their
// symbols loaded before we can put a breakpoint in the debugger.
// This will help figureing out the point when the dll was loaded.
BreakAfterLoad(GetDisplayPath());
#endif
1999-06-22 18:02:58 +04:00
return ((m_instance == NULL) ? PR_FALSE : PR_TRUE);
1999-02-02 01:40:48 +03:00
}
PRBool nsDll::Unload(void)
1999-02-02 01:40:48 +03:00
{
if (m_status != DLL_OK || m_instance == NULL)
return (PR_FALSE);
// Shutdown the dll
Shutdown();
#ifdef NS_BUILD_REFCNT_LOGGING
nsTraceRefcnt::SetActivityIsLegal(PR_FALSE);
#endif
1999-02-02 01:40:48 +03:00
PRStatus ret = PR_UnloadLibrary(m_instance);
#ifdef NS_BUILD_REFCNT_LOGGING
nsTraceRefcnt::SetActivityIsLegal(PR_TRUE);
#endif
1999-02-02 01:40:48 +03:00
if (ret == PR_SUCCESS)
{
m_instance = NULL;
return (PR_TRUE);
}
else
return (PR_FALSE);
}
void * nsDll::FindSymbol(const char *symbol)
1999-02-02 01:40:48 +03:00
{
if (symbol == NULL)
return (NULL);
// If not already loaded, load it now.
if (Load() != PR_TRUE)
return (NULL);
return(PR_FindSymbol(m_instance, symbol));
1999-02-02 01:40:48 +03:00
}
// Component dll specific functions
2000-01-25 00:28:28 +03:00
nsresult nsDll::GetDllSpec(nsIFile **fsobj)
{
NS_ASSERTION(m_dllSpec, "m_dllSpec NULL");
NS_ASSERTION(fsobj, "xcDll::GetModule : Null argument" );
*fsobj = m_dllSpec;
2000-01-25 00:28:28 +03:00
NS_ADDREF(*fsobj);
return NS_OK;
}
nsresult nsDll::GetModule(nsISupports *servMgr, nsIModule **cobj)
{
nsCOMPtr<nsIComponentManager> compMgr;
nsresult rv = NS_GetComponentManager(getter_AddRefs(compMgr));
NS_ASSERTION(compMgr, "Global Component Manager is null" );
if (NS_FAILED(rv)) return rv;
NS_ASSERTION(cobj, "xcDll::GetModule : Null argument" );
if (m_moduleObject)
{
NS_ADDREF(m_moduleObject);
*cobj = m_moduleObject;
return NS_OK;
}
// If not already loaded, load it now.
if (Load() != PR_TRUE) return NS_ERROR_FAILURE;
2000-01-25 00:28:28 +03:00
// We need a nsIFile for location. If we dont
// have one, create one.
2000-01-25 00:28:28 +03:00
if (!m_dllSpec && m_dllName)
{
// Create m_dllSpec from m_dllName
}
nsGetModuleProc proc =
(nsGetModuleProc) FindSymbol(NS_GET_MODULE_SYMBOL);
if (proc == NULL)
return NS_ERROR_FACTORY_NOT_LOADED;
rv = (*proc) (compMgr, m_dllSpec, &m_moduleObject);
if (NS_SUCCEEDED(rv))
{
NS_ADDREF(m_moduleObject);
*cobj = m_moduleObject;
}
return rv;
}
#if defined(DEBUG) && !defined(XP_BEOS)
#define SHOULD_IMPLEMENT_BREAKAFTERLOAD
#endif
// These are used by BreakAfterLoad, below.
#ifdef SHOULD_IMPLEMENT_BREAKAFTERLOAD
static nsCString *sBreakList[16];
static int sBreakListCount = 0;
#endif
nsresult nsDll::Shutdown(void)
{
// Release the module object if we got one
nsrefcnt refcnt;
if (m_moduleObject)
{
NS_RELEASE2(m_moduleObject, refcnt);
NS_ASSERTION(refcnt == 0, "Dll moduleObject refcount non zero");
}
#ifdef SHOULD_IMPLEMENT_BREAKAFTERLOAD
for (int i = 0; i < sBreakListCount; i++)
{
delete sBreakList[i];
sBreakList[i] = nsnull;
}
sBreakListCount = 0;
#endif
return NS_OK;
}
void nsDll::BreakAfterLoad(const char *nsprPath)
{
#ifdef SHOULD_IMPLEMENT_BREAKAFTERLOAD
static PRBool firstTime = PR_TRUE;
// return if invalid input
if (!nsprPath || !*nsprPath) return;
// return if nothing to break on
if (!firstTime && sBreakListCount == 0) return;
if (firstTime)
{
firstTime = PR_FALSE;
// Form the list of dlls to break on load
nsCAutoString envList(getenv("XPCOM_BREAK_ON_LOAD"));
if (envList.IsEmpty()) return;
PRInt32 ofset = 0;
PRInt32 start = 0;
do
{
ofset = envList.FindChar(':', start);
sBreakList[sBreakListCount] = new nsCString();
envList.Mid(*(sBreakList[sBreakListCount]), start, ofset);
sBreakListCount++;
start = ofset + 1;
}
while (ofset != -1 && 16 > sBreakListCount); // avoiding vc6.0 compiler issue. count < thinks it is starting a template
}
// Find the dllname part of the string
nsCString currentPath(nsprPath);
PRInt32 lastSep = currentPath.RFindCharInSet(":\\/");
for (int i=0; i<sBreakListCount; i++)
if (currentPath.Find(*(sBreakList[i]), PR_TRUE, lastSep) > 0)
{
// Loading a dll that we want to break on
// Put your breakpoint here
fprintf(stderr, "...Loading module %s\n", nsprPath);
// Break in the debugger here.
#if defined(linux) && defined(__i386)
asm("int $3");
#elif defined(VMS)
lib$signal(SS$_DEBUG);
#elif defined(XP_MACOSX)
raise(SIGTRAP);
#endif
}
#endif /* SHOULD_IMPLEMENT_BREAKAFTERLOAD */
return;
}