Smart cont try fetch config (#1717)

* SmartContract: try fetch config from c7 if it's not set

* feat: Added tvm_emulator_emulate_run_method clone with logs available

* free

* fix cast warning

* bf

---------

Co-authored-by: Marat S <[email protected]>
Co-authored-by: Yma Het <[email protected]>

Author: 3 people

crypto/smc-envelope/SmartContract.cpp

@@ -191,6 +191,42 @@ td::Ref<vm::Tuple> prepare_vm_c7(SmartContract::Args args, td::Ref<vm::Cell> cod
   return vm::make_tuple_ref(std::move(tuple_ref));
 }
 
+std::shared_ptr<const block::Config> try_fetch_config_from_c7(td::Ref<vm::Tuple> c7) {
+  if (c7.is_null() || c7->size() < 1) {
+    return nullptr;
+  }
+  auto c7_tuple = c7->at(0).as_tuple();
+  if (c7_tuple.is_null() || c7_tuple->size() < 10) {
+    return nullptr;
+  }
+  auto config_cell = c7_tuple->at(9).as_cell();
+  if (config_cell.is_null()) {
+    return nullptr;
+  }
+  auto config_dict = std::make_unique<vm::Dictionary>(config_cell, 32);
+  auto config_addr_cell = config_dict->lookup_ref(td::BitArray<32>::zero());
+  ton::StdSmcAddress config_addr;
+  if (config_addr_cell.is_null()) {
+    config_addr = ton::StdSmcAddress::zero();
+  } else {
+    auto config_addr_cs = vm::load_cell_slice(std::move(config_addr_cell));
+    if (config_addr_cs.size() != 0x100) {
+      LOG(WARNING) << "Config parameter 0 with config address has wrong size";
+      config_addr = ton::StdSmcAddress::zero();
+    } else {
+      config_addr_cs.fetch_bits_to(config_addr);
+    }
+  }
+  auto global_config = block::Config(config_cell, std::move(config_addr), 
+      block::Config::needWorkchainInfo | block::Config::needSpecialSmc | block::Config::needCapabilities);
+  auto unpack_res = global_config.unpack();
+  if (unpack_res.is_error()) {
+    LOG(ERROR) << "Failed to unpack config: " << unpack_res.error();
+    return nullptr;
+  }
+  return std::make_shared<block::Config>(std::move(global_config));
+}
+
 SmartContract::Answer run_smartcont(SmartContract::State state, td::Ref<vm::Stack> stack, td::Ref<vm::Tuple> c7,
                                     vm::GasLimits gas, bool ignore_chksig, td::Ref<vm::Cell> libraries,
                                     int vm_log_verbosity, bool debug_enabled,
@@ -314,6 +350,9 @@ td::Ref<vm::Cell> SmartContract::get_init_state() const {
 }
 
 SmartContract::Answer SmartContract::run_method(Args args) {
+  if (args.c7 && !args.config) {
+    args.config = try_fetch_config_from_c7(args.c7.value());
+  }
   if (!args.c7) {
     args.c7 = prepare_vm_c7(args, state_.code);
   }
@@ -335,6 +374,9 @@ SmartContract::Answer SmartContract::run_method(Args args) {
 }
 
 SmartContract::Answer SmartContract::run_get_method(Args args) const {
+  if (args.c7 && !args.config) {
+    args.config = try_fetch_config_from_c7(args.c7.value());
+  }
   if (!args.c7) {
     args.c7 = prepare_vm_c7(args, state_.code);
   }

emulator/emulator-extern.cpp

@@ -639,10 +639,16 @@ const char *tvm_emulator_run_get_method(void *tvm_emulator, int method_id, const
   return strdup(jb.string_builder().as_cslice().c_str());
 }
 
-const char *tvm_emulator_emulate_run_method(uint32_t len, const char *params_boc, int64_t gas_limit) {
+struct TvmEulatorEmulateRunMethodResponse
+{
+  const char *response;
+  const char *log;
+};
+
+TvmEulatorEmulateRunMethodResponse emulate_run_method(uint32_t len, const char *params_boc, int64_t gas_limit) {
   auto params_cell = vm::std_boc_deserialize(td::Slice(params_boc, len));
   if (params_cell.is_error()) {
-    return nullptr;
+    return { nullptr, nullptr };
   }
   auto params_cs = vm::load_cell_slice(params_cell.move_as_ok());
   auto code = params_cs.fetch_ref();
@@ -657,12 +663,12 @@ const char *tvm_emulator_emulate_run_method(uint32_t len, const char *params_boc
 
   td::Ref<vm::Stack> stack;
   if (!vm::Stack::deserialize_to(stack_cs, stack)) {
-    return nullptr;
+    return { nullptr, nullptr };
   }
 
   td::Ref<vm::Stack> c7;
   if (!vm::Stack::deserialize_to(c7_cs, c7)) {
-    return nullptr;
+    return { nullptr, nullptr };
   }
 
   auto emulator = new emulator::TvmEmulator(code, data);
@@ -677,7 +683,7 @@ const char *tvm_emulator_emulate_run_method(uint32_t len, const char *params_boc
 
   vm::CellBuilder stack_cb;
   if (!result.stack->serialize(stack_cb)) {
-    return nullptr;
+    return { nullptr, nullptr };
   }
 
   vm::CellBuilder cb;
@@ -687,7 +693,7 @@ const char *tvm_emulator_emulate_run_method(uint32_t len, const char *params_boc
 
   auto ser = vm::std_boc_serialize(cb.finalize());
   if (!ser.is_ok()) {
-    return nullptr;
+    return { nullptr, nullptr };
   }
   auto sok = ser.move_as_ok();
 
@@ -696,7 +702,24 @@ const char *tvm_emulator_emulate_run_method(uint32_t len, const char *params_boc
   memcpy(rn, &sz, 4);
   memcpy(rn+4, sok.data(), sz);
 
-  return rn;
+  return { rn, strdup(result.vm_log.data()) };
+}
+
+const char *tvm_emulator_emulate_run_method(uint32_t len, const char *params_boc, int64_t gas_limit) {
+  auto result = emulate_run_method(len, params_boc, gas_limit);
+  return result.response;
+}
+
+void *tvm_emulator_emulate_run_method_detailed(uint32_t len, const char *params_boc, int64_t gas_limit) {
+  auto result = emulate_run_method(len, params_boc, gas_limit);
+  return new TvmEulatorEmulateRunMethodResponse(result);
+}
+
+void run_method_detailed_result_destroy(void *detailed_result) {
+  auto result = static_cast<TvmEulatorEmulateRunMethodResponse *>(detailed_result);
+  free(const_cast<char*>(result->response));
+  free(const_cast<char*>(result->log));
+  delete result;
 }
 
 const char *tvm_emulator_send_external_message(void *tvm_emulator, const char *message_body_boc) {
@@ -773,6 +796,12 @@ void emulator_config_destroy(void *config) {
   delete static_cast<block::Config *>(config);
 }
 
+void string_destroy(const char *str) {
+  if (str != nullptr) {
+    free(const_cast<char *>(str));
+  }
+}
+
 const char* emulator_version() {
   auto version_json = td::JsonBuilder();
   auto obj = version_json.enter_object();

emulator/emulator-extern.h

@@ -254,6 +254,27 @@ EMULATOR_EXPORT const char *tvm_emulator_run_get_method(void *tvm_emulator, int
  */
 EMULATOR_EXPORT const char *tvm_emulator_emulate_run_method(uint32_t len, const char *params_boc, int64_t gas_limit);
 
+/**
+ * @brief Optimized version of "run get method" with all passed parameters in a single call. Also returns log.
+ * @param len Length of params_boc buffer
+ * @param params_boc BoC serialized parameters, scheme: request$_ code:^Cell data:^Cell stack:^VmStack params:^[c7:^VmStack libs:^Cell] method_id:(## 32)
+ * @param gas_limit Gas limit
+ * @return Pointer to struct with two fields:
+ * - response: Char* with first 4 bytes defining length, and the rest BoC serialized result
+ *         Scheme: result$_ exit_code:(## 32) gas_used:(## 32) stack:^VmStack
+ * - log: Char* with VM log string
+ */
+EMULATOR_EXPORT void *tvm_emulator_emulate_run_method_detailed(uint32_t len, const char *params_boc, int64_t gas_limit);
+
+/**
+ * @brief Destroy detailed result of "tvm_emulator_emulate_run_method_detailed"
+ * @param detailed_result Pointer to detailed result struct returned by "tvm_emulator_emulate_run_method_detailed"
+ * 
+ * Caller should not use string_destroy() for fields of this struct,
+ * as they are already freed in this function.
+ */
+EMULATOR_EXPORT void run_method_detailed_result_destroy(void *detailed_result);
+
 /**
  * @brief Send external message
  * @param tvm_emulator Pointer to TVM emulator
@@ -315,6 +336,15 @@ EMULATOR_EXPORT void tvm_emulator_destroy(void *tvm_emulator);
  */
 EMULATOR_EXPORT void emulator_config_destroy(void *config);
 
+/**
+ * @brief Destroy string created by emulator library
+ * @param string Pointer to string to destroy
+ * 
+ * This function should be used to free strings returned by emulator library functions.
+ * It is not safe to use caller's free() on them, as they may have been allocated using a different allocator.
+ */
+EMULATOR_EXPORT void string_destroy(const char *string);
+
 /**
  * @brief Get git commit hash and date of the library
  */

emulator/emulator_export_list

@@ -27,4 +27,7 @@ _tvm_emulator_send_external_message
 _tvm_emulator_send_internal_message
 _tvm_emulator_destroy
 _tvm_emulator_emulate_run_method
+_tvm_emulator_emulate_run_method_detailed
+_run_method_detailed_result_destroy
+_string_destroy
 _emulator_version