darkfi/geode/
util.rs

1/* This file is part of DarkFi (https://dark.fi)
2 *
3 * Copyright (C) 2020-2026 Dyne.org foundation
4 *
5 * This program is free software: you can redistribute it and/or modify
6 * it under the terms of the GNU Affero General Public License as
7 * published by the Free Software Foundation, either version 3 of the
8 * License, or (at your option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 * GNU Affero General Public License for more details.
14 *
15 * You should have received a copy of the GNU Affero General Public License
16 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
17 */
18
19use futures::AsyncRead;
20use smol::io::{self, AsyncReadExt};
21
22pub fn hash_to_string(hash: &blake3::Hash) -> String {
23    bs58::encode(hash.as_bytes()).into_string()
24}
25
26/// smol::fs::File::read does not guarantee that the buffer will be filled, even if the buffer is
27/// smaller than the file. This is a workaround.
28/// This reads the stream until the buffer is full or until we reached the end of the stream.
29pub async fn read_until_filled(
30    mut stream: impl AsyncRead + Unpin,
31    buffer: &mut [u8],
32) -> io::Result<&[u8]> {
33    let mut total_bytes_read = 0;
34
35    while total_bytes_read < buffer.len() {
36        let bytes_read = stream.read(&mut buffer[total_bytes_read..]).await?;
37        if bytes_read == 0 {
38            break; // EOF reached
39        }
40        total_bytes_read += bytes_read;
41    }
42
43    Ok(&buffer[..total_bytes_read])
44}