107 lines
2.8 KiB
C++
107 lines
2.8 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
|
|
*
|
|
* 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.
|
|
*
|
|
* The Original Code is mozilla.org code.
|
|
*
|
|
* The Initial Developer of the Original Code is Netscape
|
|
* Communications Corporation. Portions created by Netscape are
|
|
* Copyright (C) 1998 Netscape Communications Corporation. All
|
|
* Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
*/
|
|
|
|
#include "dllcom.h"
|
|
|
|
CProcessEntry *CProcess::m_pProcessList = NULL;
|
|
|
|
CProcessEntry::CProcessEntry(CComDll *pDll)
|
|
{
|
|
DLL_ASSERT(pDll);
|
|
|
|
m_dwProcessID = CProcess::GetProcessID();
|
|
m_pDllInstance = pDll;
|
|
|
|
m_pNext = CProcess::m_pProcessList;
|
|
CProcess::m_pProcessList = this;
|
|
}
|
|
|
|
CProcessEntry::~CProcessEntry()
|
|
{
|
|
// find ourselves in the list and remove us.
|
|
CProcessEntry **ppChange = &CProcess::m_pProcessList;
|
|
|
|
while(*ppChange != this) {
|
|
ppChange = &((*ppChange)->m_pNext);
|
|
}
|
|
|
|
*ppChange = m_pNext;
|
|
|
|
m_dwProcessID = 0;
|
|
m_pDllInstance = NULL;
|
|
m_pNext = NULL;
|
|
}
|
|
|
|
DWORD CProcess::GetProcessID()
|
|
{
|
|
DWORD dwRetval = 0;
|
|
|
|
// This is only needed to multiplex instance data on win16.
|
|
#ifdef DLL_WIN16
|
|
dwRetval = CoGetCurrentProcess();
|
|
#else
|
|
// Just use the HINSTANCE unless we need to do this on a per
|
|
// thread bases (then as above like WIN16).
|
|
dwRetval = (DWORD)CComDll::GetInstanceHandle();
|
|
#endif
|
|
|
|
return(dwRetval);
|
|
}
|
|
|
|
CComDll *CProcess::GetProcessDll()
|
|
{
|
|
CComDll *pRetval = NULL;
|
|
|
|
// Before we can find the Dll instance data, we need to
|
|
// get the ID of the current process calling us.
|
|
DWORD dwCurrent = GetProcessID();
|
|
|
|
// See if the object already exists.
|
|
CProcessEntry *pTraverse = m_pProcessList;
|
|
while(pTraverse && pTraverse->m_dwProcessID != dwCurrent) {
|
|
pTraverse = pTraverse->m_pNext;
|
|
}
|
|
|
|
if(pTraverse) {
|
|
// Have it already.
|
|
// Return it.
|
|
pRetval = pTraverse->m_pDllInstance;
|
|
|
|
// We must be sure to properly reference count the returned
|
|
// object.
|
|
pRetval->AddRef();
|
|
}
|
|
else {
|
|
// Do not have it.
|
|
// Ask the end consumer to create it.
|
|
pRetval = DLL_ConsumerCreateInstance();
|
|
}
|
|
|
|
return(pRetval);
|
|
}
|
|
|
|
BOOL CProcess::CanUnloadNow()
|
|
{
|
|
// We can unload if we have no more process entries;
|
|
// all objects are freed off.
|
|
return(m_pProcessList == NULL);
|
|
} |