Cultivation/src-tauri/src/metadata_patcher.rs

use regex::Regex;
use std::{fs, path::Path, fs::File, fs::OpenOptions, io::Read, io::Write};

// 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
}