| // Copyright 2018 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/developer/debug/zxdb/client/system_impl.h" |
| |
| #include <filesystem> |
| #include <set> |
| |
| #include "src/developer/debug/shared/logging/debug.h" |
| #include "src/developer/debug/shared/message_loop.h" |
| #include "src/developer/debug/zxdb/client/breakpoint_impl.h" |
| #include "src/developer/debug/zxdb/client/filter.h" |
| #include "src/developer/debug/zxdb/client/job_context_impl.h" |
| #include "src/developer/debug/zxdb/client/process_impl.h" |
| #include "src/developer/debug/zxdb/client/remote_api.h" |
| #include "src/developer/debug/zxdb/client/session.h" |
| #include "src/developer/debug/zxdb/client/setting_schema_definition.h" |
| #include "src/developer/debug/zxdb/client/system_observer.h" |
| #include "src/developer/debug/zxdb/client/target_impl.h" |
| #include "src/developer/debug/zxdb/client/thread_impl.h" |
| #include "src/developer/debug/zxdb/common/string_util.h" |
| #include "src/developer/debug/zxdb/symbols/loaded_module_symbols.h" |
| #include "src/developer/debug/zxdb/symbols/module_symbol_status.h" |
| #include "src/developer/debug/zxdb/symbols/module_symbols.h" |
| #include "src/lib/fxl/strings/string_printf.h" |
| |
| namespace zxdb { |
| |
| // When we want to download symbols for a build ID, we create a Download object. We then fire off |
| // requests to all symbol servers we know about asking whether they have the symbols we need. These |
| // requests are async, and the callbacks each own a shared_ptr to the Download object. If all the |
| // callbacks run and none of them are informed the request was successful, all of the shared_ptrs |
| // are dropped and the Download object is freed. The destructor of the Download object calls a |
| // callback that handles notifying the rest of the system of those results. |
| // |
| // If one of the callbacks does report that the symbols were found, a transaction to actually start |
| // the download is initiated, and its reply callback is again given a shared_ptr to the download. If |
| // we receive more notifications that other servers also have the symbol in the meantime, they are |
| // queued and will be tried as a fallback if the download fails. Again, once the download callback |
| // runs the shared_ptr is dropped, and when the Download object dies the destructor handles |
| // notifying the system. |
| class Download { |
| public: |
| explicit Download(const std::string& build_id, DebugSymbolFileType file_type, |
| SymbolServer::FetchCallback result_cb) |
| : build_id_(build_id), file_type_(file_type), result_cb_(std::move(result_cb)) {} |
| |
| ~Download() { Finish(); } |
| |
| bool active() { return !!result_cb_; } |
| |
| // Notify this download object that we have gotten the symbols if we're going to get them. |
| void Finish(); |
| |
| // Notify this Download object that one of the servers has the symbols available. |
| void Found(std::shared_ptr<Download> self, fit::callback<void(SymbolServer::FetchCallback)>); |
| |
| // Notify this Download object that a transaction failed. |
| void Error(std::shared_ptr<Download> self, const Err& err); |
| |
| // Add a symbol server to this download. |
| void AddServer(std::shared_ptr<Download> self, SymbolServer* server); |
| |
| private: |
| void RunCB(std::shared_ptr<Download> self, fit::callback<void(SymbolServer::FetchCallback)>& cb); |
| |
| std::string build_id_; |
| DebugSymbolFileType file_type_; |
| Err err_; |
| std::string path_; |
| SymbolServer::FetchCallback result_cb_; |
| std::vector<fit::callback<void(SymbolServer::FetchCallback)>> server_cbs_; |
| bool trying_ = false; |
| }; |
| |
| void Download::Finish() { |
| if (!result_cb_) |
| return; |
| |
| debug_ipc::MessageLoop::Current()->PostTask( |
| FROM_HERE, [result_cb = std::move(result_cb_), err = std::move(err_), |
| path = std::move(path_)]() mutable { result_cb(err, path); }); |
| |
| result_cb_ = nullptr; |
| } |
| |
| void Download::AddServer(std::shared_ptr<Download> self, SymbolServer* server) { |
| FXL_DCHECK(self.get() == this); |
| |
| if (!result_cb_) |
| return; |
| |
| server->CheckFetch(build_id_, file_type_, |
| [self](const Err& err, fit::callback<void(SymbolServer::FetchCallback)> cb) { |
| if (!cb) |
| self->Error(self, err); |
| else |
| self->Found(self, std::move(cb)); |
| }); |
| } |
| |
| void Download::Found(std::shared_ptr<Download> self, |
| fit::callback<void(SymbolServer::FetchCallback)> cb) { |
| FXL_DCHECK(self.get() == this); |
| |
| if (!result_cb_) |
| return; |
| |
| if (trying_) { |
| server_cbs_.push_back(std::move(cb)); |
| return; |
| } |
| |
| RunCB(self, cb); |
| } |
| |
| void Download::Error(std::shared_ptr<Download> self, const Err& err) { |
| FXL_DCHECK(self.get() == this); |
| |
| if (!result_cb_) |
| return; |
| |
| if (!err_.has_error()) { |
| err_ = err; |
| } else if (err.has_error()) { |
| err_ = Err("Multiple servers could not be reached."); |
| } |
| |
| if (!trying_ && !server_cbs_.empty()) { |
| RunCB(self, server_cbs_.back()); |
| server_cbs_.pop_back(); |
| } |
| } |
| |
| void Download::RunCB(std::shared_ptr<Download> self, |
| fit::callback<void(SymbolServer::FetchCallback)>& cb) { |
| FXL_DCHECK(!trying_); |
| trying_ = true; |
| |
| cb([self](const Err& err, const std::string& path) { |
| self->trying_ = false; |
| |
| if (path.empty()) { |
| self->Error(self, err); |
| } else { |
| self->err_ = err; |
| self->path_ = path; |
| self->Finish(); |
| } |
| }); |
| } |
| |
| SystemImpl::SystemImpl(Session* session) : System(session), symbols_(this), weak_factory_(this) { |
| // Create the default job and target. |
| AddNewJobContext(std::make_unique<JobContextImpl>(this, true)); |
| AddNewTarget(std::make_unique<TargetImpl>(this)); |
| |
| // Forward all messages from the symbol index to our observers. It's OK to bind |this| because the |
| // symbol index is owned by |this|. |
| symbols_.build_id_index().set_information_callback([this](const std::string& msg) { |
| for (auto& observer : observers()) |
| observer.OnSymbolIndexingInformation(msg); |
| }); |
| |
| // The system is the one holding the system symbols and is the one who will be updating the |
| // symbols once we get a symbol change, so the System will be listening to its own options. We |
| // don't use SystemSymbols because they live in the symbols library and we don't want it to have a |
| // client dependency. |
| settings_.AddObserver(ClientSettings::System::kDebugMode, this); |
| settings_.AddObserver(ClientSettings::System::kSymbolCache, this); |
| settings_.AddObserver(ClientSettings::System::kSymbolPaths, this); |
| settings_.AddObserver(ClientSettings::System::kSymbolRepoPaths, this); |
| settings_.AddObserver(ClientSettings::System::kSymbolServers, this); |
| |
| // Observe the session for filter matches and attach to any process koid that the system is not |
| // already attached to. |
| session->AddFilterObserver(this); |
| } |
| |
| SystemImpl::~SystemImpl() { |
| // Target destruction may depend on the symbol system. Ensure the targets get cleaned up first. |
| for (auto& target : targets_) { |
| // It's better if process destruction notifications are sent before target ones because the |
| // target owns the process. Because this class sends the target notifications, force the process |
| // destruction before doing anything. |
| target->ImplicitlyDetach(); |
| for (auto& observer : session()->target_observers()) |
| observer.WillDestroyTarget(target.get()); |
| } |
| |
| targets_.clear(); |
| |
| // Filters list may be iterated as we clean them up. Move its contents here first then let it |
| // drop so the dying objects are out of the system. |
| auto filters = std::move(filters_); |
| } |
| |
| ProcessImpl* SystemImpl::ProcessImplFromKoid(uint64_t koid) const { |
| for (const auto& target : targets_) { |
| ProcessImpl* process = target->process(); |
| if (process && process->GetKoid() == koid) |
| return process; |
| } |
| return nullptr; |
| } |
| |
| std::vector<TargetImpl*> SystemImpl::GetTargetImpls() const { |
| std::vector<TargetImpl*> result; |
| for (const auto& t : targets_) |
| result.push_back(t.get()); |
| return result; |
| } |
| |
| TargetImpl* SystemImpl::CreateNewTargetImpl(TargetImpl* clone) { |
| auto target = clone ? clone->Clone(this) : std::make_unique<TargetImpl>(this); |
| TargetImpl* to_return = target.get(); |
| AddNewTarget(std::move(target)); |
| return to_return; |
| } |
| |
| SystemSymbols* SystemImpl::GetSymbols() { return &symbols_; } |
| |
| std::vector<Target*> SystemImpl::GetTargets() const { |
| std::vector<Target*> result; |
| result.reserve(targets_.size()); |
| for (const auto& t : targets_) |
| result.push_back(t.get()); |
| return result; |
| } |
| |
| std::vector<JobContext*> SystemImpl::GetJobContexts() const { |
| std::vector<JobContext*> result; |
| result.reserve(job_contexts_.size()); |
| for (const auto& t : job_contexts_) |
| result.push_back(t.get()); |
| return result; |
| } |
| |
| std::vector<Breakpoint*> SystemImpl::GetBreakpoints() const { |
| std::vector<Breakpoint*> result; |
| result.reserve(breakpoints_.size()); |
| for (const auto& pair : breakpoints_) { |
| if (!pair.second->is_internal()) |
| result.push_back(pair.second.get()); |
| } |
| return result; |
| } |
| |
| std::vector<Filter*> SystemImpl::GetFilters() const { |
| std::vector<Filter*> result; |
| result.reserve(filters_.size()); |
| for (const auto& filter : filters_) { |
| result.push_back(filter.get()); |
| } |
| return result; |
| } |
| |
| std::vector<SymbolServer*> SystemImpl::GetSymbolServers() const { |
| std::vector<SymbolServer*> result; |
| result.reserve(symbol_servers_.size()); |
| for (const auto& item : symbol_servers_) { |
| result.push_back(item.get()); |
| } |
| return result; |
| } |
| |
| std::shared_ptr<Download> SystemImpl::GetDownload(std::string build_id, |
| DebugSymbolFileType file_type, bool quiet) { |
| if (auto existing = downloads_[{build_id, file_type}].lock()) { |
| return existing; |
| } |
| |
| DownloadStarted(); |
| |
| auto download = std::make_shared<Download>( |
| build_id, file_type, |
| [build_id, file_type, weak_this = weak_factory_.GetWeakPtr(), quiet]( |
| const Err& err, const std::string& path) { |
| if (!weak_this) { |
| return; |
| } |
| |
| if (!path.empty()) { |
| weak_this->download_success_count_++; |
| if (err.has_error()) { |
| // If we got a path but still had an error, something went wrong with the cache repo. |
| // Add the path manually. |
| weak_this->symbols_.build_id_index().AddOneFile(path); |
| } |
| |
| for (const auto& target : weak_this->targets_) { |
| if (auto process = target->process()) { |
| process->GetSymbols()->RetryLoadBuildID(build_id, file_type); |
| } |
| } |
| } else { |
| weak_this->download_fail_count_++; |
| |
| if (!quiet) { |
| weak_this->NotifyFailedToFindDebugSymbols(err, build_id, file_type); |
| } |
| } |
| |
| weak_this->DownloadFinished(); |
| }); |
| |
| downloads_[{build_id, file_type}] = download; |
| |
| if (servers_initializing_) { |
| suspended_downloads_.push_back(download); |
| } |
| |
| return download; |
| } |
| |
| void SystemImpl::DownloadStarted() { |
| if (download_count_ == 0) { |
| for (auto& observer : session()->download_observers()) { |
| observer.OnDownloadsStarted(); |
| } |
| } |
| |
| download_count_++; |
| } |
| |
| void SystemImpl::DownloadFinished() { |
| download_count_--; |
| |
| if (download_count_ == 0) { |
| for (auto& observer : session()->download_observers()) { |
| observer.OnDownloadsStopped(download_success_count_, download_fail_count_); |
| } |
| |
| download_success_count_ = download_fail_count_ = 0; |
| } |
| } |
| |
| void SystemImpl::RequestDownload(const std::string& build_id, DebugSymbolFileType file_type, |
| bool quiet) { |
| auto download = GetDownload(build_id, file_type, quiet); |
| |
| for (auto& server : symbol_servers_) { |
| if (server->state() != SymbolServer::State::kReady) { |
| continue; |
| } |
| |
| download->AddServer(download, server.get()); |
| } |
| } |
| |
| void SystemImpl::NotifyFailedToFindDebugSymbols(const Err& err, const std::string& build_id, |
| DebugSymbolFileType file_type) { |
| for (const auto& target : targets_) { |
| // Notify only those targets which are processes and which have attempted and failed to load |
| // symbols for this build ID previously. |
| auto process = target->process(); |
| if (!process) |
| continue; |
| |
| for (const auto& status : process->GetSymbols()->GetStatus()) { |
| if (status.build_id != build_id) { |
| continue; |
| } |
| |
| if (!err.has_error()) { |
| if (file_type == DebugSymbolFileType::kDebugInfo) { |
| process->OnSymbolLoadFailure(Err( |
| fxl::StringPrintf("Could not load symbols for \"%s\" because there was no mapping " |
| "for build ID \"%s\".", |
| status.name.c_str(), status.build_id.c_str()))); |
| } else { |
| process->OnSymbolLoadFailure( |
| Err(fxl::StringPrintf("Could not load binary for \"%s\" because there was no mapping " |
| "for build ID \"%s\".", |
| status.name.c_str(), status.build_id.c_str()))); |
| } |
| } else { |
| process->OnSymbolLoadFailure(err); |
| } |
| } |
| } |
| } |
| |
| void SystemImpl::OnSymbolServerBecomesReady(SymbolServer* server) { |
| for (const auto& target : targets_) { |
| auto process = target->process(); |
| if (!process) |
| continue; |
| |
| for (const auto& mod : process->GetSymbols()->GetStatus()) { |
| if (!mod.symbols || !mod.symbols->module_symbols()) { |
| auto download = GetDownload(mod.build_id, DebugSymbolFileType::kDebugInfo, true); |
| download->AddServer(download, server); |
| } else if (!mod.symbols->module_symbols()->HasBinary()) { |
| auto download = GetDownload(mod.build_id, DebugSymbolFileType::kBinary, true); |
| download->AddServer(download, server); |
| } |
| } |
| } |
| } |
| |
| Process* SystemImpl::ProcessFromKoid(uint64_t koid) const { return ProcessImplFromKoid(koid); } |
| |
| void SystemImpl::GetProcessTree(ProcessTreeCallback callback) { |
| session()->remote_api()->ProcessTree(debug_ipc::ProcessTreeRequest(), std::move(callback)); |
| } |
| |
| Target* SystemImpl::CreateNewTarget(Target* clone) { |
| return CreateNewTargetImpl(static_cast<TargetImpl*>(clone)); |
| } |
| |
| JobContext* SystemImpl::CreateNewJobContext() { |
| auto job_context = std::make_unique<JobContextImpl>(this, false); |
| JobContext* to_return = job_context.get(); |
| AddNewJobContext(std::move(job_context)); |
| return to_return; |
| } |
| |
| void SystemImpl::DeleteJobContext(JobContext* job_context) { |
| JobContextImpl* impl = static_cast<JobContextImpl*>(job_context); |
| auto found = std::find_if(job_contexts_.begin(), job_contexts_.end(), |
| [impl](auto& cur) { return impl == cur.get(); }); |
| if (found == job_contexts_.end()) { |
| FXL_NOTREACHED(); // Should always be found. |
| return; |
| } |
| |
| // Delete all filters that reference this job context. While it might be nice if the filter |
| // registered for a notification or used a weak pointer for the job, this would imply having a |
| // filter enabled/disabled state independent of the other settings which we don't have and don't |
| // currently need. Without a disabled state, clearing the job on the filter will make it apply to |
| // all jobs which the user does not want. |
| std::vector<Filter*> filters_to_remove; |
| for (auto& f : filters_) { |
| if (f->job() == job_context) |
| filters_to_remove.push_back(f.get()); |
| } |
| for (auto& f : filters_to_remove) |
| DeleteFilter(f); |
| |
| job_contexts_.erase(found); |
| } |
| |
| Breakpoint* SystemImpl::CreateNewBreakpoint() { |
| auto owning = std::make_unique<BreakpointImpl>(session(), false); |
| uint32_t id = owning->backend_id(); |
| Breakpoint* to_return = owning.get(); |
| |
| breakpoints_[id] = std::move(owning); |
| |
| // Notify observers (may mutate breakpoint list). |
| for (auto& observer : observers()) |
| observer.DidCreateBreakpoint(to_return); |
| |
| return to_return; |
| } |
| |
| Breakpoint* SystemImpl::CreateNewInternalBreakpoint() { |
| auto owning = std::make_unique<BreakpointImpl>(session(), true); |
| uint32_t id = owning->backend_id(); |
| Breakpoint* to_return = owning.get(); |
| |
| breakpoints_[id] = std::move(owning); |
| return to_return; |
| } |
| |
| void SystemImpl::DeleteBreakpoint(Breakpoint* breakpoint) { |
| BreakpointImpl* impl = static_cast<BreakpointImpl*>(breakpoint); |
| auto found = breakpoints_.find(impl->backend_id()); |
| if (found == breakpoints_.end()) { |
| // Should always have found the breakpoint. |
| FXL_NOTREACHED(); |
| return; |
| } |
| |
| // Only notify observers for non-internal breakpoints. |
| if (!found->second->is_internal()) { |
| for (auto& observer : observers()) |
| observer.WillDestroyBreakpoint(breakpoint); |
| } |
| breakpoints_.erase(found); |
| } |
| |
| Filter* SystemImpl::CreateNewFilter() { |
| Filter* to_return = filters_.emplace_back(std::make_unique<Filter>(session())).get(); |
| |
| // Notify observers (may mutate filter list). |
| for (auto& observer : observers()) |
| observer.DidCreateFilter(to_return); |
| |
| return to_return; |
| } |
| |
| void SystemImpl::DeleteFilter(Filter* filter) { |
| auto found = filters_.begin(); |
| for (; found != filters_.end(); ++found) { |
| if (found->get() == filter) { |
| break; |
| } |
| } |
| |
| if (found == filters_.end()) { |
| // Should always have found the filter. |
| FXL_NOTREACHED(); |
| return; |
| } |
| |
| for (auto& observer : observers()) |
| observer.WillDestroyFilter(filter); |
| |
| // Move this aside while we modify the list, then let it drop at the end of the function. That way |
| // the destructor doesn't see itself in the list of active filters when it emits |
| // WillDestroyFilter. |
| auto filter_ptr = std::move(*found); |
| filters_.erase(found); |
| } |
| |
| void SystemImpl::Pause(fit::callback<void()> on_paused) { |
| debug_ipc::PauseRequest request; |
| request.process_koid = 0; // 0 means all processes. |
| request.thread_koid = 0; // 0 means all threads. |
| session()->remote_api()->Pause( |
| request, [weak_system = weak_factory_.GetWeakPtr(), on_paused = std::move(on_paused)]( |
| const Err&, debug_ipc::PauseReply reply) mutable { |
| if (weak_system) { |
| // Save the newly paused thread metadata. This may need to be |
| // generalized if we add other messages that update thread metadata. |
| for (const auto& record : reply.threads) { |
| if (auto* process = weak_system->ProcessImplFromKoid(record.process_koid)) { |
| if (auto* thread = process->GetThreadImplFromKoid(record.thread_koid)) |
| thread->SetMetadata(record); |
| } |
| } |
| } |
| on_paused(); |
| }); |
| } |
| |
| void SystemImpl::Continue() { |
| // Tell each process to continue as it desires. |
| // |
| // It would be more efficient to tell the backend to resume all threads in all processes but the |
| // Thread client objects have state which needs to be updated (like the current stack) and the |
| // thread could have a controller that wants to continue in a specific way (like single-step or |
| // step in a range). |
| for (const auto& target : targets_) { |
| if (Process* process = target->GetProcess()) |
| process->Continue(); |
| } |
| } |
| |
| bool SystemImpl::HasDownload(const std::string& build_id) { |
| auto download = downloads_.find({build_id, DebugSymbolFileType::kDebugInfo}); |
| |
| if (download == downloads_.end()) { |
| download = downloads_.find({build_id, DebugSymbolFileType::kBinary}); |
| } |
| |
| if (download == downloads_.end()) { |
| return false; |
| } |
| |
| auto ptr = download->second.lock(); |
| return ptr && ptr->active(); |
| } |
| |
| std::shared_ptr<Download> SystemImpl::InjectDownloadForTesting(const std::string& build_id) { |
| return GetDownload(build_id, DebugSymbolFileType::kDebugInfo, true); |
| } |
| |
| void SystemImpl::DidConnect() { |
| // Force reload the symbol mappings after connection. This needs to be done for every connection |
| // since a new image could have been compiled and launched which will have a different build ID |
| // file. |
| symbols_.build_id_index().ClearCache(); |
| |
| // Implicitly attach a job to the root. If there was already an implicit job created (from a |
| // previous connection) re-use it since there will be settings on it about what processes to |
| // attach to that we want to preserve. |
| JobContextImpl* implicit_job = nullptr; |
| for (auto& job : job_contexts_) { |
| if (job->is_implicit_root()) { |
| implicit_job = job.get(); |
| break; |
| } |
| } |
| |
| if (!implicit_job) { |
| // No previous one, create a new implicit job. |
| auto new_job = std::make_unique<JobContextImpl>(this, true); |
| implicit_job = new_job.get(); |
| AddNewJobContext(std::move(new_job)); |
| } |
| implicit_job->AttachToSystemRoot([](fxl::WeakPtr<JobContext>, const Err&) {}); |
| } |
| |
| void SystemImpl::DidDisconnect() { |
| for (auto& target : targets_) |
| target->ImplicitlyDetach(); |
| for (auto& job : job_contexts_) |
| job->ImplicitlyDetach(); |
| } |
| |
| BreakpointImpl* SystemImpl::BreakpointImplForId(uint32_t id) { |
| auto found = breakpoints_.find(id); |
| if (found == breakpoints_.end()) |
| return nullptr; |
| return found->second.get(); |
| } |
| |
| void SystemImpl::AddNewTarget(std::unique_ptr<TargetImpl> target) { |
| Target* for_observers = target.get(); |
| |
| targets_.push_back(std::move(target)); |
| for (auto& observer : session()->target_observers()) |
| observer.DidCreateTarget(for_observers); |
| } |
| |
| void SystemImpl::AddNewJobContext(std::unique_ptr<JobContextImpl> job_context) { |
| JobContext* for_observers = job_context.get(); |
| |
| job_contexts_.push_back(std::move(job_context)); |
| for (auto& observer : observers()) |
| observer.DidCreateJobContext(for_observers); |
| } |
| |
| void SystemImpl::OnSettingChanged(const SettingStore& store, const std::string& setting_name) { |
| if (setting_name == ClientSettings::System::kSymbolPaths) { |
| auto paths = store.GetList(ClientSettings::System::kSymbolPaths); |
| BuildIDIndex& build_id_index = GetSymbols()->build_id_index(); |
| for (const std::string& path : paths) { |
| if (StringEndsWith(path, ".txt")) { |
| build_id_index.AddBuildIDMappingFile(path); |
| } else { |
| build_id_index.AddSymbolSource(path); |
| } |
| } |
| } else if (setting_name == ClientSettings::System::kSymbolRepoPaths) { |
| auto paths = store.GetList(ClientSettings::System::kSymbolPaths); |
| BuildIDIndex& build_id_index = GetSymbols()->build_id_index(); |
| for (const std::string& path : paths) { |
| build_id_index.AddRepoSymbolSource(path); |
| } |
| } else if (setting_name == ClientSettings::System::kSymbolCache) { |
| auto path = store.GetString(setting_name); |
| |
| if (!path.empty()) { |
| std::error_code ec; |
| std::filesystem::create_directory(std::filesystem::path(path) / ".build-id", ec); |
| GetSymbols()->build_id_index().AddSymbolSource(path); |
| } |
| } else if (setting_name == ClientSettings::System::kSymbolServers) { |
| auto urls = store.GetList(setting_name); |
| std::set<std::string> existing; |
| |
| for (const auto& symbol_server : symbol_servers_) { |
| existing.insert(symbol_server->name()); |
| } |
| |
| for (const auto& url : urls) { |
| if (existing.find(url) == existing.end()) { |
| symbol_servers_.push_back(SymbolServer::FromURL(session(), url)); |
| AddSymbolServer(symbol_servers_.back().get()); |
| } |
| } |
| } else if (setting_name == ClientSettings::System::kDebugMode) { |
| debug_ipc::SetDebugMode(store.GetBool(setting_name)); |
| } else { |
| FXL_LOG(WARNING) << "Unhandled setting change: " << setting_name; |
| } |
| } |
| |
| void SystemImpl::InjectSymbolServerForTesting(std::unique_ptr<SymbolServer> server) { |
| symbol_servers_.push_back(std::move(server)); |
| AddSymbolServer(symbol_servers_.back().get()); |
| } |
| |
| void SystemImpl::OnFilterMatches(JobContext* job, const std::vector<uint64_t>& matched_pids) { |
| // Go over the targets and see if we find a valid one for each pid. |
| for (uint64_t matched_pid : matched_pids) { |
| bool found = false; |
| for (auto& target : targets_) { |
| Process* process = target->GetProcess(); |
| if (process && process->GetKoid() == matched_pid) { |
| found = true; |
| break; |
| } |
| } |
| |
| // If we found an already attached process, we don't care about this match. |
| if (found) |
| continue; |
| |
| AttachToProcess(matched_pid, [matched_pid](fxl::WeakPtr<Target> target, const Err& err) { |
| if (err.has_error()) { |
| FXL_LOG(ERROR) << "Could not attach to process " << matched_pid; |
| return; |
| } |
| }); |
| } |
| } |
| |
| void SystemImpl::AttachToProcess(uint64_t pid, Target::Callback callback) { |
| // See if there is a target that is not attached. |
| Target* open_slot = nullptr; |
| for (auto& target : targets_) { |
| if (target->GetState() == zxdb::Target::State::kNone) { |
| open_slot = target.get(); |
| break; |
| } |
| } |
| |
| // If no slot was found, we create a new target. |
| if (!open_slot) |
| open_slot = CreateNewTarget(nullptr); |
| |
| open_slot->Attach(pid, std::move(callback)); |
| } |
| |
| void SystemImpl::ServerStartedInitializing() { servers_initializing_++; } |
| |
| void SystemImpl::ServerFinishedInitializing() { |
| FXL_DCHECK(servers_initializing_ > 0); |
| |
| if (!--servers_initializing_) { |
| suspended_downloads_.clear(); |
| } |
| } |
| |
| void SystemImpl::AddSymbolServer(SymbolServer* server) { |
| for (auto& observer : observers()) { |
| observer.DidCreateSymbolServer(server); |
| } |
| |
| bool initializing = false; |
| |
| if (server->state() == SymbolServer::State::kInitializing || |
| server->state() == SymbolServer::State::kBusy) { |
| initializing = true; |
| ServerStartedInitializing(); |
| } |
| |
| server->set_state_change_callback([weak_this = weak_factory_.GetWeakPtr(), initializing]( |
| SymbolServer* server, SymbolServer::State state) mutable { |
| if (!weak_this) { |
| return; |
| } |
| |
| if (state == SymbolServer::State::kReady) |
| weak_this->OnSymbolServerBecomesReady(server); |
| |
| if (initializing && state != SymbolServer::State::kBusy && |
| state != SymbolServer::State::kInitializing) { |
| initializing = false; |
| weak_this->ServerFinishedInitializing(); |
| } |
| }); |
| |
| if (server->state() == SymbolServer::State::kReady) { |
| OnSymbolServerBecomesReady(server); |
| } |
| } |
| |
| } // namespace zxdb |