ts-analyzer 0.3.0

extern crate ts_analyzer;

use std::env;
use std::fs::File;
use std::io::BufReader;

use memmem::Searcher;
use memmem::TwoWaySearcher;
use ts_analyzer::reader::TsReader;

const KLV_HEADER: &[u8; 16] =
    b"\x06\x0E\x2B\x34\x02\x0B\x01\x01\x0E\x01\x03\x01\x01\x00\x00\x00";

fn main() {
    env_logger::init();
    let filename = env::var("TEST_FILE").expect("Environment variable not set");
    println!("Reading data from {}", filename);

    let f = File::open(filename.clone()).expect("Couldn't open file");
    let buf_reader = BufReader::new(f);
    // Reader must be mutable due to internal state changing to keep track of
    // what packet is to be read next and what payloads are being tracked.
    let mut reader = TsReader::new(buf_reader)
        .expect("Transport Stream file contains no SYNC bytes.");
    let search = TwoWaySearcher::new(KLV_HEADER);

    let mut payload;
    loop {
        println!("Reading packet");
        // Run through packets until we get to one with a payload.
        payload = match reader.next_payload() {
            Ok(payload) => payload.expect("No valid complete TS payload found"),
            Err(e) => {
                panic!("An error was hit!: {}", e);
            }
        };

        if search.search_in(&payload).is_some() {
            break;
        }
    }

    println!("Found KLV payload bytes: {:02X?}", payload);
}