ipfs_unixfs/pb/
merkledag.rs1#![allow(non_snake_case)]
3#![allow(non_upper_case_globals)]
4#![allow(non_camel_case_types)]
5#![allow(unused_imports)]
6#![allow(unknown_lints)]
7#![allow(clippy::all)]
8#![cfg_attr(rustfmt, rustfmt_skip)]
9use super::*;
10use quick_protobuf::sizeofs::*;
11use quick_protobuf::{BytesReader, MessageRead, MessageWrite, Result, Writer, WriterBackend};
12use alloc::borrow::Cow;
13use core::convert::TryFrom;
14use std::io::Write;
15use core::ops::Deref;
16use core::ops::DerefMut;
17
18#[derive(Debug, Default, PartialEq, Clone)]
19pub struct PBLink<'a> {
20 pub Hash: Option<Cow<'a, [u8]>>,
21 pub Name: Option<Cow<'a, str>>,
22 pub Tsize: Option<u64>,
23}
24impl<'a> MessageRead<'a> for PBLink<'a> {
25 fn from_reader(r: &mut BytesReader, bytes: &'a [u8]) -> Result<Self> {
26 let mut msg = Self::default();
27 while !r.is_eof() {
28 match r.next_tag(bytes) {
29 Ok(10) => msg.Hash = Some(r.read_bytes(bytes).map(Cow::Borrowed)?),
30 Ok(18) => msg.Name = Some(r.read_string(bytes).map(Cow::Borrowed)?),
31 Ok(24) => msg.Tsize = Some(r.read_uint64(bytes)?),
32 Ok(t) => {
33 r.read_unknown(bytes, t)?;
34 }
35 Err(e) => return Err(e),
36 }
37 }
38 Ok(msg)
39 }
40}
41impl<'a> MessageWrite for PBLink<'a> {
42 fn get_size(&self) -> usize {
43 0 + self.Hash.as_ref().map_or(0, |m| 1 + sizeof_len((m).len()))
44 + self.Name.as_ref().map_or(0, |m| 1 + sizeof_len((m).len()))
45 + self
46 .Tsize
47 .as_ref()
48 .map_or(0, |m| 1 + sizeof_varint(*(m) as u64))
49 }
50 fn write_message<W: WriterBackend>(&self, w: &mut Writer<W>) -> Result<()> {
51 if let Some(ref s) = self.Hash {
52 w.write_with_tag(10, |w| w.write_bytes(&**s))?;
53 }
54 if let Some(ref s) = self.Name {
55 w.write_with_tag(18, |w| w.write_string(&**s))?;
56 }
57 if let Some(ref s) = self.Tsize {
58 w.write_with_tag(24, |w| w.write_uint64(*s))?;
59 }
60 Ok(())
61 }
62}
63#[derive(Debug, Default, PartialEq, Clone)]
64pub struct PBNode<'a> {
65 pub Links: Vec<PBLink<'a>>,
66 pub Data: Option<Cow<'a, [u8]>>,
67}
68impl<'a> MessageRead<'a> for PBNode<'a> {
69 fn from_reader(r: &mut BytesReader, bytes: &'a [u8]) -> Result<Self> {
70 let mut msg = Self::default();
71 while !r.is_eof() {
72 match r.next_tag(bytes) {
73 Ok(18) => msg.Links.push(r.read_message::<PBLink<'a>>(bytes)?),
74 Ok(10) => msg.Data = Some(r.read_bytes(bytes).map(Cow::Borrowed)?),
75 Ok(t) => {
76 r.read_unknown(bytes, t)?;
77 }
78 Err(e) => return Err(e),
79 }
80 }
81 Ok(msg)
82 }
83}
84impl<'a> MessageWrite for PBNode<'a> {
85 fn get_size(&self) -> usize {
86 0 + self
87 .Links
88 .iter()
89 .map(|s| 1 + sizeof_len((s).get_size()))
90 .sum::<usize>()
91 + self.Data.as_ref().map_or(0, |m| 1 + sizeof_len((m).len()))
92 }
93 fn write_message<W: WriterBackend>(&self, w: &mut Writer<W>) -> Result<()> {
94 for s in &self.Links {
95 w.write_with_tag(18, |w| w.write_message(s))?;
96 }
97 if let Some(ref s) = self.Data {
98 w.write_with_tag(10, |w| w.write_bytes(&**s))?;
99 }
100 Ok(())
101 }
102}