Update diff.rs

Author: boludoz

objdiff-gui/src/views/diff.rs

@@ -4,7 +4,7 @@ use objdiff_core::{
     diff::{
         DiffObjConfig, ObjectDiff, SymbolDiff,
         data::BYTES_PER_ROW,
-        display::{ContextItem, HoverItem, HoverItemColor, SymbolFilter, SymbolNavigationKind},
+        display::{ContextItem, HoverItem, HoverItemColor, SymbolFilter, SymbolNavigationKind, display_row},
     },
     obj::{Object, Symbol},
 };
@@ -64,6 +64,26 @@ impl<'a> DiffColumnContext<'a> {
     pub fn id(&self) -> Option<&str> { self.symbol.map(|(symbol, _, _)| symbol.name.as_str()) }
 }
 
+/// Obtains the assembly text for a given symbol diff, suitable for copying to clipboard.
+fn get_asm_text(obj: &Object, symbol_diff: &SymbolDiff, symbol_idx: usize, diff_config: &DiffObjConfig) -> String {
+    let mut asm_text = String::new();
+    
+    for ins_row in &symbol_diff.instruction_rows {
+        let mut line = String::new();
+        let result = display_row(obj, symbol_idx, ins_row, diff_config, |segment| {
+            line.push_str(&segment.text);
+            Ok(())
+        });
+        
+        if result.is_ok() {
+            asm_text.push_str(&line);
+            asm_text.push('\n');
+        }
+    }
+    
+    asm_text
+}
+
 #[must_use]
 pub fn diff_view_ui(
     ui: &mut Ui,
@@ -208,16 +228,35 @@ pub fn diff_view_ui(
 
             // Third row
             if left_ctx.has_symbol() && right_ctx.has_symbol() {
-                if (state.current_view == View::FunctionDiff
-                    && ui
-                        .button("Change target")
-                        .on_hover_text_at_pointer("Choose a different symbol to use as the target")
-                        .clicked()
-                    || hotkeys::consume_change_target_shortcut(ui.ctx()))
-                    && let Some(symbol_ref) = state.symbol_state.right_symbol.as_ref()
-                {
-                    ret = Some(DiffViewAction::SelectingLeft(symbol_ref.clone()));
-                }
+                ui.horizontal(|ui| {
+                    if state.current_view == View::FunctionDiff
+                        && ui
+                            .button("Change target")
+                            .on_hover_text_at_pointer("Choose a different symbol to use as the target")
+                            .clicked()
+                        || hotkeys::consume_change_target_shortcut(ui.ctx())
+                    {
+                        if let Some(symbol_ref) = state.symbol_state.right_symbol.as_ref() {
+                            ret = Some(DiffViewAction::SelectingLeft(symbol_ref.clone()));
+                        }
+                    }
+                    
+                    // Copy target ASM button.
+                    if state.current_view == View::FunctionDiff {
+                        if let Some((_, symbol_diff, symbol_idx)) = left_ctx.symbol {
+                            if let Some((obj, _)) = left_ctx.obj {
+                                if ui
+                                    .button("Copy ASM")
+                                    .on_hover_text_at_pointer("Copy assembly to clipboard")
+                                    .clicked()
+                                {
+                                    let asm_text = get_asm_text(obj, symbol_diff, symbol_idx, diff_config);
+                                    ui.ctx().copy_text(asm_text);
+                                }
+                            }
+                        }
+                    }
+                });
             } else if left_ctx.status.success && !left_ctx.has_symbol() {
                 ui.horizontal(|ui| {
                     let mut search = state.search.clone();
@@ -374,6 +413,20 @@ pub fn diff_view_ui(
                         {
                             ret = Some(DiffViewAction::SelectingRight(symbol_ref.clone()));
                         }
+                        
+                        /// Copy base ASM button.
+                        if let Some((_, symbol_diff, symbol_idx)) = right_ctx.symbol {
+                            if let Some((obj, _)) = right_ctx.obj {
+                                if ui
+                                    .button("Copy ASM")
+                                    .on_hover_text_at_pointer("Copy assembly to clipboard")
+                                    .clicked()
+                                {
+                                    let asm_text = get_asm_text(obj, symbol_diff, symbol_idx, diff_config);
+                                    ui.ctx().copy_text(asm_text);
+                                }
+                            }
+                        }
                     }
                 } else if right_ctx.status.success && !right_ctx.has_symbol() {
                     let mut search = state.search.clone();