ton_lib 0.0.38

A collection of types and utilities for interacting with the TON network
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

use crate::block_tlb::{BlockIdExt, BlockInfo};
use crate::clients::lite_client::client::LiteClient;
use crate::clients::lite_client::config::LiteClientConfig;
use crate::clients::tl_client::config::TLClientConfig;
use crate::clients::tl_client::tl::types::TLKeyStoreType;
use crate::error::TLError;
use futures_util::future::join_all;
use std::time::Duration;
use ton_lib_core::cell::TonCell;
use ton_lib_core::error::TLCoreError;
use ton_lib_core::traits::tlb::TLB;
use ton_liteapi::tl::response::BlockData;

pub async fn prepare_client_env(config: &mut TLClientConfig) -> Result<(), TLError> {
    if config.update_init_block {
        update_init_block(config).await?;
    }

    if let TLKeyStoreType::Directory { directory } = &config.init_opts.keystore_type {
        std::fs::create_dir_all(directory)?
    }
    Ok(())
}

async fn update_init_block(config: &mut TLClientConfig) -> Result<(), TLError> {
    log::info!("Updating init_block...");
    let lite_config = LiteClientConfig::new(&config.init_opts.config.net_config_json)?;
    let cur_init_seqno = lite_config.net_config.get_init_block_seqno();
    let lite_client = LiteClient::new(lite_config.clone())?;
    let lite_client_ref = &lite_client;
    let mut futs = vec![];
    for _ in lite_config.net_config.lite_endpoints.iter() {
        let future = async {
            let mc_info = lite_client_ref.get_mc_info().await?;
            let block = lite_client_ref.get_block(mc_info.last, None).await?;
            let seqno = parse_key_block_seqno(&block)?;
            lite_client_ref.lookup_mc_block(seqno).await
        };
        futs.push(future);
    }
    let exec_timeout = Duration::from_secs(config.update_init_block_timeout_sec.saturating_sub(1));
    let results = tokio::time::timeout(exec_timeout, join_all(futs)).await?;
    let mut max_block: Option<BlockIdExt> = None;
    for res in &results {
        match res {
            Ok(block) => {
                if max_block.is_none() || max_block.as_ref().unwrap().seqno < block.seqno {
                    max_block = Some(block.clone());
                }
            }
            Err(err) => log::warn!("Failed to get recent_init_block from node: {err:?}"),
        }
    }

    if let Some(block) = max_block {
        log::info!("init_block updated: {} -> {}", cur_init_seqno, block.seqno);
        let mut net_conf = lite_config.net_config.clone();
        net_conf.set_init_block(&block);
        config.init_opts.config.net_config_json = net_conf.to_json()?;
    }
    Ok(())
}

fn parse_key_block_seqno(block: &BlockData) -> Result<u32, TLError> {
    let block_cell = TonCell::from_boc(&block.data)?;
    if block_cell.refs.is_empty() {
        return Err(TLError::Custom("No refs in block cell".to_string()));
        // TODO make proper block parser
    }
    let mut parser = block_cell.refs[0].parser();
    let tag: usize = parser.read_num(32)?;
    if tag != BlockInfo::PREFIX.value {
        return Err(TLCoreError::TLBWrongPrefix {
            exp: BlockInfo::PREFIX.value,
            given: tag,
            bits_exp: BlockInfo::PREFIX.bits_len,
            bits_left: parser.data_bits_remaining()? + 32,
        }
        .into());
    }
    // version(32), merge_info(8), flags(8), seqno(32), vert_seqno(32), shard(104), utime(32), start/end lt(128),
    // validator_list_hash(32), catchain_seqno(32), min_ref_mc_seqno(32)
    parser.read_bits(32 + 8 + 8 + 32 + 32 + 104 + 32 + 128 + 32 + 32 + 32)?;
    let key_block_seqno = parser.read_num(32)?;
    Ok(key_block_seqno)
}