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

// Copyright (C) Microsoft Corporation and contributors. All rights reserved.

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

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

#include "stdafx.h"

void RuntimeThreadLocalData::Initialize(RuntimeThreadData* threadData)

{

this->threadData = threadData;

}

void RuntimeThreadLocalData::Uninitialize()

{

}

THREAD_LOCAL RuntimeThreadLocalData threadLocalData;

RuntimeThreadLocalData& GetRuntimeThreadLocalData()

{

return threadLocalData;

}

RuntimeThreadData::RuntimeThreadData() :

hSemaphore(nullptr),

hThread(nullptr),

sharedContent(nullptr),

receiveBroadcastCallbackFunc(nullptr),

runtime(nullptr),

context(nullptr),

parent(nullptr),

leaving(false)

{

this->hevntInitialScriptCompleted = CreateEvent(NULL, TRUE, FALSE, NULL);

this->hevntReceivedBroadcast = CreateEvent(NULL, FALSE, FALSE, NULL);

this->hevntShutdown = CreateEvent(NULL, TRUE, FALSE, NULL);

InitializeCriticalSection(&csReportQ);

}

RuntimeThreadData::~RuntimeThreadData()

{

CloseHandle(this->hevntInitialScriptCompleted);

CloseHandle(this->hevntReceivedBroadcast);

CloseHandle(this->hevntShutdown);

CloseHandle(this->hThread);

DeleteCriticalSection(&csReportQ);

}

DWORD RuntimeThreadData::ThreadProc()

{

JsValueRef scriptSource;

JsValueRef fname;

const char* fullPath = "agent source";

HRESULT hr = S_OK;

threadLocalData.Initialize(this);

IfJsErrorFailLog(ChakraRTInterface::JsCreateRuntime(JsRuntimeAttributeNone, nullptr, &runtime));

IfJsErrorFailLog(ChakraRTInterface::JsCreateContext(runtime, &context));

IfJsErrorFailLog(ChakraRTInterface::JsSetCurrentContext(context));

if (!WScriptJsrt::Initialize())

{

IfFailGo(E_FAIL);

}

IfJsErrorFailLog(ChakraRTInterface::JsCreateExternalArrayBuffer((void*)this->initialSource.c_str(),

(unsigned int)this->initialSource.size(), nullptr, nullptr, &scriptSource));

ChakraRTInterface::JsCreateString(fullPath, strlen(fullPath), &fname);

ChakraRTInterface::JsRun(scriptSource, WScriptJsrt::GetNextSourceContext(), fname, JsParseScriptAttributeNone, nullptr);

SetEvent(this->parent->hevntInitialScriptCompleted);

// loop waiting for work;

while (true)

{

HANDLE handles[] = { this->hevntReceivedBroadcast, this->hevntShutdown };

DWORD waitRet = WaitForMultipleObjects(_countof(handles), handles, false, INFINITE);

if (waitRet == WAIT_OBJECT_0)

{

JsValueRef args[3];

ChakraRTInterface::JsGetGlobalObject(&args[0]);

ChakraRTInterface::JsCreateSharedArrayBufferWithSharedContent(this->parent->sharedContent, &args[1]);

ChakraRTInterface::JsDoubleToNumber(1, &args[2]);

// notify the parent we received the data

ReleaseSemaphore(this->parent->hSemaphore, 1, NULL);

if (this->receiveBroadcastCallbackFunc)

{

ChakraRTInterface::JsCallFunction(this->receiveBroadcastCallbackFunc, args, 3, nullptr);

}

}

if (waitRet == WAIT_OBJECT_0 + 1 || this->leaving)

{

WScriptJsrt::Uninitialize();

if (this->receiveBroadcastCallbackFunc)

{

ChakraRTInterface::JsRelease(this->receiveBroadcastCallbackFunc, nullptr);

}

ChakraRTInterface::JsSetCurrentContext(nullptr);

ChakraRTInterface::JsDisposeRuntime(runtime);

threadLocalData.Uninitialize();

return 0;

}

else if (waitRet != WAIT_OBJECT_0)

{

Assert(false);

break;

}

}

Error:

ChakraRTInterface::JsSetCurrentContext(nullptr);

ChakraRTInterface::JsDisposeRuntime(runtime);

threadLocalData.Uninitialize();

return 0;

}