lib_vtop 1.0.0

A minimal libre package for VTOP
Documentation 
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107

use reqwest;
use scraper::{Html, Selector};

pub async fn university_wifi_login_logout(i: i32, username: String, password: String) -> (bool, String) {
    let client = reqwest::Client::new();
    if i == 0 {
        let res = client.get("http://172.18.10.10:1000/login?").send().await;
        match res {
            Ok(x) => {
                let res = x.text().await.unwrap_or("notext".to_string());

                let iip = parse_initil_page(res);

                if iip.0 {
                    let body = format!(
                        "magic={}&username={}&password={}",
                        iip.1, username, password
                    );
                    let res = client
                        .post("http://172.18.10.10:1000/?")
                        .body(body)
                        .send()
                        .await;
                    match res {
                        Ok(x) => {
                            let text = x.text().await.unwrap_or("notext".to_string());
                            let iip3 = get_em(text, "p");
                            if iip3.0 {
                                return (false, iip3.1);
                            } else {
                                let res = client
                                    .get("http://172.18.10.10:1000/keepalive?")
                                    .send()
                                    .await;
                                match res {
                                    Ok(x) => {
                                        let text = x.text().await.unwrap_or("notext".to_string());
                                        let iip2 = get_em(text, "h1");

                                        if iip.0 {
                                            return (true, iip2.1);
                                        } else {
                                            (false, "logined".to_string());
                                        }
                                    }
                                    Err(_e) => {
                                        return (false, "NE".to_string());
                                    }
                                }
                            }
                        }
                        Err(_e) => {
                            return (false, "NE".to_string());
                        }
                    }
                } else {
                    return (iip.0, "".to_string());
                }
            }
            Err(_e) => return (false, "NE".to_string()),
        }
    } else if i == 1 {
        let res = client.get("http://172.18.10.10:1000/logout?").send().await;
        match res {
            Ok(x) => {
                let text = x.text().await.unwrap_or("notext".to_string());
                //println!("{}", text);
                let llp = get_em(text, "H3");
                if llp.0 {
                    return (true, llp.1);
                } else {
                    return (false, "NL".to_string());
                }
            }
            Err(_e) => {
                return (false, "NE".to_string());
            }
        }
    }

    (false, "".to_string())
}

fn parse_initil_page(res: String) -> (bool, String) {
    let document = Html::parse_document(&res);
    let selector = Selector::parse("input[type=hidden][name=magic]").unwrap();
    let page = document
        .select(&selector)
        .next()
        .and_then(|element| element.value().attr("value").map(|value| value.to_string()));
    match page {
        Some(k) => return (true, k),
        None => return (false, "".to_string()),
    }
}
fn get_em(text: String, sel: &str) -> (bool, String) {
    let document = Html::parse_document(&text);
    let selector = Selector::parse(sel).unwrap();
    let page = document.select(&selector).next();
    match page {
        Some(k) => {
            let o = k.text().collect::<Vec<_>>()[0];
            return (true, o.trim().to_string());
        }
        None => return (false, "".to_string()),
    }
}