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
122
123
124
125
126
127
128
129
130
131
132

//! Provides overloaded methods for
//! [Accumulator](../struct.Accumulator.html) and
//! [Update](../struct.Update.html) that automatically serialize the element.
use generic_array::ArrayLength;
use serde::Serialize;
use crate::bigint::BigInt;
use crate::{Accumulator, Update, Witness};
use crate::mapper::Mapper;

/// Trait for an accumulator that automatically serializes elements into
/// [VelocyPack](https://github.com/arangodb/velocypack) format.
trait VpackAccumulator<T: BigInt> {

    fn add<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize
    ) -> Witness<T>;

    fn del<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize,
        w: &Witness<T>
    ) -> Result<(), &'static str>;

    fn prove<Map: Mapper, N: ArrayLength<u8>>(
        &self,
        x: &impl Serialize
    ) -> Result<Witness<T>, &'static str>;

    fn verify<Map: Mapper, N: ArrayLength<u8>>(
        &self,
        x: &impl Serialize,
        w: &Witness<T>
    ) -> Result<(), &'static str>;
}

impl<T: BigInt> VpackAccumulator<T> for Accumulator<T> {

    fn add<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize
    ) -> Witness<T> {
        self.add::<Map, N>(&velocypack::to_bytes(x).unwrap())
    }

    fn del<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize,
        w: &Witness<T>
    ) -> Result<(), &'static str> {
        self.del::<Map, N>(&velocypack::to_bytes(x).unwrap(), w)
    }

    fn prove<Map: Mapper, N: ArrayLength<u8>>(
        &self,
        x: &impl Serialize
    ) -> Result<Witness<T>, &'static str> {
        self.prove::<Map, N>(&velocypack::to_bytes(x).unwrap())
    }

    fn verify<Map: Mapper, N: ArrayLength<u8>>(
        &self,
        x: &impl Serialize,
        w: &Witness<T>
    ) -> Result<(), &'static str> {
        self.verify::<Map, N>(&velocypack::to_bytes(x).unwrap(), w)
    }
}

/// Trait for an update that automatically serializes elements into
/// [VelocyPack](https://github.com/arangodb/velocypack) format.
trait VpackUpdate<T: BigInt> {

    fn add<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize,
        w: &Witness<T>
    );

    fn del<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize,
        w: &Witness<T>
    );

    fn undo_add<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize,
        w: &Witness<T>
    );

    fn undo_del<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize,
        w: &Witness<T>
    );
}

impl<T: BigInt> VpackUpdate<T> for Update<T> {

    fn add<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize,
        w: &Witness<T>
    ) {
        self.add::<Map, N>(&velocypack::to_bytes(x).unwrap(), w)
    }

    fn del<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize,
        w: &Witness<T>
    ) {
        self.del::<Map, N>(&velocypack::to_bytes(x).unwrap(), w)
    }

    fn undo_add<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize,
        w: &Witness<T>
    ) {
        self.undo_add::<Map, N>(&velocypack::to_bytes(x).unwrap(), w)
    }

    fn undo_del<Map: Mapper, N: ArrayLength<u8>>(
        &mut self,
        x: &impl Serialize,
        w: &Witness<T>
    ) {
        self.undo_del::<Map, N>(&velocypack::to_bytes(x).unwrap(), w)
    }
}