m3u8_down/
m3u8_down.rs

1use reqrio::{json, HlsError, Method, ReqExt, ScReq, Timeout, ALPN, Cipher};
2use std::fs::File;
3use std::io::Write;
4
5struct M3u8DownEngine {
6    req: ScReq,
7    key_url: String,
8    sequence: u128,
9    index_url: String,
10    ts_urls: Vec<String>,
11    cipher: Cipher,
12}
13
14impl M3u8DownEngine {
15    fn new(index: impl ToString) -> M3u8DownEngine {
16        let mut req = ScReq::new().with_alpn(ALPN::Http20); //.with_proxy(Proxy::new_http_plain("127.0.0.1", 10809));
17        req.set_headers_json(json::object! {
18            "Host": "",
19            "sec-ch-ua-platform": "Android",
20            "user-agent": "Mozilla/5.0 (Linux; Android 13; Pixel 7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/116.0.0.0 Mobile Safari/537.36",
21            "sec-ch-ua": r#""Android WebView";v="135", "Not-A.Brand";v="8", "Chromium";v="135""#,
22            "sec-ch-ua-mobile": "?1",
23            "accept": "*/*",
24            "origin": "",
25            "x-requested-with": "mark.via",
26            "sec-fetch-site": "cross-site",
27            "sec-fetch-mode": "cors",
28            "sec-fetch-dest": "empty",
29            "referer": "",
30            "accept-encoding": "gzip, deflate, br, zstd",
31            "accept-language": "en,zh-CN;q=0.9,zh;q=0.8,en-US;q=0.7",
32            "priority": "u=1, i"
33        }).unwrap();
34        M3u8DownEngine {
35            req,
36            index_url: index.to_string(),
37            sequence: 0,
38            key_url: "".to_string(),
39            ts_urls: vec![],
40            cipher: Cipher::aes_128_ecb(),
41        }
42    }
43
44    fn down_ts(&mut self) -> Result<(), HlsError> {
45        let mut file = File::create("4.mp4")?;
46        for (index, ts_url) in self.ts_urls.iter().enumerate() {
47            println!("Downloading: {:3}/{}; url:{}", index, self.ts_urls.len(), ts_url);
48            self.req.set_url(ts_url)?;
49            let body = self.req.get()?.bytes()?;
50            file.write_all(&if self.key_url.is_empty() { body } else { self.cipher.decrypt(body)? })?;
51        }
52        Ok(())
53    }
54
55    fn get_key(&mut self) -> Result<(), HlsError> {
56        println!("key url: {}", self.key_url);
57        self.req.set_url(self.key_url.as_str())?;
58        let key = self.req.send_check(Method::GET)?.text()?;
59        println!("key: {}; sequence: {}", key, self.sequence);
60        self.cipher.set_secret_key(key.into_bytes(), Some(self.sequence.to_be_bytes().to_vec()));
61        Ok(())
62    }
63
64    fn download(&mut self) -> Result<(), HlsError> {
65        self.req.set_url(self.index_url.as_str())?;
66        let body = self.req.send_check(Method::GET)?.text()?;
67        // println!("{}", body);
68        for line in body.split("\n") {
69            if line.starts_with("#EXT-X-MEDIA-SEQUENCE:") {
70                self.sequence = line.trim().replace("#EXT-X-MEDIA-SEQUENCE:", "").parse()?;
71                continue;
72            }
73            if line.starts_with("#EXT-X-KEY") {
74                let pos = line.find("URI=\"");
75                if let Some(pos) = pos {
76                    self.key_url = line[pos + 4..].trim().replace("\"", "");
77                }
78                if line.contains("=AES-128,") {
79                    self.cipher = Cipher::aes_128_cbc();
80                }
81                continue;
82            }
83            if line.starts_with("http") {
84                self.ts_urls.push(line.trim().to_string());
85            }
86        }
87        let mut timeout = Timeout::new_same(5000, 30000);
88        timeout.set_handle_times(10);
89        self.req.set_timeout(timeout);
90        if !self.key_url.is_empty() { self.get_key()?; }
91        self.down_ts()?;
92        Ok(())
93    }
94    
95}
96
97
98fn main() {
99    let index = "";
100    let mut engine = M3u8DownEngine::new(index);
101    engine.download().unwrap();
102    println!("{:?} {:?}", engine.key_url, engine.ts_urls);
103}
m3u8_down/m3u8_down.rs

m3u8_down/
m3u8_down.rs
