修改获取disk序列号的方法,改为使用逻辑卷的序列号。因为可能由于权限的原因获取不到磁盘的序列号
This commit is contained in:
parent
db28e8525c
commit
9c2fbe847e
|
@ -7,6 +7,8 @@ use block_modes::block_padding::Pkcs7;
|
||||||
use base64::{Engine as _, engine::general_purpose};
|
use base64::{Engine as _, engine::general_purpose};
|
||||||
type Aes128Cbc = Cbc<Aes128, Pkcs7>;
|
type Aes128Cbc = Cbc<Aes128, Pkcs7>;
|
||||||
|
|
||||||
|
use std::ffi::CString;
|
||||||
|
use winapi::um::fileapi::GetVolumeInformationA;
|
||||||
use std::fmt;
|
use std::fmt;
|
||||||
|
|
||||||
#[derive(Debug)]
|
#[derive(Debug)]
|
||||||
|
@ -24,22 +26,49 @@ impl fmt::Display for EncryptErr {
|
||||||
|
|
||||||
impl std::error::Error for EncryptErr {}
|
impl std::error::Error for EncryptErr {}
|
||||||
|
|
||||||
|
// pub fn get_key() -> String {
|
||||||
|
// let output = Command::new("wmic")
|
||||||
|
// .args(&["DISKDRIVE", "get", "serialnumber"])
|
||||||
|
// .output()
|
||||||
|
// .expect("Failed to execute command");
|
||||||
|
|
||||||
|
// let serial = String::from_utf8_lossy(&output.stdout)
|
||||||
|
// .lines()
|
||||||
|
// .nth(1) // 获取第二行,通常是序列号
|
||||||
|
// .unwrap()
|
||||||
|
// .trim()
|
||||||
|
// .to_string();
|
||||||
|
// log::trace!("hdd serial is {}", serial);
|
||||||
|
// serial
|
||||||
|
// }
|
||||||
pub fn get_key() -> String {
|
pub fn get_key() -> String {
|
||||||
let output = Command::new("wmic")
|
let drive: &str = "C:\\";
|
||||||
.args(&["DISKDRIVE", "get", "serialnumber"])
|
let mut volume_name: [i8; 100] = [0; 100];
|
||||||
.output()
|
let mut file_system_name: [i8; 100] = [0; 100];
|
||||||
.expect("Failed to execute command");
|
let mut volume_serial_number: u32 = 0;
|
||||||
|
let mut max_component_length: u32 = 0;
|
||||||
|
let mut file_system_flags: u32 = 0;
|
||||||
|
let tstr = CString::new(drive).unwrap();
|
||||||
|
let success = unsafe {
|
||||||
|
GetVolumeInformationA(
|
||||||
|
tstr.as_ptr(),
|
||||||
|
volume_name.as_mut_ptr(),
|
||||||
|
volume_name.len() as u32,
|
||||||
|
&mut volume_serial_number,
|
||||||
|
&mut max_component_length,
|
||||||
|
&mut file_system_flags,
|
||||||
|
file_system_name.as_mut_ptr(),
|
||||||
|
file_system_name.len() as u32,
|
||||||
|
)
|
||||||
|
};
|
||||||
|
|
||||||
let serial = String::from_utf8_lossy(&output.stdout)
|
if success != 0 {
|
||||||
.lines()
|
return volume_serial_number.to_string()
|
||||||
.nth(1) // 获取第二行,通常是序列号
|
} else {
|
||||||
.unwrap()
|
log::warn!("Failed to get volume information.");
|
||||||
.trim()
|
return "".to_string();
|
||||||
.to_string();
|
}
|
||||||
log::trace!("hdd serial is {}", serial);
|
|
||||||
serial
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pub fn is_base64(encrypted_data: &str) -> bool {
|
pub fn is_base64(encrypted_data: &str) -> bool {
|
||||||
match general_purpose::STANDARD.decode(encrypted_data) {
|
match general_purpose::STANDARD.decode(encrypted_data) {
|
||||||
Ok(_) => true,
|
Ok(_) => true,
|
||||||
|
|
Loading…
Reference in New Issue
Block a user