Rust网络协议库tor-linkspec的使用：Tor链接规范解析与安全通信实现

完整示例代码

下面是一个完整的示例，展示了如何使用tor-linkspec库进行Tor链接规范的解析、构建和安全通信：

use tor_linkspec::{
    ChanTarget, EncodedLinkSpec, LinkSpec, OwnedChanTargetBuilder, 
    RelayIds, RelayIdsBuilder
};
use std::net::{IpAddr, Ipv4Addr, SocketAddr};

fn main() {
    // 示例1: 创建基本RelayIds
    example_create_relay_ids();
    
    // 示例2: 解析链接描述符
    let raw_data = vec![
        0x01, 0x00, 0x02, 0x01, 0xBB, 0x02,  // OR Port: 192.0.2.1:443
        0x03, 0x12, 0x12, 0x12, 0x12, 0x12,  // Ed25519 ID
        0x12, 0x12, 0x12, 0x12, 0x12, 0x12,
        0x12, 0x12, 0x12, 0x12, 0x12, 0x12,
        0x12, 0x12, 0x12, 0x12, 0x12, 0x12,
        0x12, 0x12, 0x12, 0x12, 0x12, 0x12,
        0x04, 0x34, 0x34, 0x34, 0x34, 0x34,  // RSA ID
        0x34, 0x34, 0x34, 0x34, 0x34, 0x34,
        0x34, 0x34, 0x34, 0x34, 0x34, 0x34,
        0x34, 0x34, 0x34, 0x34
    ];
    example_parse_link_specifiers(&raw_data);
    
    // 示例3: 构建自定义Relay
    example_build_custom_relay();
    
    // 示例4: 自定义链接规范
    example_custom_link_specifier();
}

/// 示例1: 创建基本RelayIds
fn example_create_relay_ids() {
    println!("\n=== 示例1: 创建基本RelayIds ===");
    
    let relay_ids = RelayIds::builder()
        .ed_identity([0x12; 32].into())  // 设置Ed25519身份密钥
        .rsa_identity([0x34; 20].into()) // 设置RSA身份指纹
        .build()
        .unwrap();
    
    println!("Relay IDs: {:?}", relay_ids);
    
    // 作为通道目标使用
    let target: &dyn ChanTarget = &relay_ids;
    println!("Link specifiers:");
    for ls in target.link_specifiers() {
        println!("  {:?}", ls);
    }
    
    // 编码链接规范
    let encoded = EncodedLinkSpec::new(target.link_specifiers());
    println!("Encoded link specifiers: {:?}", encoded);
}

/// 示例2: 解析链接描述符
fn example_parse_link_specifiers(raw_data: &[u8]) {
    println!("\n=== 示例2: 解析链接描述符 ===");
    
    let parsed = LinkSpec::decode_list(raw_data).unwrap();
    
    for spec in parsed {
        match spec {
            LinkSpec::OrPort(addr_port) => {
                println!("OR端口: {}:{}", addr_port.ip(), addr_port.port());
            }
            LinkSpec::Ed25519Id(key) => {
                println!("Ed25519身份ID: {:?}", key);
            }
            LinkSpec::RsaId(fingerprint) => {
                println!("RSA身份指纹: {:?}", fingerprint);
            }
            _ => println!("其他链接描述符: {:?}", spec),
        }
    }
}

/// 示例3: 构建自定义Relay
fn example_build_custom_relay() {
    println!("\n=== 示例3: 构建自定义Relay ===");
    
    let mut builder = RelayIdsBuilder::default();
    
    // 设置Ed25519身份密钥
    builder.ed_identity([0x56; 32].into());
    
    // 设置RSA身份指纹
    builder.rsa_identity([0x78; 20].into());
    
    // 添加OR端口
    let addr = SocketAddr::new(IpAddr::V4(Ipv4Addr::new(192, 0, 2, 1)), 443);
    builder.add_or_port(addr);
    
    let relay_ids = builder.build().unwrap();
    println!("构建的Relay: {:?}", relay_ids);
}

/// 示例4: 自定义链接规范
fn example_custom_link_specifier() {
    println!("\n=== 示例4: 自定义链接规范 ===");
    
    let custom_data = vec![0x01, 0x02, 0x03, 0x04];
    let custom_spec = LinkSpec::Unknown(0x80, custom_data); // 0x80是自定义类型
    
    let mut builder = RelayIdsBuilder::default();
    builder.add_link_specifier(custom_spec);
    
    let relay_ids = builder.build().unwrap();
    println!("带有自定义描述符的Relay: {:?}", relay_ids);
}

代码说明

1. 创建RelayIds:

   • 使用RelayIds::builder()创建构建器
   • 设置Ed25519和RSA身份信息
   • 构建并打印结果

2. 解析链接描述符:

   • 使用LinkSpec::decode_list()解析原始字节数据
   • 匹配不同类型的链接描述符并打印

3. 构建自定义Relay:

   • 使用RelayIdsBuilder创建自定义Relay
   • 添加身份信息和OR端口
   • 构建并打印结果

4. 自定义链接规范:

   • 创建自定义类型的链接描述符(LinkSpec::Unknown)
   • 添加到Relay构建器中
   • 构建并打印结果

运行结果示例

运行上述代码将输出类似以下内容:

=== 示例1: 创建基本RelayIds ===
Relay IDs: RelayIds { ... }
Link specifiers:
  Ed25519Id(...)
  RsaId(...)
Encoded link specifiers: EncodedLinkSpec(...)

=== 示例2: 解析链接描述符 ===
OR端口: 192.0.2.1:443
Ed25519身份ID: Ed25519Id(...)
RSA身份指纹: RsaId(...)

=== 示例3: 构建自定义Relay ===
构建的Relay: RelayIds { ... }

=== 示例4: 自定义链接规范 ===
带有自定义描述符的Relay: RelayIds { ... }