Automatically apply RSA patch

src-tauri/src/main.rs

@@ -22,7 +22,6 @@ mod downloader;
 mod file_helpers;
 mod gamebanana;
 mod lang;
-mod metadata_patcher;
 mod proxy;
 mod system_helpers;
 mod unzip;
@@ -119,8 +118,7 @@ fn main() {
         web::web_get,
         gamebanana::get_download_links,
         gamebanana::list_submissions,
-        gamebanana::list_mods,
-        metadata_patcher::patch_metadata
+        gamebanana::list_mods
       ])
       .run(tauri::generate_context!())
       .expect("error while running tauri application");

src-tauri/src/metadata_patcher.rs

@@ -1,153 +0,0 @@
-use regex::Regex;
-use std::{fs, fs::File, fs::OpenOptions, io::Read, io::Write, path::Path};
-
-// For these two functions, a non-zero return value indicates failure.
-extern "C" {
-  fn decrypt_global_metadata(data: *mut u8, size: usize) -> i32;
-  fn encrypt_global_metadata(data: *mut u8, size: usize) -> i32;
-}
-
-#[tauri::command]
-pub fn patch_metadata(metadata_folder: &str) -> bool {
-  let metadata_file = &(metadata_folder.to_owned() + "\\global-metadata-unpatched.dat");
-  // check if metadata_file exists
-  if !Path::new(metadata_file).exists() {
-    println!("Metadata file not found");
-    return false;
-  }
-
-  println!("Patching metadata file: {}", metadata_file);
-  let decrypted = decrypt_metadata(metadata_file);
-  if do_vecs_match(&decrypted, &Vec::new()) {
-    println!("Failed to decrypt metadata file.");
-    return false;
-  }
-
-  let modified = replace_keys(&decrypted);
-  if do_vecs_match(&modified, &Vec::new()) {
-    println!("Failed to replace keys in metadata file.");
-    return false;
-  }
-
-  let encrypted = encrypt_metadata(&modified);
-  if do_vecs_match(&encrypted, &Vec::new()) {
-    println!("Failed to re-encrypt metadata file.");
-    return false;
-  }
-
-  //write encrypted to file
-  let mut file = match OpenOptions::new()
-    .create(true)
-    .write(true)
-    .open(&(metadata_folder.to_owned() + "\\global-metadata-patched.dat"))
-  {
-    Ok(file) => file,
-    Err(e) => {
-      println!("Failed to open global-metadata: {}", e);
-      return false;
-    }
-  };
-
-  file.write_all(&encrypted).unwrap();
-
-  true
-}
-
-fn decrypt_metadata(file_path: &str) -> Vec<u8> {
-  let mut file = match File::open(file_path) {
-    Ok(file) => file,
-    Err(e) => {
-      println!("Failed to open global-metadata: {}", e);
-      return Vec::new();
-    }
-  };
-  let mut data = Vec::new();
-
-  // Read metadata file
-  match file.read_to_end(&mut data) {
-    Ok(_) => (),
-    Err(e) => {
-      println!("Failed to read global-metadata: {}", e);
-      return Vec::new();
-    }
-  }
-
-  // Decrypt metadata file
-  let success = unsafe { decrypt_global_metadata(data.as_mut_ptr(), data.len()) } == 0;
-  if success {
-    println!("Successfully decrypted global-metadata");
-    data
-  } else {
-    println!("Failed to decrypt global-metadata");
-    Vec::new()
-  }
-}
-
-fn replace_keys(data: &[u8]) -> Vec<u8> {
-  let mut new_data = String::new();
-
-  unsafe {
-    let data_str = String::from_utf8_unchecked(data.to_vec());
-
-    let re = Regex::new(r"<RSAKeyValue>((.|\n|\r)*?)</RSAKeyValue>").unwrap();
-    let matches = re.find_iter(&data_str);
-
-    // dispatch key is index 3
-    // password key is index 2
-
-    for (i, rmatch) in matches.enumerate() {
-      let key = rmatch.as_str();
-
-      if i == 2 {
-        println!("Replacing password key");
-        new_data = replace_rsa_key(&data_str, key, "passwordKey.txt");
-      } else if i == 3 {
-        println!("Replacing dispatch key");
-        new_data = replace_rsa_key(&new_data, key, "dispatchKey.txt");
-      }
-    }
-  }
-
-  return new_data.as_bytes().to_vec();
-}
-
-fn replace_rsa_key(old_data: &str, to_replace: &str, file_name: &str) -> String {
-  // Read dispatch key file
-  unsafe {
-    // Get key path from current directory
-    let key_file_path = std::env::current_dir()
-      .unwrap()
-      .join("keys")
-      .join(file_name);
-
-    let key_data = match fs::read(&key_file_path) {
-      Ok(file) => file,
-      Err(e) => {
-        println!("Failed to open {}: {}", key_file_path.to_str().unwrap(), e);
-        return String::new();
-      }
-    };
-
-    let new_key = String::from_utf8_unchecked(key_data.to_vec());
-
-    // Replace old key with new key
-    old_data.replace(to_replace, &new_key)
-  }
-}
-
-fn encrypt_metadata(old_data: &[u8]) -> Vec<u8> {
-  let mut data = old_data.to_vec();
-
-  let success = unsafe { encrypt_global_metadata(data.as_mut_ptr(), data.len()) } == 0;
-  if success {
-    println!("Successfully encrypted global-metadata");
-    data
-  } else {
-    println!("Failed to encrypt global-metadata");
-    Vec::new()
-  }
-}
-
-fn do_vecs_match<T: PartialEq>(a: &Vec<T>, b: &Vec<T>) -> bool {
-  a == b
-}