pookie 0.0.2

Load cookie from your web browsers
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

use std::path::PathBuf;

use eyre::Result;
use libesedb::EseDb;

use crate::common::{date, enums::Cookie};

/// Returns cookies from IE based browsers
pub fn internet_explorer_based(
    db_path: PathBuf,
    domains: Option<Vec<String>>,
) -> Result<Vec<Cookie>> {
    unsafe {
        if let Some(path) = db_path.to_str() {
            crate::windows::restart_manager::release_file_lock(path);
        }
    }
    let db = EseDb::open(db_path)?;
    let mut cookies: Vec<Cookie> = vec![];

    for table in db.iter_tables()? {
        let table = table?;
        let name: String = table.name()?;

        if name.starts_with("CookieEntry") {
            for rec in table.iter_records()? {
                let rec = rec?;
                let host = rec.value(8)?;
                let host = host.as_str().unwrap_or("");
                let path = rec.value(9)?;
                let path = path.as_str().unwrap_or("");
                let name: Vec<u8> =
                    rec.value(10)?.as_bytes().unwrap_or(&[]).to_vec();
                let name = String::from_utf8(name)
                    .unwrap_or("".to_string())
                    .trim_matches('\0')
                    .to_string();
                let value = rec.value(11)?;
                let same_site = 0;
                let value =
                    String::from_utf8(value.as_bytes().unwrap_or(&[]).to_vec())
                        .unwrap_or("".to_string())
                        .trim_matches('\0')
                        .to_string();
                let secure = false;
                let expires = rec.value(4)?.to_u64().unwrap_or(0);
                let expires = date::internet_explorer_timestamp(expires);
                let http_only = false;

                let should_append = domains.is_none()
                    || domains.iter().any(|d| d.contains(&host.to_string()));
                if should_append {
                    cookies.push(Cookie {
                        domain: host.to_string(),
                        path: path.to_string(),
                        secure,
                        expires,
                        name,
                        value,
                        http_only,
                        same_site,
                    })
                }
            }
        }
    }
    Ok(cookies)
}