Fix: no panic when check magic_number or version fail (tikv#11)

brson · Feb 18, 2020 · ccc914e · ccc914e
1 parent 16f7f7a
commit ccc914e
Showing 1 changed file with 16 additions and 2 deletions.
diff --git a/src/tiflash_ffi/invoke.rs b/src/tiflash_ffi/invoke.rs
@@ -176,8 +176,22 @@ impl TiFlashServerHelper {
 
     pub fn check(&self) {
         assert_eq!(std::mem::align_of::<Self>(), std::mem::align_of::<u64>());
-        assert_eq!(self.magic_number, 0x13579BDF);
-        assert_eq!(self.version, 1);
+        const MAGIC_NUMBER: u32 = 0x13579BDF;
+        const VERSION: u32 = 1;
+
+        if self.magic_number != MAGIC_NUMBER {
+            eprintln!(
+                "TiFlash Proxy FFI magic number not match: expect {} got {}",
+                MAGIC_NUMBER, self.magic_number
+            );
+            std::process::exit(-1);
+        } else if self.version != VERSION {
+            eprintln!(
+                "TiFlash Proxy FFI version not match: expect {} got {}",
+                VERSION, self.version
+            );
+            std::process::exit(-1);
+        }
     }
 
     pub fn handle_admin_raft_cmd(