1 files changed, 92 insertions, 0 deletions
diff --git a/src/blockchain_db/asset_db.cpp b/src/blockchain_db/asset_db.cpp
index d80b61028..800ea78fb 100644
--- a/src/blockchain_db/asset_db.cpp
+++ b/src/blockchain_db/asset_db.cpp
@@ -3,6 +3,8 @@
 #include <algorithm>
 #include <cstring>
 
+#include "ringct/rctOps.h"
+
 namespace cryptonote
 {
 namespace assets
@@ -92,5 +94,95 @@ bool apply_block_extensions_to_db(BlockchainDB& db,
   asset_ids = std::move(candidate_ids);
   return true;
 }
+
+bool verify_asset_ownership_against_db(const BlockchainDB& db,
+  const asset_ownership_proof& proof, network_type expected_network,
+  const crypto::hash& carrier_prefix_hash, std::string* error)
+{
+  if (db.has_asset_key_image(proof.key_image))
+    return fail(error, "asset key image is already spent");
+
+  for (const asset_ring_member& member : proof.ring)
+  {
+    asset_output_data_t stored{};
+    if (!db.get_asset_output(member.output_id, stored))
+      return fail(error, "asset ownership ring references an unknown output");
+    if (stored.asset_id != member.asset_id || member.asset_id != proof.asset_id)
+      return fail(error, "asset ownership ring member has the wrong asset id");
+    if (!rct::equalKeys(stored.destination, member.public_output.dest))
+      return fail(error, "asset ownership ring destination does not match consensus state");
+    if (!rct::equalKeys(stored.commitment, member.public_output.mask))
+      return fail(error, "asset ownership ring commitment does not match consensus state");
+  }
+  return verify_asset_ownership_proof(
+    proof, expected_network, carrier_prefix_hash, error);
+}
+
+bool apply_asset_transaction_to_db(BlockchainDB& db,
+  const asset_transaction_payload& payload, network_type expected_network,
+  const crypto::hash& expected_carrier_prefix_hash, uint64_t height,
+  std::vector<crypto::hash>& output_ids, std::string* error)
+{
+  asset_registry registry;
+  if (!load_registry_from_db(db, expected_network, registry, error)
+      || !verify_asset_transaction_payload(payload, registry.known_assets(),
+           expected_network, expected_carrier_prefix_hash, error))
+    return false;
+  for (const asset_ownership_proof& proof : payload.ownership_proofs)
+    if (!verify_asset_ownership_against_db(db, proof, expected_network,
+          expected_carrier_prefix_hash, error))
+      return false;
+
+  crypto::hash issued_id{};
+  std::vector<uint8_t> encoded_issuance;
+  if (payload.issuance)
+  {
+    if (!registry.apply_issuance(*payload.issuance, height, issued_id, error)
+        || !encode_issuance_payload(*payload.issuance, encoded_issuance, error))
+      return false;
+  }
+
+  std::vector<crypto::hash> candidate_ids;
+  uint32_t global_output_index = 0;
+  for (size_t group = 0; group < payload.balances.size(); ++group)
+  {
+    const confidential_asset_balance& balance = payload.balances[group];
+    for (size_t index = 0; index < balance.outputs.size(); ++index)
+    {
+      confidential_asset_output output{
+        payload.output_destinations[group][index], balance.outputs[index]};
+      crypto::hash output_id{};
+      if (!derive_asset_output_id(expected_network, expected_carrier_prefix_hash,
+            balance.asset_id, global_output_index++, output, output_id, error))
+        return false;
+      asset_output_data_t existing{};
+      if (db.get_asset_output(output_id, existing))
+        return fail(error, "asset output identity already exists");
+      candidate_ids.push_back(output_id);
+    }
+  }
+
+  if (payload.issuance)
+  {
+    const blobdata_ref bytes{reinterpret_cast<const char*>(encoded_issuance.data()),
+      encoded_issuance.size()};
+    db.add_asset_record(issued_id, height, bytes);
+  }
+  for (const asset_ownership_proof& proof : payload.ownership_proofs)
+    db.add_asset_key_image(proof.key_image, height);
+  size_t candidate_index = 0;
+  for (size_t group = 0; group < payload.balances.size(); ++group)
+  {
+    const confidential_asset_balance& balance = payload.balances[group];
+    for (size_t index = 0; index < balance.outputs.size(); ++index)
+    {
+      const asset_output_data_t output{balance.asset_id,
+        payload.output_destinations[group][index], balance.outputs[index], height};
+      db.add_asset_output(candidate_ids[candidate_index++], output);
+    }
+  }
+  output_ids = std::move(candidate_ids);
+  return true;
+}
 }
 }