add metering for deserialization

rust/candid/benches/benchmark.rs

@@ -2,6 +2,8 @@ use candid::{CandidType, Decode, Deserialize, Encode, Int, Nat, Principal};
 use criterion::{criterion_group, criterion_main, BatchSize, BenchmarkId, Criterion};
 use std::collections::BTreeMap;
 
+const COST: usize = 20_000_000;
+
 fn bench_blob(c: &mut Criterion) {
     use serde_bytes::{ByteBuf, Bytes};
     let vec: Vec<u8> = vec![0x61; 524288];
@@ -11,7 +13,7 @@ fn bench_blob(c: &mut Criterion) {
             || vec.clone(),
             |vec| {
                 let bytes = Encode!(&ByteBuf::from(vec)).unwrap();
-                Decode!(&bytes, ByteBuf).unwrap();
+                Decode!([COST]; &bytes, ByteBuf).unwrap();
             },
             BatchSize::SmallInput,
         )
@@ -21,7 +23,7 @@ fn bench_blob(c: &mut Criterion) {
             || vec.clone(),
             |vec| {
                 let bytes = Encode!(&Bytes::new(vec)).unwrap();
-                Decode!(&bytes, &Bytes).unwrap();
+                Decode!([COST]; &bytes, &Bytes).unwrap();
             },
             BatchSize::SmallInput,
         )
@@ -32,7 +34,7 @@ fn bench_blob(c: &mut Criterion) {
             || text.clone(),
             |text| {
                 let bytes = Encode!(&text).unwrap();
-                Decode!(&bytes, String).unwrap();
+                Decode!([COST]; &bytes, String).unwrap();
             },
             BatchSize::SmallInput,
         )
@@ -42,7 +44,7 @@ fn bench_blob(c: &mut Criterion) {
             || text.clone(),
             |text| {
                 let bytes = Encode!(text).unwrap();
-                Decode!(&bytes, &str).unwrap();
+                Decode!([COST]; &bytes, &str).unwrap();
             },
             BatchSize::SmallInput,
         )
@@ -59,7 +61,7 @@ fn bench_collections(c: &mut Criterion) {
                 || vec8.clone(),
                 |vec| {
                     let bytes = Encode!(&vec).unwrap();
-                    Decode!(&bytes, Vec<u8>).unwrap();
+                    Decode!([COST]; &bytes, Vec<u8>).unwrap();
                 },
                 BatchSize::SmallInput,
             )
@@ -72,7 +74,7 @@ fn bench_collections(c: &mut Criterion) {
                 || vec64.clone(),
                 |vec| {
                     let bytes = Encode!(&vec).unwrap();
-                    Decode!(&bytes, Vec<i64>).unwrap();
+                    Decode!([COST]; &bytes, Vec<i64>).unwrap();
                 },
                 BatchSize::SmallInput,
             )
@@ -85,7 +87,7 @@ fn bench_collections(c: &mut Criterion) {
                 || vec.clone(),
                 |vec| {
                     let bytes = Encode!(&vec).unwrap();
-                    Decode!(&bytes, Vec<candid::Int>).unwrap();
+                    Decode!([COST]; &bytes, Vec<candid::Int>).unwrap();
                 },
                 BatchSize::SmallInput,
             )
@@ -100,7 +102,7 @@ fn bench_collections(c: &mut Criterion) {
                 || map.clone(),
                 |map| {
                     let bytes = Encode!(&map).unwrap();
-                    Decode!(&bytes, BTreeMap<String, Nat>).unwrap();
+                    Decode!([COST]; &bytes, BTreeMap<String, Nat>).unwrap();
                 },
                 BatchSize::SmallInput,
             )
@@ -130,7 +132,7 @@ fn bench_recursion(c: &mut Criterion) {
         c.bench_with_input(BenchmarkId::new("option list", n), &list, |b, list| {
             b.iter(|| {
                 let bytes = Encode!(list).unwrap();
-                Decode!(&bytes, Option<Box<List>>).unwrap()
+                Decode!([COST]; &bytes, Option<Box<List>>).unwrap()
             })
         });
     }
@@ -141,7 +143,7 @@ fn bench_recursion(c: &mut Criterion) {
         c.bench_with_input(BenchmarkId::new("variant list", n), &list, |b, list| {
             b.iter(|| {
                 let bytes = Encode!(list).unwrap();
-                Decode!(&bytes, VariantList).unwrap()
+                Decode!([COST]; &bytes, VariantList).unwrap()
             })
         });
     }
@@ -177,7 +179,7 @@ fn bench_profile(c: &mut Criterion) {
         |b, vec| {
             b.iter(|| {
                 let bytes = Encode!(vec).unwrap();
-                Decode!(&bytes, Vec<Profile>).unwrap()
+                Decode!([COST]; &bytes, Vec<Profile>).unwrap()
             })
         },
     );