From 6c94811ba217acc3be17b5057928a1adaf7f0ab0 Mon Sep 17 00:00:00 2001
From: Joe Rozner <joe@deadbytes.net>
Date: Thu, 27 Jan 2022 23:53:31 -0800
Subject: rust: fix semantic class for flag write cb

Current implementation returns the flag write id rather than the flag
class id
---
 rust/src/architecture.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'rust/src')

diff --git a/rust/src/architecture.rs b/rust/src/architecture.rs
index 6df8b19c..7ce672b8 100644
--- a/rust/src/architecture.rs
+++ b/rust/src/architecture.rs
@@ -1729,7 +1729,8 @@ where
         let custom_arch = unsafe { &*(ctxt as *mut A) };
         custom_arch
             .flag_write_from_id(write_type)
-            .map(|w| w.id())
+            .map(|w| w.class())
+            .and_then(|c| c.map(|c| c.id()))
             .unwrap_or(0)
     }
 
-- 
cgit v1.3.1