/*
 * Copyright (C) 2024 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "berberis/base/logging.h"
#include "berberis/guest_state/get_cpu_state_opaque.h"
#include "berberis/guest_state/guest_state_arch.h"
#include "berberis/guest_state/guest_state_opaque.h"
#include "native_bridge_support/guest_state_accessor/accessor.h"

namespace berberis {

extern "C" __attribute__((visibility("default"))) int LoadGuestStateRegisters(
    const void* guest_state_data,
    size_t guest_state_data_size,
    NativeBridgeGuestRegs* guest_regs) {
  if (guest_state_data_size < sizeof(ThreadState)) {
    ALOGE("The guest state data size is invalid: %zu", guest_state_data_size);
    return NATIVE_BRIDGE_GUEST_STATE_ACCESSOR_ERROR_INVALID_STATE;
  }
  guest_regs->guest_arch = NATIVE_BRIDGE_ARCH_ARM64;
  return GetCpuState(guest_regs, &(static_cast<const ThreadState*>(guest_state_data))->cpu);
}

int GetCpuState(NativeBridgeGuestRegs* guest_regs, const CPUState* state) {
  if (guest_regs->guest_arch != NATIVE_BRIDGE_ARCH_ARM64) {
    ALOGE("The guest architecture is unmatched: %lu", guest_regs->guest_arch);
    return NATIVE_BRIDGE_GUEST_STATE_ACCESSOR_ERROR_UNSUPPORTED_ARCH;
  }
  memcpy(&guest_regs->regs_arm64.x, &state->x, sizeof(guest_regs->regs_arm64.x));
  memcpy(&guest_regs->regs_arm64.sp, &state->sp, sizeof(guest_regs->regs_arm64.sp));
  memcpy(&guest_regs->regs_arm64.ip, &state->insn_addr, sizeof(guest_regs->regs_arm64.ip));
  memcpy(&guest_regs->regs_arm64.v, &state->v, sizeof(guest_regs->regs_arm64.v));
  return 0;
}
}  // namespace berberis