95 lines
2.5 KiB
Rust
95 lines
2.5 KiB
Rust
|
//! # Raw PSBT Key-Value Pairs
|
||
|
|
//!
|
||
|
|
//! Raw PSBT key-value pairs as defined at
|
||
|
|
//! https://github.com/bitcoin/bips/blob/master/bip-0174.mediawiki.
|
||
|
|
|
||
|
|
use std::fmt;
|
||
|
|
|
||
|
|
use consensus::encode::{Decodable, Encodable, VarInt, MAX_VEC_SIZE};
|
||
|
|
use consensus::encode::{self, Decoder, Encoder};
|
||
|
|
use util::psbt::Error;
|
||
|
|
|
||
|
|
/// A PSBT key in its raw byte form.
|
||
|
|
#[derive(Debug, PartialEq, Hash, Eq, Clone)]
|
||
|
|
pub struct Key {
|
||
|
|
/// The type of this PSBT key.
|
||
|
|
pub type_value: u8,
|
||
|
|
/// The key itself in raw byte form.
|
||
|
|
pub key: Vec<u8>,
|
||
|
|
}
|
||
|
|
|
||
|
|
/// A PSBT key-value pair in its raw byte form.
|
||
|
|
#[derive(Debug, PartialEq)]
|
||
|
|
pub struct Pair {
|
||
|
|
/// The key of this key-value pair.
|
||
|
|
pub key: Key,
|
||
|
|
/// The value of this key-value pair in raw byte form.
|
||
|
|
pub value: Vec<u8>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl fmt::Display for Key {
|
||
|
|
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||
|
|
use hex;
|
||
|
|
|
||
|
|
write!(f, "type: {:#x}, key: {}", self.type_value, hex::encode(&self.key))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<D: Decoder> Decodable<D> for Key {
|
||
|
|
fn consensus_decode(d: &mut D) -> Result<Self, encode::Error> {
|
||
|
|
let VarInt(byte_size): VarInt = Decodable::consensus_decode(d)?;
|
||
|
|
|
||
|
|
if byte_size == 0 {
|
||
|
|
return Err(Error::NoMorePairs.into());
|
||
|
|
}
|
||
|
|
|
||
|
|
let key_byte_size: u64 = byte_size - 1;
|
||
|
|
|
||
|
|
if key_byte_size > MAX_VEC_SIZE as u64 {
|
||
|
|
return Err(encode::Error::OversizedVectorAllocation { requested: key_byte_size as usize, max: MAX_VEC_SIZE } )
|
||
|
|
}
|
||
|
|
|
||
|
|
let type_value: u8 = Decodable::consensus_decode(d)?;
|
||
|
|
|
||
|
|
let mut key = Vec::with_capacity(key_byte_size as usize);
|
||
|
|
for _ in 0..key_byte_size {
|
||
|
|
key.push(Decodable::consensus_decode(d)?);
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(Key {
|
||
|
|
type_value: type_value,
|
||
|
|
key: key,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<S: Encoder> Encodable<S> for Key {
|
||
|
|
fn consensus_encode(&self, s: &mut S) -> Result<(), encode::Error> {
|
||
|
|
VarInt((self.key.len() + 1) as u64).consensus_encode(s)?;
|
||
|
|
|
||
|
|
self.type_value.consensus_encode(s)?;
|
||
|
|
|
||
|
|
for key in &self.key {
|
||
|
|
key.consensus_encode(s)?
|
||
|
|
}
|
||
|
|
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<S: Encoder> Encodable<S> for Pair {
|
||
|
|
fn consensus_encode(&self, s: &mut S) -> Result<(), encode::Error> {
|
||
|
|
self.key.consensus_encode(s)?;
|
||
|
|
self.value.consensus_encode(s)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl<D: Decoder> Decodable<D> for Pair {
|
||
|
|
fn consensus_decode(d: &mut D) -> Result<Self, encode::Error> {
|
||
|
|
Ok(Pair {
|
||
|
|
key: Decodable::consensus_decode(d)?,
|
||
|
|
value: Decodable::consensus_decode(d)?,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|