Enhance/warning architecture#478

src/main.rs

@@ -23,14 +23,15 @@ use hhmmss::Hhmmss;
 use pbr::ProgressBar;
 use serde_json::Value;
 use std::cmp::Ordering;
-use std::ffi::OsStr;
+use std::ffi::{OsStr, OsString};
 use std::fmt::Display;
 use std::fs::create_dir;
 use std::io::BufWriter;
 use std::path::Path;
 use std::sync::Arc;
 use std::time::SystemTime;
 use std::{
+    env,
     fs::{self, File},
     path::PathBuf,
     vec,
@@ -40,7 +41,7 @@ use tokio::spawn;
 use tokio::task::JoinHandle;
 
 #[cfg(target_os = "windows")]
-use {is_elevated::is_elevated, std::env};
+use is_elevated::is_elevated;
 
 // 一度にtimelineやdetectionを実行する行数
 const MAX_DETECT_RECORDS: usize = 5000;
@@ -81,6 +82,17 @@ impl App {
                 &analysis_start_time.day().to_owned()
             ));
         }
+
+        if !self.is_matched_architecture_and_binary() {
+            AlertMessage::alert(
+                &mut BufWriter::new(std::io::stderr().lock()),
+                "The hayabusa version you ran does not match your PC architecture.\n Please use the correct architecture. (Binary ending in -x64.exe for 64-bit and -x86.exe for 32-bit.)",
+            )
+            .ok();
+            println!();
+            return;
+        }
+
         if configs::CONFIG
             .read()
             .unwrap()
@@ -709,6 +721,22 @@ impl App {
             Ok("You currently have the latest rules.".to_string())
         }
     }
+
+    /// check architecture
+    fn is_matched_architecture_and_binary(&self) -> bool {
+        if cfg!(target_os = "windows") {
+            let is_processor_arch_32bit = env::var_os("PROCESSOR_ARCHITECTURE")
+                .unwrap_or_default()
+                .eq("x86");
+            // PROCESSOR_ARCHITEW6432は32bit環境には存在しないため、環境変数存在しなかった場合は32bit環境であると判断する
+            let not_wow_flag = env::var_os("PROCESSOR_ARCHITEW6432")
+                .unwrap_or_else(|| OsString::from("x86"))
+                .eq("x86");
+            return (cfg!(target_pointer_width = "64") && !is_processor_arch_32bit)
+                || (cfg!(target_pointer_width = "32") && is_processor_arch_32bit && not_wow_flag);
+        }
+        true
+    }
 }
 
 #[cfg(test)]