// Copyright 2019 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/expr/eval_dwarf_expr.h"

#include "src/developer/debug/shared/message_loop.h"
#include "src/developer/debug/zxdb/expr/eval_context.h"
#include "src/developer/debug/zxdb/expr/resolve_ptr_ref.h"
#include "src/lib/fxl/strings/string_printf.h"

namespace zxdb {

void DwarfExprToValue(UnitSymbolFactory symbol_factory,
                      const fxl::RefPtr<EvalContext>& eval_context,
                      const SymbolContext& symbol_context, DwarfExpr expr, fxl::RefPtr<Type> type,
                      EvalCallback cb) {
  auto evaluator = fxl::MakeRefCounted<AsyncDwarfExprEvalValue>(
      std::move(symbol_factory), eval_context->GetDataProvider(), symbol_context, eval_context,
      std::move(type), std::move(cb));
  evaluator->Eval(std::move(expr));
}

void DwarfExprEvalToValue(const fxl::RefPtr<EvalContext>& context, DwarfExprEval& eval,
                          fxl::RefPtr<Type> type, EvalCallback cb) {
  if (eval.GetResultType() == DwarfExprEval::ResultType::kValue) {
    // Get the concrete type since we need the byte size. But don't use this to actually construct
    // the variable since it will strip "const" and stuff that the user will expect to see.
    fxl::RefPtr<Type> concrete_type = context->GetConcreteType(type.get());

    // The DWARF expression produced the exact value (it's not in memory). The expression may (but
    // usually won't) be annotated with its own type which should match the size of the type we
    // expect here. We don't bother validating that for now and instead just copy out the expected
    // number of bytes.
    uint32_t type_size = concrete_type->byte_size();
    if (type_size > sizeof(DwarfStackEntry::UnsignedType)) {
      return cb(
          Err(fxl::StringPrintf("Result size insufficient for type of size %u. "
                                "Please file a bug with a repro case.",
                                type_size)));
    }

    // When the result was read directly from a register or is known to be constant, preserve that
    // so the user can potentially write to it (or give a good error message about writing to it).
    ExprValueSource source(ExprValueSource::Type::kTemporary);
    if (eval.current_register_id() != debug::RegisterID::kUnknown)
      source = ExprValueSource(eval.current_register_id());
    else if (eval.result_is_constant())
      source = ExprValueSource(ExprValueSource::Type::kConstant);

    // Assuming little-endian and that the types in the DwarfExprEval are in a union, we can
    // copy the bytes out using the unsigned value without type checking.
    DwarfStackEntry result = eval.GetResult();
    auto result_value = result.unsigned_value();

    std::vector<uint8_t> data;
    data.resize(type_size);
    memcpy(data.data(), &result_value, type_size);
    cb(ExprValue(type, std::move(data), source));
  } else if (eval.GetResultType() == DwarfExprEval::ResultType::kData) {
    // The DWARF result is a block of data.
    //
    // Here we assume the data size is correct. If it doesn't match the type, that should be caught
    // later when it's interpreted.
    //
    // TODO(bug 39630) we have no source locations for this case.
    cb(ExprValue(type, eval.TakeResultData(), ExprValueSource(ExprValueSource::Type::kComposite)));
  } else {
    // The DWARF result is a pointer to the value.
    DwarfStackEntry result = eval.GetResult();
    if (!result.TreatAsUnsigned())
      return cb(Err("DWARF expression produced an unexpected type."));
    ResolvePointer(context, result.unsigned_value(), type,
                   [cb = std::move(cb)](ErrOrValue value) mutable { cb(std::move(value)); });
  }
}

void AsyncDwarfExprEval::Eval(DwarfExpr expr) {
  dwarf_eval_.Eval(std::move(expr), [this_ref = RefPtrTo(this)](DwarfExprEval*, const Err& err) {
    this_ref->dwarf_callback_(this_ref->dwarf_eval_, err);

    // Prevent the DwarfExprEval from getting reentrantly deleted from within its
    // own callback by posting a reference back to the message loop.
    debug::MessageLoop::Current()->PostTask(FROM_HERE, [this_ref = std::move(this_ref)]() {});
  });
}

// The callback passed to the base class can capture |this| because we're the same object.
AsyncDwarfExprEvalValue::AsyncDwarfExprEvalValue(UnitSymbolFactory symbol_factory,
                                                 fxl::RefPtr<SymbolDataProvider> data_provider,
                                                 const SymbolContext& expr_symbol_context,
                                                 const fxl::RefPtr<EvalContext>& context,
                                                 fxl::RefPtr<Type> type, EvalCallback cb)
    : AsyncDwarfExprEval(std::move(symbol_factory), std::move(data_provider), expr_symbol_context,
                         [this](DwarfExprEval&, const Err& err) { OnEvalComplete(err); }),
      context_(context),
      type_(std::move(type)),
      value_callback_(std::move(cb)) {}

void AsyncDwarfExprEvalValue::OnEvalComplete(const Err& err) {
  if (err.has_error())
    return value_callback_(err);
  DwarfExprEvalToValue(context_, dwarf_eval(), type_, std::move(value_callback_));
}

}  // namespace zxdb