// Copyright 2017 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 <inttypes.h>
#include <stdint.h>
#include <stdio.h>

#include <zircon/syscalls.h>

#include "inspector/inspector.h"
#include "utils-impl.h"

__EXPORT zx_status_t inspector_read_general_regs(zx_handle_t thread,
                                                 zx_thread_state_general_regs_t* regs) {
  auto status = zx_thread_read_state(thread, ZX_THREAD_STATE_GENERAL_REGS, regs, sizeof(*regs));
  if (status < 0) {
    print_zx_error("unable to access general regs", status);
    return status;
  }
  return ZX_OK;
}

#if defined(__x86_64__)

__EXPORT void inspector_print_general_regs(FILE* f, const zx_thread_state_general_regs_t* regs,
                                           const inspector_excp_data_t* excp_data) {
  fprintf(f, " CS:  %#18llx RIP: %#18" PRIx64 " EFL: %#18" PRIx64, 0ull, regs->rip, regs->rflags);
  if (excp_data) {
    fprintf(f, " CR2: %#18" PRIx64, excp_data->cr2);
  }
  fprintf(f, "\n");
  fprintf(f, " RAX: %#18" PRIx64 " RBX: %#18" PRIx64 " RCX: %#18" PRIx64 " RDX: %#18" PRIx64 "\n",
          regs->rax, regs->rbx, regs->rcx, regs->rdx);
  fprintf(f, " RSI: %#18" PRIx64 " RDI: %#18" PRIx64 " RBP: %#18" PRIx64 " RSP: %#18" PRIx64 "\n",
          regs->rsi, regs->rdi, regs->rbp, regs->rsp);
  fprintf(f, "  R8: %#18" PRIx64 "  R9: %#18" PRIx64 " R10: %#18" PRIx64 " R11: %#18" PRIx64 "\n",
          regs->r8, regs->r9, regs->r10, regs->r11);
  fprintf(f, " R12: %#18" PRIx64 " R13: %#18" PRIx64 " R14: %#18" PRIx64 " R15: %#18" PRIx64 "\n",
          regs->r12, regs->r13, regs->r14, regs->r15);
  fprintf(f, " fs.base: %#18" PRIx64 " gs.base: %#18" PRIx64 "\n", regs->fs_base, regs->gs_base);
  if (excp_data) {
    // errc value is 17 on purpose, errc is 4 characters
    fprintf(f, " errc: %#17" PRIx64 "\n", excp_data->err_code);
  }
}

#elif defined(__aarch64__)

__EXPORT void inspector_print_general_regs(FILE* f, const zx_thread_state_general_regs_t* regs,
                                           const inspector_excp_data_t* excp_data) {
  fprintf(f, " x0  %#18" PRIx64 " x1  %#18" PRIx64 " x2  %#18" PRIx64 " x3  %#18" PRIx64 "\n",
          regs->r[0], regs->r[1], regs->r[2], regs->r[3]);
  fprintf(f, " x4  %#18" PRIx64 " x5  %#18" PRIx64 " x6  %#18" PRIx64 " x7  %#18" PRIx64 "\n",
          regs->r[4], regs->r[5], regs->r[6], regs->r[7]);
  fprintf(f, " x8  %#18" PRIx64 " x9  %#18" PRIx64 " x10 %#18" PRIx64 " x11 %#18" PRIx64 "\n",
          regs->r[8], regs->r[9], regs->r[10], regs->r[11]);
  fprintf(f, " x12 %#18" PRIx64 " x13 %#18" PRIx64 " x14 %#18" PRIx64 " x15 %#18" PRIx64 "\n",
          regs->r[12], regs->r[13], regs->r[14], regs->r[15]);
  fprintf(f, " x16 %#18" PRIx64 " x17 %#18" PRIx64 " x18 %#18" PRIx64 " x19 %#18" PRIx64 "\n",
          regs->r[16], regs->r[17], regs->r[18], regs->r[19]);
  fprintf(f, " x20 %#18" PRIx64 " x21 %#18" PRIx64 " x22 %#18" PRIx64 " x23 %#18" PRIx64 "\n",
          regs->r[20], regs->r[21], regs->r[22], regs->r[23]);
  fprintf(f, " x24 %#18" PRIx64 " x25 %#18" PRIx64 " x26 %#18" PRIx64 " x27 %#18" PRIx64 "\n",
          regs->r[24], regs->r[25], regs->r[26], regs->r[27]);
  fprintf(f, " x28 %#18" PRIx64 " x29 %#18" PRIx64 " lr  %#18" PRIx64 " sp  %#18" PRIx64 "\n",
          regs->r[28], regs->r[29], regs->lr, regs->sp);
  fprintf(f, " pc  %#18" PRIx64 " psr %#18" PRIx64 "\n", regs->pc, regs->cpsr);
}

#else  // unsupported arch

__EXPORT void inspector_print_general_regs(const zx_thread_state_general_regs_t* regs,
                                           const inspector_excp_data_t* excp_data) {
  printf("unsupported architecture\n");
}

#endif