/* * Authored by Alex Hultman, 2018-2020. * Intellectual property of third-party. * Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0 * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include "App.h" #include #include "Utilities.h" using namespace v8; /* uWS.App.ws('/pattern', behavior) */ template void uWS_App_ws(const FunctionCallbackInfo &args) { /* pattern, behavior */ if (missingArguments(2, args)) { return; } Isolate *isolate = args.GetIsolate(); PerContextData *perContextData = (PerContextData *) Local::Cast(args.Data())->Value(); APP *app = (APP *) args.Holder()->GetAlignedPointerFromInternalField(0); /* This one is default constructed with defaults */ typename APP::WebSocketBehavior behavior = {}; NativeString pattern(args.GetIsolate(), args[0]); if (pattern.isInvalid(args)) { return; } UniquePersistent upgradePf; UniquePersistent openPf; UniquePersistent messagePf; UniquePersistent drainPf; UniquePersistent closePf; UniquePersistent pingPf; UniquePersistent pongPf; /* Get the behavior object */ if (args.Length() == 2) { Local behaviorObject = Local::Cast(args[1]); /* maxPayloadLength or default */ MaybeLocal maybeMaxPayloadLength = behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "maxPayloadLength", NewStringType::kNormal).ToLocalChecked()); if (!maybeMaxPayloadLength.IsEmpty() && !maybeMaxPayloadLength.ToLocalChecked()->IsUndefined()) { behavior.maxPayloadLength = maybeMaxPayloadLength.ToLocalChecked()->Int32Value(isolate->GetCurrentContext()).ToChecked(); } /* idleTimeout or default */ MaybeLocal maybeIdleTimeout = behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "idleTimeout", NewStringType::kNormal).ToLocalChecked()); if (!maybeIdleTimeout.IsEmpty() && !maybeIdleTimeout.ToLocalChecked()->IsUndefined()) { behavior.idleTimeout = maybeIdleTimeout.ToLocalChecked()->Int32Value(isolate->GetCurrentContext()).ToChecked(); } /* Compression or default, map from 0, 1, 2 to disabled, shared, dedicated. This is actually the enum */ MaybeLocal maybeCompression = behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "compression", NewStringType::kNormal).ToLocalChecked()); if (!maybeCompression.IsEmpty() && !maybeCompression.ToLocalChecked()->IsUndefined()) { behavior.compression = (uWS::CompressOptions) maybeCompression.ToLocalChecked()->Int32Value(isolate->GetCurrentContext()).ToChecked(); } /* maxBackpressure or default */ MaybeLocal maybeMaxBackpressure = behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "maxBackpressure", NewStringType::kNormal).ToLocalChecked()); if (!maybeMaxBackpressure.IsEmpty() && !maybeMaxBackpressure.ToLocalChecked()->IsUndefined()) { behavior.maxBackpressure = maybeMaxBackpressure.ToLocalChecked()->Int32Value(isolate->GetCurrentContext()).ToChecked(); } /* Upgrade */ upgradePf.Reset(args.GetIsolate(), Local::Cast(behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "upgrade", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked())); /* Open */ openPf.Reset(args.GetIsolate(), Local::Cast(behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "open", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked())); /* Message */ messagePf.Reset(args.GetIsolate(), Local::Cast(behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "message", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked())); /* Drain */ drainPf.Reset(args.GetIsolate(), Local::Cast(behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "drain", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked())); /* Close */ closePf.Reset(args.GetIsolate(), Local::Cast(behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "close", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked())); /* Ping */ pingPf.Reset(args.GetIsolate(), Local::Cast(behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "ping", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked())); /* Pong */ pongPf.Reset(args.GetIsolate(), Local::Cast(behaviorObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "pong", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked())); } /* Upgrade handler is always optional */ if (upgradePf != Undefined(isolate)) { behavior.upgrade = [upgradePf = std::move(upgradePf), perContextData](auto *res, auto *req, auto *context) { Isolate *isolate = perContextData->isolate; HandleScope hs(isolate); Local upgradeLf = Local::New(isolate, upgradePf); Local resObject = perContextData->resTemplate[getAppTypeIndex()].Get(isolate)->Clone(); resObject->SetAlignedPointerInInternalField(0, res); Local reqObject = perContextData->reqTemplate.Get(isolate)->Clone(); reqObject->SetAlignedPointerInInternalField(0, req); Local argv[3] = {resObject, reqObject, External::New(isolate, (void *) context)}; CallJS(isolate, upgradeLf, 3, argv); /* Properly invalidate req */ reqObject->SetAlignedPointerInInternalField(0, nullptr); /* µWS itself will terminate if not responded and not attached * onAborted handler, so we can assume it's done */ }; } /* Open handler is NOT optional for the wrapper */ behavior.open = [openPf = std::move(openPf), perContextData](auto *ws) { Isolate *isolate = perContextData->isolate; HandleScope hs(isolate); /* Create a new websocket object */ Local wsObject = perContextData->wsTemplate[getAppTypeIndex()].Get(isolate)->Clone(); wsObject->SetAlignedPointerInInternalField(0, ws); /* Retrieve temporary userData object */ PerSocketData *perSocketData = (PerSocketData *) ws->getUserData(); /* Copy entires from userData, only if we have it set (not the case for default constructor) */ if (!perSocketData->socketPf.IsEmpty()) { /* socketPf points to a stack allocated UniquePersistent, or nullptr, at this point */ Local userData = Local::New(isolate, perSocketData->socketPf); /* Merge userData and wsObject; this code is exceedingly horrible */ Local keys; if (userData->GetOwnPropertyNames(isolate->GetCurrentContext()).ToLocal(&keys)) { for (int i = 0; i < keys->Length(); i++) { wsObject->Set(isolate->GetCurrentContext(), keys->Get(isolate->GetCurrentContext(), i).ToLocalChecked(), userData->Get(isolate->GetCurrentContext(), keys->Get(isolate->GetCurrentContext(), i).ToLocalChecked()).ToLocalChecked() ).ToChecked(); } } } /* Attach a new V8 object with pointer to us, to it */ perSocketData->socketPf.Reset(isolate, wsObject); Local openLf = Local::New(isolate, openPf); if (!openLf->IsUndefined()) { Local argv[] = {wsObject}; CallJS(isolate, openLf, 1, argv); } }; /* Message handler is always optional */ if (messagePf != Undefined(isolate)) { behavior.message = [messagePf = std::move(messagePf), isolate](auto *ws, std::string_view message, uWS::OpCode opCode) { HandleScope hs(isolate); Local messageArrayBuffer = ArrayBuffer::New(isolate, (void *) message.data(), message.length()); PerSocketData *perSocketData = (PerSocketData *) ws->getUserData(); Local argv[3] = {Local::New(isolate, perSocketData->socketPf), messageArrayBuffer, Boolean::New(isolate, opCode == uWS::OpCode::BINARY)}; CallJS(isolate, Local::New(isolate, messagePf), 3, argv); /* Important: we clear the ArrayBuffer to make sure it is not invalidly used after return */ NeuterArrayBuffer(messageArrayBuffer); }; } /* Drain handler is always optional */ if (drainPf != Undefined(isolate)) { behavior.drain = [drainPf = std::move(drainPf), isolate](auto *ws) { HandleScope hs(isolate); PerSocketData *perSocketData = (PerSocketData *) ws->getUserData(); Local argv[1] = {Local::New(isolate, perSocketData->socketPf) }; CallJS(isolate, Local::New(isolate, drainPf), 1, argv); }; } /* Ping handler is always optional */ if (pingPf != Undefined(isolate)) { behavior.ping = [pingPf = std::move(pingPf), isolate](auto *ws) { HandleScope hs(isolate); PerSocketData *perSocketData = (PerSocketData *) ws->getUserData(); Local argv[1] = {Local::New(isolate, perSocketData->socketPf)}; CallJS(isolate, Local::New(isolate, pingPf), 1, argv); }; } /* Pong handler is always optional */ if (pongPf != Undefined(isolate)) { behavior.pong = [pongPf = std::move(pongPf), isolate](auto *ws) { HandleScope hs(isolate); PerSocketData *perSocketData = (PerSocketData *) ws->getUserData(); Local argv[1] = {Local::New(isolate, perSocketData->socketPf)}; CallJS(isolate, Local::New(isolate, pongPf), 1, argv); }; } /* Close handler is NOT optional for the wrapper */ behavior.close = [closePf = std::move(closePf), isolate](auto *ws, int code, std::string_view message) { HandleScope hs(isolate); Local messageArrayBuffer = ArrayBuffer::New(isolate, (void *) message.data(), message.length()); PerSocketData *perSocketData = (PerSocketData *) ws->getUserData(); Local wsObject = Local::New(isolate, perSocketData->socketPf); /* Invalidate this wsObject */ wsObject->SetAlignedPointerInInternalField(0, nullptr); /* Only call close handler if we have one set */ Local closeLf = Local::New(isolate, closePf); if (!closeLf->IsUndefined()) { Local argv[3] = {wsObject, Integer::New(isolate, code), messageArrayBuffer}; CallJS(isolate, closeLf, 3, argv); } /* This should technically not be required */ perSocketData->socketPf.Reset(); /* Again, here we clear the buffer to avoid strange bugs */ NeuterArrayBuffer(messageArrayBuffer); }; app->template ws(std::string(pattern.getString()), std::move(behavior)); /* Return this */ args.GetReturnValue().Set(args.Holder()); } /* This method wraps get, post and all http methods */ template void uWS_App_get(F f, const FunctionCallbackInfo &args) { APP *app = (APP *) args.Holder()->GetAlignedPointerFromInternalField(0); /* Pattern */ NativeString pattern(args.GetIsolate(), args[0]); if (pattern.isInvalid(args)) { return; } /* Handler */ Callback checkedCallback(args.GetIsolate(), args[1]); if (checkedCallback.isInvalid(args)) { return; } UniquePersistent cb = checkedCallback.getFunction(); /* This function requires perContextData */ PerContextData *perContextData = (PerContextData *) Local::Cast(args.Data())->Value(); (app->*f)(std::string(pattern.getString()), [cb = std::move(cb), perContextData](auto *res, auto *req) { Isolate *isolate = perContextData->isolate; HandleScope hs(isolate); Local resObject = perContextData->resTemplate[getAppTypeIndex()].Get(isolate)->Clone(); resObject->SetAlignedPointerInInternalField(0, res); Local reqObject = perContextData->reqTemplate.Get(isolate)->Clone(); reqObject->SetAlignedPointerInInternalField(0, req); Local argv[] = {resObject, reqObject}; CallJS(isolate, cb.Get(isolate), 2, argv); /* Properly invalidate req */ reqObject->SetAlignedPointerInInternalField(0, nullptr); /* µWS itself will terminate if not responded and not attached * onAborted handler, so we can assume it's done */ }); args.GetReturnValue().Set(args.Holder()); } template void uWS_App_listen(const FunctionCallbackInfo &args) { APP *app = (APP *) args.Holder()->GetAlignedPointerFromInternalField(0); Isolate *isolate = args.GetIsolate(); /* Require at least two arguments */ if (missingArguments(2, args)) { return; } /* Callback is last */ auto cb = [&args, isolate](auto *token) { /* Return a false boolean if listen failed */ Local argv[] = {token ? Local::Cast(External::New(isolate, token)) : Local::Cast(Boolean::New(isolate, false))}; /* Immediate call cannot be CallJS */ Local::Cast(args[args.Length() - 1])->Call(isolate->GetCurrentContext(), isolate->GetCurrentContext()->Global(), 1, argv).IsEmpty(); }; /* Host is first, if present */ std::string host; if (!args[0]->IsNumber()) { NativeString h(isolate, args[0]); if (h.isInvalid(args)) { return; } host = h.getString(); } /* Port, options are in the middle, if present */ std::vector numbers; for (int i = std::min(1, host.length()); i < args.Length() - 1; i++) { numbers.push_back(args[i]->Uint32Value(args.GetIsolate()->GetCurrentContext()).ToChecked()); } /* We only use the most complete overload */ app->listen(host, numbers.size() ? numbers[0] : 0, numbers.size() > 1 ? numbers[1] : 0, std::move(cb)); args.GetReturnValue().Set(args.Holder()); } template void uWS_App_publish(const FunctionCallbackInfo &args) { APP *app = (APP *) args.Holder()->GetAlignedPointerFromInternalField(0); Isolate *isolate = args.GetIsolate(); /* topic, message [isBinary, compress] */ if (missingArguments(2, args)) { return; } NativeString topic(isolate, args[0]); if (topic.isInvalid(args)) { return; } NativeString message(isolate, args[1]); if (message.isInvalid(args)) { return; } app->publish(topic.getString(), message.getString(), BooleanValue(isolate, args[2]) ? uWS::OpCode::BINARY : uWS::OpCode::TEXT, BooleanValue(isolate, args[3])); } /* This one modified per-thread static strings temporarily */ std::pair readOptionsObject(const FunctionCallbackInfo &args, int index) { Isolate *isolate = args.GetIsolate(); /* Read the options object if any */ uWS::SocketContextOptions options = {}; thread_local std::string keyFileName, certFileName, passphrase, dhParamsFileName, caFileName; if (args.Length() > index) { Local optionsObject = Local::Cast(args[index]); /* Key file name */ NativeString keyFileNameValue(isolate, optionsObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "key_file_name", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked()); if (keyFileNameValue.isInvalid(args)) { return {}; } if (keyFileNameValue.getString().length()) { keyFileName = keyFileNameValue.getString(); options.key_file_name = keyFileName.c_str(); } /* Cert file name */ NativeString certFileNameValue(isolate, optionsObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "cert_file_name", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked()); if (certFileNameValue.isInvalid(args)) { return {}; } if (certFileNameValue.getString().length()) { certFileName = certFileNameValue.getString(); options.cert_file_name = certFileName.c_str(); } /* Passphrase */ NativeString passphraseValue(isolate, optionsObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "passphrase", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked()); if (passphraseValue.isInvalid(args)) { return {}; } if (passphraseValue.getString().length()) { passphrase = passphraseValue.getString(); options.passphrase = passphrase.c_str(); } /* DH params file name */ NativeString dhParamsFileNameValue(isolate, optionsObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "dh_params_file_name", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked()); if (dhParamsFileNameValue.isInvalid(args)) { return {}; } if (dhParamsFileNameValue.getString().length()) { dhParamsFileName = dhParamsFileNameValue.getString(); options.dh_params_file_name = dhParamsFileName.c_str(); } /* CA file name */ NativeString caFileNameValue(isolate, optionsObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "ca_file_name", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked()); if (caFileNameValue.isInvalid(args)) { return {}; } if (caFileNameValue.getString().length()) { caFileName = caFileNameValue.getString(); options.ca_file_name = caFileName.c_str(); } /* ssl_prefer_low_memory_usage */ options.ssl_prefer_low_memory_usage = BooleanValue(isolate, optionsObject->Get(isolate->GetCurrentContext(), String::NewFromUtf8(isolate, "ssl_prefer_low_memory_usage", NewStringType::kNormal).ToLocalChecked()).ToLocalChecked()); } return {options, true}; } template void uWS_App_addServerName(const FunctionCallbackInfo &args) { APP *app = (APP *) args.Holder()->GetAlignedPointerFromInternalField(0); Isolate *isolate = args.GetIsolate(); NativeString hostnamePatternValue(isolate, args[0]); if (hostnamePatternValue.isInvalid(args)) { return; } std::string hostnamePattern; if (hostnamePatternValue.getString().length()) { hostnamePattern = hostnamePatternValue.getString(); } auto [options, valid] = readOptionsObject(args, 1); if (!valid) { return; } app->addServerName(hostnamePattern.c_str(), options); args.GetReturnValue().Set(args.Holder()); } template void uWS_App_removeServerName(const FunctionCallbackInfo &args) { APP *app = (APP *) args.Holder()->GetAlignedPointerFromInternalField(0); Isolate *isolate = args.GetIsolate(); NativeString hostnamePatternValue(isolate, args[0]); if (hostnamePatternValue.isInvalid(args)) { return; } std::string hostnamePattern; if (hostnamePatternValue.getString().length()) { hostnamePattern = hostnamePatternValue.getString(); } app->removeServerName(hostnamePattern.c_str()); args.GetReturnValue().Set(args.Holder()); } template void uWS_App_missingServerName(const FunctionCallbackInfo &args) { APP *app = (APP *) args.Holder()->GetAlignedPointerFromInternalField(0); Isolate *isolate = args.GetIsolate(); UniquePersistent missingPf; missingPf.Reset(args.GetIsolate(), Local::Cast(args[0])); app->missingServerName([missingPf = std::move(missingPf), isolate](const char *hostname) { /* We hand a JavaScript string here */ HandleScope hs(isolate); Local missingLf = Local::New(isolate, missingPf); Local argv[1] = {String::NewFromUtf8(isolate, hostname, NewStringType::kNormal).ToLocalChecked()}; CallJS(isolate, missingLf, 1, argv); }); args.GetReturnValue().Set(args.Holder()); } template void uWS_App(const FunctionCallbackInfo &args) { Isolate *isolate = args.GetIsolate(); Local appTemplate = FunctionTemplate::New(isolate); appTemplate->SetClassName(String::NewFromUtf8(isolate, std::is_same::value ? "uWS.SSLApp" : "uWS.App", NewStringType::kNormal).ToLocalChecked()); auto [options, valid] = readOptionsObject(args, 0); if (!valid) { return; } /* uSockets copies strings here */ APP *app = new APP(options); /* Throw if we failed to construct the app */ if (app->constructorFailed()) { delete app; args.GetReturnValue().Set(isolate->ThrowException(String::NewFromUtf8(isolate, "App construction failed", NewStringType::kNormal).ToLocalChecked())); return; } appTemplate->InstanceTemplate()->SetInternalFieldCount(1); /* All the http methods */ appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "get", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, [](auto &args) { uWS_App_get(&APP::get, args); }, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "post", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, [](auto &args) { uWS_App_get(&APP::post, args); }, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "options", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, [](auto &args) { uWS_App_get(&APP::options, args); }, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "del", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, [](auto &args) { uWS_App_get(&APP::del, args); }, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "patch", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, [](auto &args) { uWS_App_get(&APP::patch, args); }, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "put", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, [](auto &args) { uWS_App_get(&APP::put, args); }, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "head", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, [](auto &args) { uWS_App_get(&APP::head, args); }, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "connect", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, [](auto &args) { uWS_App_get(&APP::connect, args); }, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "trace", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, [](auto &args) { uWS_App_get(&APP::trace, args); }, args.Data())); /* Any http method */ appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "any", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, [](auto &args) { uWS_App_get(&APP::any, args); }, args.Data())); /* ws, listen */ appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "ws", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, uWS_App_ws, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "listen", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, uWS_App_listen, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "publish", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, uWS_App_publish, args.Data())); /* SNI */ appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "addServerName", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, uWS_App_addServerName, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "removeServerName", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, uWS_App_removeServerName, args.Data())); appTemplate->PrototypeTemplate()->Set(String::NewFromUtf8(isolate, "missingServerName", NewStringType::kNormal).ToLocalChecked(), FunctionTemplate::New(isolate, uWS_App_missingServerName, args.Data())); Local localApp = appTemplate->GetFunction(isolate->GetCurrentContext()).ToLocalChecked()->NewInstance(isolate->GetCurrentContext()).ToLocalChecked(); localApp->SetAlignedPointerInInternalField(0, app); PerContextData *perContextData = (PerContextData *) Local::Cast(args.Data())->Value(); /* Add this to our delete list */ if constexpr (std::is_same::value) { perContextData->sslApps.emplace_back(app); } else { perContextData->apps.emplace_back(app); } args.GetReturnValue().Set(localApp); }