| // Copyright 2022 The Fuchsia Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "src/sys/fuzzing/common/component-context.h" |
| |
| #include <lib/async-loop/default.h> |
| #include <lib/sys/cpp/component_context.h> |
| |
| namespace fuzzing { |
| |
| std::unique_ptr<ComponentContext> ComponentContext::Create() { |
| static bool once = true; |
| FX_CHECK(once) << "ComponentContext::Create called more than once."; |
| once = false; |
| auto context = sys::ComponentContext::Create(); |
| auto loop = std::make_unique<async::Loop>(&kAsyncLoopConfigAttachToCurrentThread); |
| auto executor = MakeExecutor(loop->dispatcher()); |
| auto svc = context->svc(); |
| auto outgoing = context->outgoing(); |
| return std::make_unique<ComponentContext>(std::move(loop), std::move(executor), std::move(svc), |
| std::move(outgoing)); |
| } |
| |
| std::unique_ptr<ComponentContext> ComponentContext::CreateAuxillary() { |
| auto loop = std::make_unique<async::Loop>(&kAsyncLoopConfigNoAttachToCurrentThread); |
| auto executor = MakeExecutor(loop->dispatcher()); |
| auto svc = sys::ServiceDirectory::CreateFromNamespace(); |
| std::unique_ptr<sys::OutgoingDirectory> outgoing; |
| return std::make_unique<ComponentContext>(std::move(loop), std::move(executor), std::move(svc), |
| std::move(outgoing)); |
| } |
| |
| std::unique_ptr<ComponentContext> ComponentContext::CreateWithExecutor(ExecutorPtr executor) { |
| auto context = sys::ComponentContext::Create(); |
| std::unique_ptr<async::Loop> loop; |
| auto svc = context->svc(); |
| auto outgoing = context->outgoing(); |
| return std::make_unique<ComponentContext>(std::move(loop), std::move(executor), std::move(svc), |
| std::move(outgoing)); |
| } |
| |
| ComponentContext::ComponentContext(LoopPtr loop, ExecutorPtr executor, ServiceDirectoryPtr svc, |
| OutgoingDirectoryPtr outgoing) |
| : loop_(std::move(loop)), |
| executor_(std::move(executor)), |
| svc_(std::move(svc)), |
| outgoing_(std::move(outgoing)) {} |
| |
| ComponentContext::~ComponentContext() { |
| if (loop_ && !outgoing_) { |
| // Auxiliary context. |
| loop_->Shutdown(); |
| loop_->JoinThreads(); |
| } |
| } |
| |
| // Runs the message loop on the current thread. This method should only be called at most once. |
| zx_status_t ComponentContext::Run() { |
| FX_CHECK(loop_); |
| if (!outgoing_) { |
| // Auxiliary context. |
| return loop_->StartThread(); |
| } |
| outgoing_->ServeFromStartupInfo(loop_->dispatcher()); |
| return loop_->Run(); |
| } |
| |
| zx_status_t ComponentContext::RunUntilIdle() { |
| FX_CHECK(loop_); |
| return loop_->RunUntilIdle(); |
| } |
| |
| } // namespace fuzzing |