Function convert_registry_path

archive/ntreg/src/registry_key.rs:501–544

(path: &str)

}

fn convert_registry_path(path: &str) -> Result<String, NtStatusError> {
    if path == r"\Registry\Machine" || path.starts_with(r"\Registry\Machine\") ||
        path == r"\Registry\User" || path.starts_with(r"\Registry\User\") {
        return Ok(path.to_string());
    }

    let hive : &str;
    let mut key_path = String::from("");
    if path.contains('\\') {
        hive = path.split('\\').next().unwrap();
        key_path = path.split('\\').skip(1).collect::<Vec<&str>>().join("\\");
    }
    else {
        hive = path;
    }

    let mut path: String = match hive.to_uppercase().as_str() {
        "HKEY_CLASSES_ROOT" | "HKCR" => {
            format!("\\Registry\\Machine\\Software\\Classes\\{}", key_path)
        },
        "HKEY_CURRENT_USER" | "HKCU" => {
            let user = NtCurrentUserInfo::new()?;
            format!("\\Registry\\User\\{}\\{}", user.sid, key_path)
        },
        "HKEY_LOCAL_MACHINE" | "HKLM" => {
            format!("\\Registry\\Machine\\{}", key_path)
        },
        "HKEY_USERS" | "HKU" => {
            format!("\\Registry\\User\\{}", key_path)
        },
        "HKEY_CURRENT_CONFIG" | "HKCC" => {
            format!("\\Registry\\Machine\\System\\CurrentControlSet\\Hardware Profiles\\Current\\{}", key_path)
        },
        _ => {
            return Err(NtStatusError::new(STATUS_OBJECT_PATH_SYNTAX_BAD, "Invalid registry path."));
        }
    };

    if path.ends_with('\\') {
        path.pop();
    }

    Ok(path)
}