321 lines
8.5 KiB
C++
321 lines
8.5 KiB
C++
#include "stdafx.h"
|
|
|
|
#include "JSSocketWorker.h"
|
|
#include "ScriptAPI.h"
|
|
|
|
#pragma warning(disable : 4702) // disable unreachable code warning
|
|
|
|
static CJSSocketWorker * GetThisSocket(duk_context * ctx);
|
|
static duk_ret_t RequireBufferDataOrString(duk_context * ctx, duk_idx_t idx, const char ** data, duk_size_t * size);
|
|
static duk_int_t RegisterWriteCallback(duk_context * ctx, duk_idx_t idx);
|
|
static void RegisterWriteEndCallback(duk_context * ctx, duk_idx_t idx);
|
|
|
|
void ScriptAPI::Define_Socket(duk_context * ctx)
|
|
{
|
|
const DukPropListEntry prototype[] = {
|
|
{"connect", DukCFunction(js_Socket_connect)},
|
|
{"write", DukCFunction(js_Socket_write)},
|
|
{"end", DukCFunction(js_Socket_end)},
|
|
{"close", DukCFunction(js_Socket_close)},
|
|
{"on", DukCFunction(js__Emitter_on)},
|
|
{"off", DukCFunction(js__Emitter_off)},
|
|
{"localAddress", DukGetter(js_Socket__get_localAddress)},
|
|
{"localPort", DukGetter(js_Socket__get_localPort)},
|
|
{"remoteAddress", DukGetter(js_Socket__get_remoteAddress)},
|
|
{"remotePort", DukGetter(js_Socket__get_remotePort)},
|
|
{"addressFamily", DukGetter(js_Socket__get_addressFamily)},
|
|
{nullptr},
|
|
};
|
|
|
|
DefineGlobalClass(ctx, "Socket", js_Socket__constructor, prototype, nullptr);
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket__constructor(duk_context * ctx)
|
|
{
|
|
CheckArgs(ctx, {Arg_OptObject});
|
|
|
|
if (!duk_is_constructor_call(ctx))
|
|
{
|
|
return DUK_RET_ERROR;
|
|
}
|
|
|
|
bool bAllowHalfOpen = false;
|
|
|
|
CScriptInstance * inst = GetInstance(ctx);
|
|
|
|
if (duk_is_object(ctx, 0))
|
|
{
|
|
if (duk_has_prop_string(ctx, 0, "allowHalfOpen"))
|
|
{
|
|
duk_get_prop_string(ctx, 0, "allowHalfOpen");
|
|
bAllowHalfOpen = duk_get_boolean_default(ctx, -1, 0) != 0;
|
|
duk_pop(ctx);
|
|
}
|
|
}
|
|
|
|
duk_push_this(ctx);
|
|
void * objectHeapPtr = duk_get_heapptr(ctx, -1);
|
|
|
|
InitEmitter(ctx, -1, {"data", "end", "connect", "error", "close", "drain", "lookup"});
|
|
|
|
duk_push_uint(ctx, 0);
|
|
duk_put_prop_string(ctx, -2, HS_socketNextWriteCallbackId);
|
|
|
|
duk_push_object(ctx);
|
|
duk_put_prop_string(ctx, -2, HS_socketWriteCallbacks);
|
|
|
|
duk_push_array(ctx);
|
|
duk_put_prop_string(ctx, -2, HS_socketWriteEndCallbacks);
|
|
|
|
CJSSocketWorker * socketWorker = new CJSSocketWorker(inst, objectHeapPtr, bAllowHalfOpen);
|
|
|
|
duk_push_pointer(ctx, (void *)socketWorker);
|
|
duk_put_prop_string(ctx, -2, HS_socketWorkerPtr);
|
|
|
|
duk_push_c_function(ctx, js_Socket__finalizer, 1);
|
|
duk_set_finalizer(ctx, -2);
|
|
|
|
return 0;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket__finalizer(duk_context * ctx)
|
|
{
|
|
duk_get_prop_string(ctx, 0, HS_socketWorkerPtr);
|
|
CJSSocketWorker * socketWorker = (CJSSocketWorker *)duk_get_pointer(ctx, -1);
|
|
|
|
if (socketWorker == nullptr)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
UnrefObject(ctx, 0);
|
|
delete socketWorker;
|
|
|
|
return 0;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket_connect(duk_context * ctx)
|
|
{
|
|
CheckArgs(ctx, {Arg_Number, Arg_String, Arg_OptFunction});
|
|
|
|
CJSSocketWorker * socketWorker = GetThisSocket(ctx);
|
|
|
|
duk_idx_t nargs = duk_get_top(ctx);
|
|
unsigned short port = (unsigned short)duk_get_uint(ctx, 0);
|
|
const char * host = duk_get_string(ctx, 1);
|
|
|
|
if (nargs == 3)
|
|
{
|
|
// add 'connect' event listener
|
|
duk_push_c_function(ctx, js__Emitter_on, 2); // todo once
|
|
duk_push_this(ctx);
|
|
duk_push_string(ctx, "connect");
|
|
duk_pull(ctx, 2);
|
|
duk_pcall_method(ctx, 2);
|
|
duk_pop(ctx);
|
|
}
|
|
|
|
duk_push_this(ctx);
|
|
RefObject(ctx, -1);
|
|
|
|
socketWorker->Init(host, port);
|
|
socketWorker->StartWorkerProc();
|
|
|
|
return 0;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket_write(duk_context * ctx)
|
|
{
|
|
CJSSocketWorker * socketWorker = GetThisSocket(ctx);
|
|
|
|
const char * data;
|
|
duk_size_t size;
|
|
duk_int_t callbackId = -1;
|
|
|
|
duk_idx_t nargs = duk_get_top(ctx);
|
|
RequireBufferDataOrString(ctx, 0, &data, &size);
|
|
|
|
if (nargs == 2 && duk_is_function(ctx, 1))
|
|
{
|
|
callbackId = RegisterWriteCallback(ctx, 1);
|
|
}
|
|
|
|
socketWorker->Write((char *)data, size, callbackId, false);
|
|
return 0;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket_end(duk_context * ctx)
|
|
{
|
|
CJSSocketWorker * socketWorker = GetThisSocket(ctx);
|
|
|
|
const char * data;
|
|
duk_size_t size;
|
|
duk_int_t callbackId = -1;
|
|
|
|
duk_idx_t nargs = duk_get_top(ctx);
|
|
RequireBufferDataOrString(ctx, 0, &data, &size);
|
|
|
|
if (nargs == 2 && duk_is_function(ctx, 1))
|
|
{
|
|
RegisterWriteEndCallback(ctx, 1);
|
|
}
|
|
|
|
socketWorker->Write((char *)data, size, callbackId, true);
|
|
return 0;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket__invokeWriteCallback(duk_context * ctx)
|
|
{
|
|
duk_int_t callbackId = duk_get_int(ctx, 0);
|
|
|
|
duk_push_this(ctx);
|
|
duk_get_prop_string(ctx, -1, HS_socketWriteCallbacks);
|
|
duk_get_prop_index(ctx, -1, callbackId);
|
|
duk_dup(ctx, -3);
|
|
|
|
if (duk_pcall_method(ctx, 0) != 0)
|
|
{
|
|
duk_throw(ctx);
|
|
}
|
|
|
|
duk_pop(ctx);
|
|
duk_del_prop_index(ctx, -1, callbackId);
|
|
|
|
return 0;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket__invokeWriteEndCallbacks(duk_context * ctx)
|
|
{
|
|
duk_push_this(ctx);
|
|
duk_get_prop_string(ctx, -1, HS_socketWriteEndCallbacks);
|
|
duk_size_t numCallbacks = duk_get_length(ctx, -1);
|
|
for (duk_size_t i = 0; i < numCallbacks; i++)
|
|
{
|
|
duk_get_prop_index(ctx, -1, (int)((UINT_PTR)i));
|
|
duk_dup(ctx, -3);
|
|
|
|
if (duk_pcall_method(ctx, 0) != 0)
|
|
{
|
|
duk_throw(ctx);
|
|
}
|
|
|
|
duk_pop(ctx);
|
|
}
|
|
|
|
duk_pop(ctx);
|
|
|
|
// reset array
|
|
duk_push_array(ctx);
|
|
duk_put_prop_string(ctx, -2, HS_socketWriteEndCallbacks);
|
|
|
|
return 0;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket_close(duk_context * ctx)
|
|
{
|
|
CJSSocketWorker * socketWorker = GetThisSocket(ctx);
|
|
socketWorker->StopWorkerProc();
|
|
return 0;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket__get_localAddress(duk_context * ctx)
|
|
{
|
|
CJSSocketWorker * socketWorker = GetThisSocket(ctx);
|
|
duk_push_string(ctx, socketWorker->GetLocalAddress().c_str());
|
|
return 1;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket__get_localPort(duk_context * ctx)
|
|
{
|
|
CJSSocketWorker * socketWorker = GetThisSocket(ctx);
|
|
duk_push_uint(ctx, socketWorker->GetLocalPort());
|
|
return 1;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket__get_remoteAddress(duk_context * ctx)
|
|
{
|
|
CJSSocketWorker * socketWorker = GetThisSocket(ctx);
|
|
duk_push_string(ctx, socketWorker->GetRemoteAddress().c_str());
|
|
return 1;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket__get_remotePort(duk_context * ctx)
|
|
{
|
|
CJSSocketWorker * socketWorker = GetThisSocket(ctx);
|
|
duk_push_uint(ctx, socketWorker->GetRemotePort());
|
|
return 1;
|
|
}
|
|
|
|
duk_ret_t ScriptAPI::js_Socket__get_addressFamily(duk_context * ctx)
|
|
{
|
|
CJSSocketWorker * socketWorker = GetThisSocket(ctx);
|
|
duk_push_string(ctx, socketWorker->GetFamily());
|
|
return 1;
|
|
}
|
|
|
|
CJSSocketWorker * GetThisSocket(duk_context * ctx)
|
|
{
|
|
duk_push_this(ctx);
|
|
duk_get_prop_string(ctx, -1, HS_socketWorkerPtr);
|
|
CJSSocketWorker * socketWorker = (CJSSocketWorker *)duk_get_pointer(ctx, -1);
|
|
duk_pop_n(ctx, 2);
|
|
|
|
if (socketWorker == nullptr)
|
|
{
|
|
duk_push_error_object(ctx, DUK_ERR_ERROR, "internal socket object is null");
|
|
return duk_throw(ctx);
|
|
}
|
|
|
|
return socketWorker;
|
|
}
|
|
|
|
duk_ret_t RequireBufferDataOrString(duk_context * ctx, duk_idx_t idx, const char ** data, duk_size_t * size)
|
|
{
|
|
if (duk_is_buffer_data(ctx, idx))
|
|
{
|
|
*data = (const char *)duk_get_buffer_data(ctx, idx, size);
|
|
return 0;
|
|
}
|
|
else if (duk_is_string(ctx, idx))
|
|
{
|
|
*data = duk_get_lstring(ctx, idx, size);
|
|
return 0;
|
|
}
|
|
|
|
duk_push_error_object(ctx, DUK_ERR_TYPE_ERROR, "arg %d must be buffer-like or string", idx);
|
|
return duk_throw(ctx);
|
|
}
|
|
|
|
duk_int_t RegisterWriteCallback(duk_context * ctx, duk_idx_t idx)
|
|
{
|
|
idx = duk_normalize_index(ctx, idx);
|
|
|
|
duk_push_this(ctx);
|
|
duk_get_prop_string(ctx, -1, HS_socketNextWriteCallbackId);
|
|
duk_int_t callbackId = duk_get_int(ctx, -1);
|
|
duk_pop(ctx);
|
|
|
|
duk_get_prop_string(ctx, -1, HS_socketWriteCallbacks);
|
|
duk_dup(ctx, idx);
|
|
duk_put_prop_index(ctx, -2, callbackId);
|
|
duk_pop(ctx);
|
|
|
|
duk_push_int(ctx, callbackId + 1);
|
|
duk_put_prop_string(ctx, -2, HS_socketNextWriteCallbackId);
|
|
|
|
return callbackId;
|
|
}
|
|
|
|
void RegisterWriteEndCallback(duk_context * ctx, duk_idx_t idx)
|
|
{
|
|
idx = duk_normalize_index(ctx, idx);
|
|
|
|
duk_push_this(ctx);
|
|
|
|
duk_get_prop_string(ctx, -1, HS_socketWriteEndCallbacks);
|
|
duk_size_t callbackIdx = duk_get_length(ctx, -1);
|
|
duk_dup(ctx, idx);
|
|
duk_put_prop_index(ctx, -2, (int)((UINT_PTR)callbackIdx));
|
|
duk_pop(ctx);
|
|
}
|