summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authoriximeow <me@iximeow.net>2019-03-20 03:14:39 -0700
committeriximeow <me@iximeow.net>2020-01-12 17:07:40 -0800
commite380a481b8f129b50ee49be818ad889bedcd3b2f (patch)
tree397264f8f2ea9c0d7402afefc833b7e0eb22ece7
parentdad3bed68d442f23e484a963eda9a6812e67c7c4 (diff)
update and impl new display-related triats
-rw-r--r--Cargo.toml1
-rw-r--r--src/display.rs66
-rw-r--r--src/lib.rs1
3 files changed, 45 insertions, 23 deletions
diff --git a/Cargo.toml b/Cargo.toml
index 58e3a3c..384bae6 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -11,6 +11,7 @@ A rust msp430 decoder (specifically the microcorruption flavor)
[dependencies]
yaxpeax-arch = { path = "../../yaxpeax-arch" }
+"termion" = "1.4.0"
[[test]]
name = "test"
diff --git a/src/display.rs b/src/display.rs
index 7864248..b51ae9b 100644
--- a/src/display.rs
+++ b/src/display.rs
@@ -1,25 +1,38 @@
-use std;
-use std::fmt::{Display, Formatter};
+use ::{MSP430, Operand, Opcode, Instruction, Width};
-use ::{Operand, Opcode, Instruction, Width};
+use std::collections::HashMap;
+use std::fmt::{Display, Formatter};
+use std::hash::Hash;
+use std::rc::Rc;
+use std;
+use termion::color;
+use yaxpeax_arch::{Arch, Colorize, ColorSettings, LengthedInstruction, ShowContextual};
impl Display for Instruction {
fn fmt(&self, f: &mut Formatter) -> Result<(), std::fmt::Error> {
- write!(f, "{}", self.opcode)?;
+ let mut s = String::new();
+ self.contextualize(None, 0, None, &mut s).unwrap();
+ write!(f, "{}", s)
+ }
+}
+
+impl <T: std::fmt::Write> ShowContextual<<MSP430 as Arch>::Address, [Option<String>], T> for Instruction {
+ fn contextualize(&self, colors: Option<&ColorSettings>, address: <MSP430 as Arch>::Address, context: Option<&[Option<String>]>, out: &mut T) -> std::fmt::Result {
+ write!(out, "{}", self.opcode)?;
match self.op_width {
- Width::B => { write!(f, ".b")? },
+ Width::B => { write!(out, ".b")? },
Width::W => { }
};
match self.operands[0] {
Operand::Nothing => return Ok(()),
x @ _ => {
- write!(f, " {}", x)?;
+ write!(out, " {}", x)?;
}
};
match self.operands[1] {
Operand::Nothing => return Ok(()),
x @ _ => {
- write!(f, ", {}", x)?;
+ write!(out, ", {}", x)?;
}
};
Ok(())
@@ -64,6 +77,14 @@ impl Display for Opcode {
impl Display for Operand {
fn fmt(&self, f: &mut Formatter) -> Result<(), std::fmt::Error> {
+ let mut s = String::new();
+ self.colorize(None, &mut s).unwrap();
+ write!(f, "{}", s)
+ }
+}
+
+impl <T: std::fmt::Write> Colorize<T> for Operand {
+ fn colorize(&self, colors: Option<&ColorSettings>, out: &mut T) -> std::fmt::Result {
fn signed_hex(num: i16) -> String {
if num >= 0 {
format!("+{:#x}", num)
@@ -73,51 +94,50 @@ impl Display for Operand {
}
match self {
Operand::Register(reg) => {
- write!(f, "R{}", reg)
+ write!(out, "R{}", reg)
},
Operand::Indexed(reg, offset) => {
- write!(f, "{}(R{})", signed_hex(*offset as i16), reg)
+ write!(out, "{}(R{})", signed_hex(*offset as i16), reg)
},
Operand::RegisterIndirect(reg) => {
- write!(f, "@R{}", reg)
+ write!(out, "@R{}", reg)
},
Operand::IndirectAutoinc(reg) => {
- write!(f, "@R{}+", reg)
+ write!(out, "@R{}+", reg)
},
Operand::Offset(offset) => {
- write!(f, "${}", signed_hex(*offset as i16))
+ write!(out, "${}", signed_hex(*offset as i16))
},
Operand::Symbolic(offset) => {
- write!(f, "{}(PC)", signed_hex(*offset as i16))
+ write!(out, "{}(PC)", signed_hex(*offset as i16))
},
Operand::Immediate(imm) => {
- write!(f, "#0x{:x}", imm)
+ write!(out, "#0x{:x}", imm)
},
Operand::Absolute(offset) => {
- write!(f, "&0x{:x}", offset)
+ write!(out, "&0x{:x}", offset)
},
Operand::Const4 => {
- write!(f, "4")
+ write!(out, "4")
},
Operand::Const8 => {
- write!(f, "8")
+ write!(out, "8")
},
Operand::Const0 => {
- write!(f, "0")
+ write!(out, "0")
},
Operand::Const1 => {
- write!(f, "1")
+ write!(out, "1")
},
Operand::Const2 => {
- write!(f, "2")
+ write!(out, "2")
},
Operand::ConstNeg1 => {
- write!(f, "-1")
+ write!(out, "-1")
},
Operand::Nothing => {
- write!(f, "<No Operand>")
+ write!(out, "<No Operand>")
}
}
}
}
-
diff --git a/src/lib.rs b/src/lib.rs
index c215b68..81ceceb 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,4 +1,5 @@
extern crate yaxpeax_arch;
+extern crate termion;
use yaxpeax_arch::{Arch, Decodable, LengthedInstruction};