LibJS: Convert Console to use MarkedVector<Value>

Using a Vector<Value> is unsafe as GC cannot see the stored values.
This is then vended to outside users of ConsoleClient, e.g. LibWeb and
WebContent, which is then outside of LibJS's control.

An example issue is if the client stores it for later use and forgets
to visit the stored values, meaning they can be destroyed at any time.
We can save the client from this by vending a MarkedVector<Value> to
them.
This commit is contained in:
Luke Wilde 2022-05-07 00:20:22 +01:00 committed by Linus Groh
parent f04911e777
commit 05748ed607
Notes: sideshowbarker 2024-07-18 05:37:06 +09:00
5 changed files with 42 additions and 28 deletions

View file

@ -138,7 +138,8 @@ ThrowCompletionOr<Value> Console::count()
String concat = String::formatted("{}: {}", label, map.get(label).value()); String concat = String::formatted("{}: {}", label, map.get(label).value());
// 5. Perform Logger("count", « concat »). // 5. Perform Logger("count", « concat »).
Vector<Value> concat_as_vector { js_string(vm(), concat) }; MarkedVector<Value> concat_as_vector { global_object().heap() };
concat_as_vector.append(js_string(vm(), concat));
if (m_client) if (m_client)
TRY(m_client->logger(LogLevel::Count, concat_as_vector)); TRY(m_client->logger(LogLevel::Count, concat_as_vector));
return js_undefined(); return js_undefined();
@ -163,7 +164,8 @@ ThrowCompletionOr<Value> Console::count_reset()
// that the given label does not have an associated count. // that the given label does not have an associated count.
auto message = String::formatted("\"{}\" doesn't have a count", label); auto message = String::formatted("\"{}\" doesn't have a count", label);
// 2. Perform Logger("countReset", « message »); // 2. Perform Logger("countReset", « message »);
Vector<Value> message_as_vector { js_string(vm(), message) }; MarkedVector<Value> message_as_vector { global_object().heap() };
message_as_vector.append(js_string(vm(), message));
if (m_client) if (m_client)
TRY(m_client->logger(LogLevel::CountReset, message_as_vector)); TRY(m_client->logger(LogLevel::CountReset, message_as_vector));
} }
@ -183,7 +185,7 @@ ThrowCompletionOr<Value> Console::assert_()
auto message = js_string(vm(), "Assertion failed"); auto message = js_string(vm(), "Assertion failed");
// NOTE: Assemble `data` from the function arguments. // NOTE: Assemble `data` from the function arguments.
Vector<Value> data; MarkedVector<Value> data { global_object().heap() };
if (vm().argument_count() > 1) { if (vm().argument_count() > 1) {
data.ensure_capacity(vm().argument_count() - 1); data.ensure_capacity(vm().argument_count() - 1);
for (size_t i = 1; i < vm().argument_count(); ++i) { for (size_t i = 1; i < vm().argument_count(); ++i) {
@ -309,8 +311,11 @@ ThrowCompletionOr<Value> Console::time()
// 1. If the associated timer table contains an entry with key label, return, optionally reporting // 1. If the associated timer table contains an entry with key label, return, optionally reporting
// a warning to the console indicating that a timer with label `label` has already been started. // a warning to the console indicating that a timer with label `label` has already been started.
if (m_timer_table.contains(label)) { if (m_timer_table.contains(label)) {
if (m_client) if (m_client) {
TRY(m_client->printer(LogLevel::Warn, { Vector<Value> { js_string(vm(), String::formatted("Timer '{}' already exists.", label)) } })); MarkedVector<Value> timer_already_exists_warning_message_as_vector { global_object().heap() };
timer_already_exists_warning_message_as_vector.append(js_string(vm(), String::formatted("Timer '{}' already exists.", label)));
TRY(m_client->printer(LogLevel::Warn, move(timer_already_exists_warning_message_as_vector)));
}
return js_undefined(); return js_undefined();
} }
@ -332,8 +337,11 @@ ThrowCompletionOr<Value> Console::time_log()
// NOTE: Warn if the timer doesn't exist. Not part of the spec yet, but discussed here: https://github.com/whatwg/console/issues/134 // NOTE: Warn if the timer doesn't exist. Not part of the spec yet, but discussed here: https://github.com/whatwg/console/issues/134
if (maybe_start_time == m_timer_table.end()) { if (maybe_start_time == m_timer_table.end()) {
if (m_client) if (m_client) {
TRY(m_client->printer(LogLevel::Warn, { Vector<Value> { js_string(vm(), String::formatted("Timer '{}' does not exist.", label)) } })); MarkedVector<Value> timer_does_not_exist_warning_message_as_vector { global_object().heap() };
timer_does_not_exist_warning_message_as_vector.append(js_string(vm(), String::formatted("Timer '{}' does not exist.", label)));
TRY(m_client->printer(LogLevel::Warn, move(timer_does_not_exist_warning_message_as_vector)));
}
return js_undefined(); return js_undefined();
} }
auto start_time = maybe_start_time->value; auto start_time = maybe_start_time->value;
@ -345,7 +353,7 @@ ThrowCompletionOr<Value> Console::time_log()
auto concat = String::formatted("{}: {}", label, duration); auto concat = String::formatted("{}: {}", label, duration);
// 5. Prepend concat to data. // 5. Prepend concat to data.
Vector<Value> data; MarkedVector<Value> data { global_object().heap() };
data.ensure_capacity(vm().argument_count()); data.ensure_capacity(vm().argument_count());
data.append(js_string(vm(), concat)); data.append(js_string(vm(), concat));
for (size_t i = 1; i < vm().argument_count(); ++i) for (size_t i = 1; i < vm().argument_count(); ++i)
@ -353,7 +361,7 @@ ThrowCompletionOr<Value> Console::time_log()
// 6. Perform Printer("timeLog", data). // 6. Perform Printer("timeLog", data).
if (m_client) if (m_client)
TRY(m_client->printer(LogLevel::TimeLog, data)); TRY(m_client->printer(LogLevel::TimeLog, move(data)));
return js_undefined(); return js_undefined();
} }
@ -370,8 +378,11 @@ ThrowCompletionOr<Value> Console::time_end()
// NOTE: Warn if the timer doesn't exist. Not part of the spec yet, but discussed here: https://github.com/whatwg/console/issues/134 // NOTE: Warn if the timer doesn't exist. Not part of the spec yet, but discussed here: https://github.com/whatwg/console/issues/134
if (maybe_start_time == m_timer_table.end()) { if (maybe_start_time == m_timer_table.end()) {
if (m_client) if (m_client) {
TRY(m_client->printer(LogLevel::Warn, { Vector<Value> { js_string(vm(), String::formatted("Timer '{}' does not exist.", label)) } })); MarkedVector<Value> timer_does_not_exist_warning_message_as_vector { global_object().heap() };
timer_does_not_exist_warning_message_as_vector.append(js_string(vm(), String::formatted("Timer '{}' does not exist.", label)));
TRY(m_client->printer(LogLevel::Warn, move(timer_does_not_exist_warning_message_as_vector)));
}
return js_undefined(); return js_undefined();
} }
auto start_time = maybe_start_time->value; auto start_time = maybe_start_time->value;
@ -387,15 +398,16 @@ ThrowCompletionOr<Value> Console::time_end()
// 6. Perform Printer("timeEnd", « concat »). // 6. Perform Printer("timeEnd", « concat »).
if (m_client) { if (m_client) {
Vector<Value> concat_as_vector { js_string(vm(), concat) }; MarkedVector<Value> concat_as_vector { global_object().heap() };
TRY(m_client->printer(LogLevel::TimeEnd, concat_as_vector)); concat_as_vector.append(js_string(vm(), concat));
TRY(m_client->printer(LogLevel::TimeEnd, move(concat_as_vector)));
} }
return js_undefined(); return js_undefined();
} }
Vector<Value> Console::vm_arguments() MarkedVector<Value> Console::vm_arguments()
{ {
Vector<Value> arguments; MarkedVector<Value> arguments { global_object().heap() };
arguments.ensure_capacity(vm().argument_count()); arguments.ensure_capacity(vm().argument_count());
for (size_t i = 0; i < vm().argument_count(); ++i) { for (size_t i = 0; i < vm().argument_count(); ++i) {
arguments.append(vm().argument(i)); arguments.append(vm().argument(i));
@ -429,7 +441,7 @@ void Console::output_debug_message([[maybe_unused]] LogLevel log_level, [[maybe_
#endif #endif
} }
ThrowCompletionOr<String> Console::value_vector_to_string(Vector<Value>& values) ThrowCompletionOr<String> Console::value_vector_to_string(MarkedVector<Value> const& values)
{ {
StringBuilder builder; StringBuilder builder;
for (auto const& item : values) { for (auto const& item : values) {
@ -471,7 +483,7 @@ VM& ConsoleClient::vm()
} }
// 2.1. Logger(logLevel, args), https://console.spec.whatwg.org/#logger // 2.1. Logger(logLevel, args), https://console.spec.whatwg.org/#logger
ThrowCompletionOr<Value> ConsoleClient::logger(Console::LogLevel log_level, Vector<Value>& args) ThrowCompletionOr<Value> ConsoleClient::logger(Console::LogLevel log_level, MarkedVector<Value> const& args)
{ {
auto& global_object = this->global_object(); auto& global_object = this->global_object();
@ -487,8 +499,9 @@ ThrowCompletionOr<Value> ConsoleClient::logger(Console::LogLevel log_level, Vect
// 4. If rest is empty, perform Printer(logLevel, « first ») and return. // 4. If rest is empty, perform Printer(logLevel, « first ») and return.
if (rest_size == 0) { if (rest_size == 0) {
auto first_as_vector = Vector { first }; MarkedVector<Value> first_as_vector { global_object.heap() };
return printer(log_level, first_as_vector); first_as_vector.append(first);
return printer(log_level, move(first_as_vector));
} }
// 5. If first does not contain any format specifiers, perform Printer(logLevel, args). // 5. If first does not contain any format specifiers, perform Printer(logLevel, args).
@ -505,7 +518,7 @@ ThrowCompletionOr<Value> ConsoleClient::logger(Console::LogLevel log_level, Vect
} }
// 2.2. Formatter(args), https://console.spec.whatwg.org/#formatter // 2.2. Formatter(args), https://console.spec.whatwg.org/#formatter
ThrowCompletionOr<Vector<Value>> ConsoleClient::formatter(Vector<Value>& args) ThrowCompletionOr<MarkedVector<Value>> ConsoleClient::formatter(MarkedVector<Value> const& args)
{ {
// TODO: Actually implement formatting // TODO: Actually implement formatting
return args; return args;

View file

@ -61,7 +61,7 @@ public:
GlobalObject const& global_object() const { return m_global_object; } GlobalObject const& global_object() const { return m_global_object; }
VM& vm(); VM& vm();
Vector<Value> vm_arguments(); MarkedVector<Value> vm_arguments();
HashMap<String, unsigned>& counters() { return m_counters; } HashMap<String, unsigned>& counters() { return m_counters; }
HashMap<String, unsigned> const& counters() const { return m_counters; } HashMap<String, unsigned> const& counters() const { return m_counters; }
@ -86,7 +86,7 @@ public:
void output_debug_message(LogLevel log_level, String output) const; void output_debug_message(LogLevel log_level, String output) const;
private: private:
ThrowCompletionOr<String> value_vector_to_string(Vector<Value>&); ThrowCompletionOr<String> value_vector_to_string(MarkedVector<Value> const&);
ThrowCompletionOr<String> format_time_since(Core::ElapsedTimer timer); ThrowCompletionOr<String> format_time_since(Core::ElapsedTimer timer);
GlobalObject& m_global_object; GlobalObject& m_global_object;
@ -104,10 +104,10 @@ public:
{ {
} }
using PrinterArguments = Variant<Console::Group, Console::Trace, Vector<Value>>; using PrinterArguments = Variant<Console::Group, Console::Trace, MarkedVector<Value>>;
ThrowCompletionOr<Value> logger(Console::LogLevel log_level, Vector<Value>& args); ThrowCompletionOr<Value> logger(Console::LogLevel log_level, MarkedVector<Value> const& args);
ThrowCompletionOr<Vector<Value>> formatter(Vector<Value>& args); ThrowCompletionOr<MarkedVector<Value>> formatter(MarkedVector<Value> const& args);
virtual ThrowCompletionOr<Value> printer(Console::LogLevel log_level, PrinterArguments) = 0; virtual ThrowCompletionOr<Value> printer(Console::LogLevel log_level, PrinterArguments) = 0;
virtual void clear() = 0; virtual void clear() = 0;

View file

@ -4,6 +4,7 @@
* SPDX-License-Identifier: BSD-2-Clause * SPDX-License-Identifier: BSD-2-Clause
*/ */
#include <LibJS/Heap/MarkedVector.h>
#include <LibJS/Runtime/Completion.h> #include <LibJS/Runtime/Completion.h>
#include <LibWeb/HTML/WorkerDebugConsoleClient.h> #include <LibWeb/HTML/WorkerDebugConsoleClient.h>
@ -52,7 +53,7 @@ JS::ThrowCompletionOr<JS::Value> WorkerDebugConsoleClient::printer(JS::Console::
return JS::js_undefined(); return JS::js_undefined();
} }
auto output = String::join(" ", arguments.get<Vector<JS::Value>>()); auto output = String::join(" ", arguments.get<JS::MarkedVector<JS::Value>>());
m_console.output_debug_message(log_level, output); m_console.output_debug_message(log_level, output);
switch (log_level) { switch (log_level) {

View file

@ -162,7 +162,7 @@ JS::ThrowCompletionOr<JS::Value> WebContentConsoleClient::printer(JS::Console::L
return JS::js_undefined(); return JS::js_undefined();
} }
auto output = String::join(" ", arguments.get<Vector<JS::Value>>()); auto output = String::join(" ", arguments.get<JS::MarkedVector<JS::Value>>());
m_console.output_debug_message(log_level, output); m_console.output_debug_message(log_level, output);
StringBuilder html; StringBuilder html;

View file

@ -1374,7 +1374,7 @@ public:
return JS::js_undefined(); return JS::js_undefined();
} }
auto output = String::join(" ", arguments.get<Vector<JS::Value>>()); auto output = String::join(" ", arguments.get<JS::MarkedVector<JS::Value>>());
m_console.output_debug_message(log_level, output); m_console.output_debug_message(log_level, output);
switch (log_level) { switch (log_level) {