sui_move_natives_v1/
object.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
// Copyright (c) Mysten Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

use crate::{object_runtime::ObjectRuntime, NativesCostTable};
use move_binary_format::errors::PartialVMResult;
use move_core_types::{account_address::AccountAddress, gas_algebra::InternalGas};
use move_vm_runtime::{native_charge_gas_early_exit, native_functions::NativeContext};
use move_vm_types::{
    loaded_data::runtime_types::Type,
    natives::function::NativeResult,
    pop_arg,
    values::{StructRef, Value},
};
use smallvec::smallvec;
use std::collections::VecDeque;

#[derive(Clone)]
pub struct BorrowUidCostParams {
    pub object_borrow_uid_cost_base: InternalGas,
}
/***************************************************************************************************
 * native fun borrow_uid
 * Implementation of the Move native function `borrow_uid<T: key>(obj: &T): &UID`
 *   gas cost: object_borrow_uid_cost_base                | this is hard to calculate as it's very sensitive to `borrow_field` impl. Making it flat
 **************************************************************************************************/
pub fn borrow_uid(
    context: &mut NativeContext,
    ty_args: Vec<Type>,
    mut args: VecDeque<Value>,
) -> PartialVMResult<NativeResult> {
    debug_assert!(ty_args.len() == 1);
    debug_assert!(args.len() == 1);

    let borrow_uid_cost_params = context
        .extensions_mut()
        .get::<NativesCostTable>()
        .borrow_uid_cost_params
        .clone();

    // Charge base fee
    native_charge_gas_early_exit!(context, borrow_uid_cost_params.object_borrow_uid_cost_base);

    let obj = pop_arg!(args, StructRef);
    let id_field = obj.borrow_field(0)?;

    Ok(NativeResult::ok(context.gas_used(), smallvec![id_field]))
}

#[derive(Clone)]
pub struct DeleteImplCostParams {
    pub object_delete_impl_cost_base: InternalGas,
}
/***************************************************************************************************
 * native fun delete_impl
 * Implementation of the Move native function `delete_impl(id: address)`
 *   gas cost: cost_base                | this is a simple ID deletion
 **************************************************************************************************/
pub fn delete_impl(
    context: &mut NativeContext,
    ty_args: Vec<Type>,
    mut args: VecDeque<Value>,
) -> PartialVMResult<NativeResult> {
    debug_assert!(ty_args.is_empty());
    debug_assert!(args.len() == 1);

    let delete_impl_cost_params = context
        .extensions_mut()
        .get::<NativesCostTable>()
        .delete_impl_cost_params
        .clone();

    // Charge base fee
    native_charge_gas_early_exit!(
        context,
        delete_impl_cost_params.object_delete_impl_cost_base
    );

    // unwrap safe because the interface of native function guarantees it.
    let uid_bytes = pop_arg!(args, AccountAddress);

    let obj_runtime: &mut ObjectRuntime = context.extensions_mut().get_mut();
    obj_runtime.delete_id(uid_bytes.into())?;
    Ok(NativeResult::ok(context.gas_used(), smallvec![]))
}

#[derive(Clone)]
pub struct RecordNewIdCostParams {
    pub object_record_new_uid_cost_base: InternalGas,
}
/***************************************************************************************************
 * native fun record_new_uid
 * Implementation of the Move native function `record_new_uid(id: address)`
 *   gas cost: object_record_new_uid_cost_base                | this is a simple ID addition
 **************************************************************************************************/
pub fn record_new_uid(
    context: &mut NativeContext,
    ty_args: Vec<Type>,
    mut args: VecDeque<Value>,
) -> PartialVMResult<NativeResult> {
    debug_assert!(ty_args.is_empty());
    debug_assert!(args.len() == 1);

    let record_new_id_cost_params = context
        .extensions_mut()
        .get::<NativesCostTable>()
        .record_new_id_cost_params
        .clone();

    // Charge base fee
    native_charge_gas_early_exit!(
        context,
        record_new_id_cost_params.object_record_new_uid_cost_base
    );

    // unwrap safe because the interface of native function guarantees it.
    let uid_bytes = pop_arg!(args, AccountAddress);

    let obj_runtime: &mut ObjectRuntime = context.extensions_mut().get_mut();
    obj_runtime.new_id(uid_bytes.into())?;
    Ok(NativeResult::ok(context.gas_used(), smallvec![]))
}