//-------------------------------------------------------------------------------------------------------

// Copyright (C) Microsoft. All rights reserved.

// Licensed under the MIT license. See LICENSE.txt file in the project root for full license information.

//-------------------------------------------------------------------------------------------------------

#include "Runtime.h"

#include "core\AtomLockGuids.h"

#include "core\ConfigParser.h"

#include "Base\ThreadContextTLSEntry.h"

#include "Base\ThreadBoundThreadContextManager.h"

#ifdef DYNAMIC_PROFILE_STORAGE

#include "Language\DynamicProfileStorage.h"

#endif

#include "jsrtcontext.h"

#include "TestHooks.h"

extern HANDLE g_hInstance;

static ATOM lockedDll = 0;

static BOOL AttachProcess(HANDLE hmod)

{

if (!ThreadContextTLSEntry::InitializeProcess() || !JsrtContext::Initialize())

{

return FALSE;

}

g_hInstance = hmod;

AutoSystemInfo::SaveModuleFileName(hmod);

#if defined(_M_IX86)

// Enable SSE2 math functions in CRT if SSE2 is available

#pragma prefast(suppress:6031, "We don't require SSE2, but will use it if available")

_set_SSE2_enable(TRUE);

#endif

#ifdef ENABLE_TEST_HOOKS

if (FAILED(OnChakraCoreLoaded()))

{

return FALSE;

}

#endif

{

CmdLineArgsParser parser;

ConfigParser::ParseOnModuleLoad(parser, hmod);

}

#ifdef ENABLE_JS_ETW

EtwTrace::Register();

#endif

ValueType::Initialize();

ThreadContext::GlobalInitialize();

wchar_t *engine = szChakraCoreLock;

if (::FindAtom(szJScript9Lock) != 0)

{

AssertMsg(FALSE, "Expecting to load chakracore.dll but process already loaded jscript9.dll");

Binary_Inconsistency_fatal_error();

}

if (::FindAtom(szChakraLock) != 0)

{

AssertMsg(FALSE, "Expecting to load chakracore.dll but process already loaded chakra.dll");

Binary_Inconsistency_fatal_error();

}

lockedDll = ::AddAtom(engine);

AssertMsg(lockedDll, "Failed to lock chakracore.dll");

#ifdef ENABLE_BASIC_TELEMETRY

g_TraceLoggingClient = NoCheckHeapNewStruct(TraceLoggingClient);

#endif

#ifdef DYNAMIC_PROFILE_STORAGE

return DynamicProfileStorage::Initialize();

#else

return TRUE;

#endif

}

static void DetachProcess()

{

ThreadBoundThreadContextManager::DestroyAllContextsAndEntries();

// In JScript, we never unload except for when the app shuts down

// because DllCanUnloadNow always returns S_FALSE. As a result

// it's okay that we never try to cleanup. Attempting to cleanup on

// shutdown is bad because we shouldn't free objects built into

// other dlls.

JsrtRuntime::Uninitialize();

JsrtContext::Uninitialize();

// thread-bound entrypoint should be able to get cleanup correctly, however tlsentry

// for current thread might be left behind if this thread was initialized.

ThreadContextTLSEntry::CleanupThread();

ThreadContextTLSEntry::CleanupProcess();

#if PROFILE_DICTIONARY

DictionaryStats::OutputStats();

#endif

g_hInstance = NULL;

}

/****************************** Public Functions *****************************/

#if _WIN32 || _WIN64

EXTERN_C BOOL WINAPI DllMain(HINSTANCE hmod, DWORD dwReason, PVOID pvReserved)

{

switch (dwReason)

{

case DLL_PROCESS_ATTACH:

{

return AttachProcess(hmod);

}

case DLL_THREAD_ATTACH:

ThreadContextTLSEntry::InitializeThread();

#ifdef HEAP_TRACK_ALLOC

HeapAllocator::InitializeThread();

#endif

return TRUE;

case DLL_THREAD_DETACH:

// If we are not doing DllCanUnloadNow, so we should clean up. Otherwise, DllCanUnloadNow is already running,

// so the ThreadContext global lock is already taken. If we try to clean up, we will block on the ThreadContext

// global lock while holding the loader lock, which DllCanUnloadNow may block on waiting for thread termination

// which requires the loader lock. DllCanUnloadNow will clean up for us anyway, so we can just skip the whole thing.

ThreadBoundThreadContextManager::DestroyContextAndEntryForCurrentThread();

return TRUE;

case DLL_PROCESS_DETACH:

lockedDll = ::DeleteAtom(lockedDll);

AssertMsg(lockedDll == 0, "Failed to release the lock for chakracore.dll");

#ifdef DYNAMIC_PROFILE_STORAGE

DynamicProfileStorage::Uninitialize();

#endif

#ifdef ENABLE_JS_ETW

// Do this before DetachProcess() so that we won't have ETW rundown callbacks while destroying threadContexts.

EtwTrace::UnRegister();

#endif

// don't do anything if we are in forceful shutdown

// try to clean up handles in graceful shutdown

if (pvReserved == NULL)

{

DetachProcess();

}

#if defined(CHECK_MEMORY_LEAK) || defined(LEAK_REPORT)

else

{

ThreadContext::ReportAndCheckLeaksOnProcessDetach();

}

#endif

return TRUE;

default:

AssertMsg(FALSE, "DllMain() called with unrecognized dwReason.");

return FALSE;

}

}

#endif // _WIN32 || _WIN64

void ChakraBinaryAutoSystemInfoInit(AutoSystemInfo * autoSystemInfo)

{

autoSystemInfo->buildDateHash = JsUtil::CharacterBuffer<char>::StaticGetHashCode(__DATE__, _countof(__DATE__));

autoSystemInfo->buildTimeHash = JsUtil::CharacterBuffer<char>::StaticGetHashCode(__TIME__, _countof(__TIME__));

}