// Copyright (c) 2022 Microsoft, Inc. // Use of this source code is governed by the MIT license that can be // found in the LICENSE file. #ifndef ELECTRON_SHELL_BROWSER_API_ELECTRON_API_UTILITY_PROCESS_H_ #define ELECTRON_SHELL_BROWSER_API_ELECTRON_API_UTILITY_PROCESS_H_ #include #include #include #include #include "base/containers/id_map.h" #include "base/environment.h" #include "base/memory/weak_ptr.h" #include "base/process/process_handle.h" #include "gin/wrappable.h" #include "mojo/public/cpp/bindings/connector.h" #include "mojo/public/cpp/bindings/message.h" #include "mojo/public/cpp/bindings/remote.h" #include "shell/browser/event_emitter_mixin.h" #include "shell/common/gin_helper/pinnable.h" #include "shell/services/node/public/mojom/node_service.mojom.h" #include "v8/include/v8.h" namespace gin { class Arguments; template class Handle; } // namespace gin namespace base { class Process; } // namespace base namespace electron { namespace api { class UtilityProcessWrapper : public gin::Wrappable, public gin_helper::Pinnable, public gin_helper::EventEmitterMixin, public mojo::MessageReceiver { public: enum class IOHandle : size_t { STDIN = 0, STDOUT = 1, STDERR = 2 }; enum class IOType { IO_PIPE, IO_INHERIT, IO_IGNORE }; ~UtilityProcessWrapper() override; static gin::Handle Create(gin::Arguments* args); static raw_ptr FromProcessId(base::ProcessId pid); void Shutdown(int exit_code); // gin::Wrappable static gin::WrapperInfo kWrapperInfo; gin::ObjectTemplateBuilder GetObjectTemplateBuilder( v8::Isolate* isolate) override; const char* GetTypeName() override; private: UtilityProcessWrapper(node::mojom::NodeServiceParamsPtr params, std::u16string display_name, std::map stdio, base::EnvironmentMap env_map, base::FilePath current_working_directory, bool use_plugin_helper); void OnServiceProcessDisconnected(uint32_t error_code, const std::string& description); void OnServiceProcessLaunched(const base::Process& process); void CloseConnectorPort(); void PostMessage(gin::Arguments* args); bool Kill() const; v8::Local GetOSProcessId(v8::Isolate* isolate) const; // mojo::MessageReceiver bool Accept(mojo::Message* mojo_message) override; base::ProcessId pid_ = base::kNullProcessId; #if BUILDFLAG(IS_WIN) // Non-owning handles, these will be closed when the // corresponding FD are closed via _close. HANDLE stdout_read_handle_; HANDLE stderr_read_handle_; #endif int stdout_read_fd_ = -1; int stderr_read_fd_ = -1; bool connector_closed_ = false; std::unique_ptr connector_; blink::MessagePortDescriptor host_port_; mojo::Remote node_service_remote_; base::WeakPtrFactory weak_factory_{this}; }; } // namespace api } // namespace electron #endif // ELECTRON_SHELL_BROWSER_API_ELECTRON_API_UTILITY_PROCESS_H_