src/developer/debug/debug_agent/mock_debug_agent_harness.cc

// Copyright 2020 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/debug_agent/mock_debug_agent_harness.h"

#include "src/developer/debug/debug_agent/mock_process.h"
#include "src/developer/debug/ipc/protocol.h"

namespace debug_agent {

MockDebugAgentHarness::MockDebugAgentHarness(std::unique_ptr<MockSystemInterface> system_interface)
    : system_interface_(system_interface.get()), agent_(std::move(system_interface)) {
  auto stream_backend = std::make_unique<MockStreamBackend>();
  stream_backend_ = stream_backend.get();
  agent_.Connect(std::move(stream_backend));

  // Make sure DebugAgent is initialized with the current protocol version.
  debug_ipc::HelloRequest request;
  request.version = debug_ipc::kCurrentProtocolVersion;

  debug_ipc::HelloReply reply;
  agent_.OnHello(request, &reply);
}

MockProcess* MockDebugAgentHarness::AddProcess(zx_koid_t process_koid) {
  auto owning_process = std::make_unique<MockProcess>(debug_agent(), process_koid);
  MockProcess* result = owning_process.get();

  debug_agent()->InjectProcessForTest(std::move(owning_process));
  return result;
}

debug::Status MockDebugAgentHarness::AddOrChangeBreakpoint(uint32_t breakpoint_id,
                                                           zx_koid_t process_koid, uint64_t address,
                                                           debug_ipc::BreakpointType type) {
  debug_ipc::ProcessBreakpointSettings location;
  location.id.process = process_koid;
  location.address = address;

  debug_ipc::AddOrChangeBreakpointRequest request;
  request.breakpoint.id = breakpoint_id;
  request.breakpoint.type = type;
  request.breakpoint.name = "Injected breakpoint";
  request.breakpoint.locations.push_back(location);

  debug_ipc::AddOrChangeBreakpointReply reply;
  debug_agent()->OnAddOrChangeBreakpoint(request, &reply);
  return reply.status;
}

debug::Status MockDebugAgentHarness::AddOrChangeBreakpoint(uint32_t breakpoint_id,
                                                           zx_koid_t process_koid,
                                                           zx_koid_t thread_koid,
                                                           const debug::AddressRange& range,
                                                           debug_ipc::BreakpointType type) {
  debug_ipc::ProcessBreakpointSettings location;
  location.id = {.process = process_koid, .thread = thread_koid};
  location.address_range = range;

  debug_ipc::AddOrChangeBreakpointRequest request;
  request.breakpoint.id = breakpoint_id;
  request.breakpoint.type = type;
  request.breakpoint.name = "Injected breakpoint";
  request.breakpoint.locations.push_back(location);

  debug_ipc::AddOrChangeBreakpointReply reply;
  debug_agent()->OnAddOrChangeBreakpoint(request, &reply);
  return reply.status;
}

void MockDebugAgentHarness::Pause(zx_koid_t process_koid, zx_koid_t thread_koid) {
  debug_ipc::PauseRequest request;
  if (process_koid != 0)
    request.ids.push_back({.process = process_koid, .thread = thread_koid});

  debug_ipc::PauseReply reply;
  debug_agent()->OnPause(request, &reply);
}

void MockDebugAgentHarness::Resume(debug_ipc::ResumeRequest::How how, zx_koid_t process_koid,
                                   zx_koid_t thread_koid) {
  debug_ipc::ResumeRequest request;
  request.how = how;
  if (process_koid)
    request.ids.push_back({.process = process_koid, .thread = thread_koid});

  debug_ipc::ResumeReply reply;
  debug_agent()->OnResume(request, &reply);
}

}  // namespace debug_agent