xref: /aosp_15_r20/tools/netsim/rust/http-proxy/tests/integration_test.rs (revision cf78ab8cffb8fc9207af348f23af247fb04370a6)

// Copyright 2024 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

use capture::pcap;
use std::io::Cursor;
use std::net::{IpAddr, Ipv6Addr};
use std::str::FromStr;
use tokio::io::BufReader;

fn ipv6_from_str(addr: &str) -> Result<IpAddr, std::io::Error> {
    match Ipv6Addr::from_str(addr) {
        Ok(addr) => Ok(addr.into()),
        Err(err) => Err(std::io::Error::new(std::io::ErrorKind::InvalidInput, err.to_string())),
    }
}

#[tokio::test]
async fn dns_manager() -> Result<(), std::io::Error> {
    const DATA: &[u8] = include_bytes!("../../capture/data/dns.cap");

    let mut reader = BufReader::new(Cursor::new(DATA));
    let header = pcap::read_file_header(&mut reader).await?;
    assert_eq!(header.linktype, pcap::LinkType::Ethernet.into());
    let mut dns_manager = http_proxy::DnsManager::new();
    loop {
        match pcap::read_record(&mut reader).await {
            Ok((_hdr, record)) => {
                dns_manager.add_from_ethernet_slice(&record);
            }
            Err(e) if e.kind() == std::io::ErrorKind::UnexpectedEof => {
                break;
            }
            Err(e) => {
                println!("Error: {:?}", e);
                assert!(false);
            }
        }
    }
    assert_eq!(dns_manager.len(), 4);

    //  0xf0d4 AAAA www.netbsd.org AAAA
    assert_eq!(
        dns_manager.get(&ipv6_from_str("2001:4f8:4:7:2e0:81ff:fe52:9a6b")?),
        Some("www.netbsd.org".into())
    );

    Ok(())
}