fix: macos chrome cookie plugin (#217)

Author: wendajiang

Cargo.toml

@@ -22,7 +22,6 @@ clap = { version = "4.5", features = ["cargo", "derive"] }
 colored = "3.1.1"
 dirs = "6.0.0"
 env_logger = "0.11.6"
-keyring = "3.6.3"
 log = "0.4.29"
 openssl = "0.10"
 pyo3 = { version = "0.27", optional = true }
@@ -37,9 +36,10 @@ clap_complete = "4.5"
 thiserror = "2.0.18"
 unicode-width = "0.2"
 notify = "8.2.0"
+rookie = "0.5.6"
 
 [dependencies.diesel]
-version = "2.2.12"
+version = "2.3.6"
 features = ["sqlite"]
 
 [dependencies.reqwest]

src/err.rs

@@ -67,8 +67,6 @@ pub enum Error {
     #[error(transparent)]
     Anyhow(#[from] anyhow::Error),
     #[error(transparent)]
-    Keyring(#[from] keyring::Error),
-    #[error(transparent)]
     OpenSSL(#[from] openssl::error::ErrorStack),
     #[cfg(feature = "pym")]
     #[error(transparent)]

src/plugins/chrome.rs

@@ -1,32 +1,6 @@
-use crate::{Error, Result, cache};
-use anyhow::anyhow;
-use diesel::prelude::*;
-use keyring::Entry;
-use openssl::{hash, pkcs5, symm};
+use crate::{Error, Result};
 use std::{collections::HashMap, fmt::Display};
 
-/// LeetCode Cookies Schema
-mod schema {
-    table! {
-        cookies (host_key) {
-            encrypted_value -> Binary,
-            host_key -> Text,
-            name -> Text,
-        }
-    }
-}
-
-/// Please make sure the order
-///
-/// The order between table and struct must be same.
-#[derive(Queryable, Debug, Clone)]
-struct Cookies {
-    pub encrypted_value: Vec<u8>,
-    #[allow(dead_code)]
-    pub host_key: String,
-    pub name: String,
-}
-
 /// Spawn cookies to cookie format
 #[derive(Debug)]
 pub struct Ident {
@@ -54,38 +28,13 @@ pub fn cookies() -> Result<Ident> {
         });
     }
 
-    // If doesn't config SESSION and csrftoken
-    use self::schema::cookies::dsl::*;
-    trace!("Derive cookies from google chrome...");
-
-    let home = dirs::home_dir().ok_or(Error::NoneError)?;
-    let p = match std::env::consts::OS {
-        "macos" => home.join("Library/Application Support/Google/Chrome/Default/Cookies"),
-        "linux" => home.join(".config/google-chrome/Default/Cookies"),
-        _ => panic!("Opps...only works on OSX or Linux now..."),
-    };
-
-    debug!("Chrome Cookies path is {:?}", &p);
-    let mut conn = cache::conn(p.to_string_lossy().to_string());
-    let res = cookies
-        .filter(host_key.like(format!("#{}", ccfg.site)))
-        .load::<Cookies>(&mut conn)
-        .expect("Loading cookies from google chrome failed.");
-
-    debug!("res {:?}", &res);
-    if res.is_empty() {
-        return Err(Error::CookieError);
-    }
-
-    // Get system password
-    let ring = Entry::new("Chrome Safe Storage", "Chrome")?;
-    let pass = ring.get_password().expect("Get Password failed");
-
-    // Decode cookies
     let mut m: HashMap<String, String> = HashMap::new();
-    for c in res {
+
+    let domains = vec![format!("{}", ccfg.site)];
+    let cookies = rookie::chrome(Some(domains)).unwrap();
+    for c in cookies {
         if (c.name == "csrftoken") || (c.name == "LEETCODE_SESSION") {
-            m.insert(c.name, decode_cookies(&pass, c.encrypted_value)?);
+            m.insert(c.name, c.value);
         }
     }
 
@@ -98,56 +47,3 @@ pub fn cookies() -> Result<Ident> {
     })
 }
 
-/// Decode cookies from chrome
-fn decode_cookies(pass: &str, v: Vec<u8>) -> Result<String> {
-    let mut key = [0_u8; 16];
-    match std::env::consts::OS {
-        "macos" => {
-            pkcs5::pbkdf2_hmac(
-                pass.as_bytes(),
-                b"saltysalt",
-                1003,
-                hash::MessageDigest::sha1(),
-                &mut key,
-            )
-            .expect("pbkdf2 hmac went error.");
-        }
-        "linux" => {
-            pkcs5::pbkdf2_hmac(
-                b"peanuts",
-                b"saltysalt",
-                1,
-                hash::MessageDigest::sha1(),
-                &mut key,
-            )
-            .expect("pbkdf2 hmac went error.");
-        }
-        _ => return Err(anyhow!("only supports OSX or Linux for now").into()),
-    }
-
-    chrome_decrypt(v, key)
-}
-
-/// Decrypt chrome cookie value with aes-128-cbc
-fn chrome_decrypt(v: Vec<u8>, key: [u8; 16]) -> Result<String> {
-    // <space>: \u16
-    let iv = vec![32_u8; 16];
-    let mut decrypter = symm::Crypter::new(
-        symm::Cipher::aes_128_cbc(),
-        symm::Mode::Decrypt,
-        &key,
-        Some(&iv),
-    )?;
-
-    let data_len = v.len() - 3;
-    let block_size = symm::Cipher::aes_128_cbc().block_size();
-    let mut plaintext = vec![0; data_len + block_size];
-
-    decrypter.pad(false);
-
-    let count = decrypter.update(&v[3..], &mut plaintext)?;
-    decrypter.finalize(&mut plaintext[count..])?;
-    plaintext.retain(|x| x >= &20_u8);
-
-    Ok(String::from_utf8_lossy(&plaintext.to_vec()).to_string())
-}