From fb9ead64e6c7d96c02627b01b6a85921e2d8f7a4 Mon Sep 17 00:00:00 2001 From: iximeow Date: Sat, 15 Aug 2020 13:15:29 -0700 Subject: add RegSpec constructors, consts, and const fns --- src/long_mode/mod.rs | 177 +++++++++++++++++++++++++++++++++++++--------- src/protected_mode/mod.rs | 151 ++++++++++++++++++++++++++++++--------- 2 files changed, 258 insertions(+), 70 deletions(-) (limited to 'src') diff --git a/src/long_mode/mod.rs b/src/long_mode/mod.rs index e8bc646..94e34ae 100644 --- a/src/long_mode/mod.rs +++ b/src/long_mode/mod.rs @@ -60,6 +60,9 @@ pub enum ConditionCode { #[allow(non_snake_case)] impl RegSpec { + /// the register `rip`. this register is in the class `rip`, which contains only it. + pub const RIP: RegSpec = RegSpec::rip(); + pub fn num(&self) -> u8 { self.num } @@ -75,7 +78,7 @@ impl RegSpec { } #[inline] - fn st(num: u8) -> RegSpec { + pub fn st(num: u8) -> RegSpec { if num >= 8 { panic!("invalid x87 reg st({})", num); } @@ -86,6 +89,110 @@ impl RegSpec { } } + /// construct a `RegSpec` for xmm reg `num` + #[inline] + pub fn xmm(num: u8) -> RegSpec { + if num >= 32 { + panic!("invalid x86 xmm reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::X + } + } + + /// construct a `RegSpec` for ymm reg `num` + #[inline] + pub fn ymm(num: u8) -> RegSpec { + if num >= 32 { + panic!("invalid x86 ymm reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::Y + } + } + + /// construct a `RegSpec` for zmm reg `num` + #[inline] + pub fn zmm(num: u8) -> RegSpec { + if num >= 32 { + panic!("invalid x86 zmm reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::Z + } + } + + /// construct a `RegSpec` for qword reg `num` + #[inline] + pub fn q(num: u8) -> RegSpec { + if num >= 16 { + panic!("invalid x86 qword reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::Q + } + } + + /// construct a `RegSpec` for dword reg `num` + #[inline] + pub fn d(num: u8) -> RegSpec { + if num >= 16 { + panic!("invalid x86 dword reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::D + } + } + + /// construct a `RegSpec` for word reg `num` + #[inline] + pub fn w(num: u8) -> RegSpec { + if num >= 16 { + panic!("invalid x86 word reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::W + } + } + + /// construct a `RegSpec` for non-rex byte reg `num` + #[inline] + pub fn rb(num: u8) -> RegSpec { + if num >= 18 { + panic!("invalid x86 rex-byte reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::rB + } + } + + /// construct a `RegSpec` for non-rex byte reg `num` + #[inline] + pub fn b(num: u8) -> RegSpec { + if num >= 8 { + panic!("invalid x86 non-rex byte reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::B + } + } + #[inline] fn from_parts(num: u8, extended: bool, bank: RegisterBank) -> RegSpec { RegSpec { @@ -103,7 +210,7 @@ impl RegSpec { } #[inline] - pub fn rip() -> RegSpec { + pub const fn rip() -> RegSpec { RegSpec { num: 0, bank: RegisterBank::RIP @@ -111,7 +218,7 @@ impl RegSpec { } #[inline] - pub fn eip() -> RegSpec { + pub const fn eip() -> RegSpec { RegSpec { num: 0, bank: RegisterBank::EIP @@ -119,7 +226,7 @@ impl RegSpec { } #[inline] - pub fn eflags() -> RegSpec { + pub const fn eflags() -> RegSpec { RegSpec { num: 0, bank: RegisterBank::EFlags @@ -127,7 +234,7 @@ impl RegSpec { } #[inline] - pub fn rflags() -> RegSpec { + pub const fn rflags() -> RegSpec { RegSpec { num: 0, bank: RegisterBank::RFlags @@ -135,7 +242,7 @@ impl RegSpec { } #[inline] - pub fn rbp() -> RegSpec { + pub const fn rbp() -> RegSpec { RegSpec { num: 5, bank: RegisterBank::Q @@ -143,7 +250,7 @@ impl RegSpec { } #[inline] - pub fn rsp() -> RegSpec { + pub const fn rsp() -> RegSpec { RegSpec { num: 4, bank: RegisterBank::Q @@ -151,7 +258,7 @@ impl RegSpec { } #[inline] - pub fn esp() -> RegSpec { + pub const fn esp() -> RegSpec { RegSpec { num: 4, bank: RegisterBank::D @@ -159,7 +266,7 @@ impl RegSpec { } #[inline] - pub fn sp() -> RegSpec { + pub const fn sp() -> RegSpec { RegSpec { num: 4, bank: RegisterBank::W @@ -167,132 +274,132 @@ impl RegSpec { } #[inline] - pub fn fs() -> RegSpec { + pub const fn fs() -> RegSpec { RegSpec { bank: RegisterBank::S, num: 4 } } #[inline] - pub fn gs() -> RegSpec { + pub const fn gs() -> RegSpec { RegSpec { bank: RegisterBank::S, num: 5 } } #[inline] - pub fn rax() -> RegSpec { + pub const fn rax() -> RegSpec { RegSpec { bank: RegisterBank::Q, num: 0 } } #[inline] - pub fn rcx() -> RegSpec { + pub const fn rcx() -> RegSpec { RegSpec { bank: RegisterBank::Q, num: 1 } } #[inline] - pub fn rdx() -> RegSpec { + pub const fn rdx() -> RegSpec { RegSpec { bank: RegisterBank::Q, num: 2 } } #[inline] - pub fn rsi() -> RegSpec { + pub const fn rsi() -> RegSpec { RegSpec { bank: RegisterBank::Q, num: 6 } } #[inline] - pub fn rdi() -> RegSpec { + pub const fn rdi() -> RegSpec { RegSpec { bank: RegisterBank::Q, num: 7 } } #[inline] - pub fn r8() -> RegSpec { + pub const fn r8() -> RegSpec { RegSpec { bank: RegisterBank::Q, num: 8 } } #[inline] - pub fn r9() -> RegSpec { + pub const fn r9() -> RegSpec { RegSpec { bank: RegisterBank::Q, num: 9 } } #[inline] - pub fn eax() -> RegSpec { + pub const fn eax() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 0 } } #[inline] - pub fn ecx() -> RegSpec { + pub const fn ecx() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 1 } } #[inline] - pub fn edx() -> RegSpec { + pub const fn edx() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 2 } } #[inline] - pub fn ebx() -> RegSpec { + pub const fn ebx() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 3 } } #[inline] - pub fn ax() -> RegSpec { + pub const fn ax() -> RegSpec { RegSpec { bank: RegisterBank::W, num: 0 } } #[inline] - pub fn dx() -> RegSpec { + pub const fn dx() -> RegSpec { RegSpec { bank: RegisterBank::W, num: 2 } } #[inline] - pub fn al() -> RegSpec { + pub const fn al() -> RegSpec { RegSpec { bank: RegisterBank::B, num: 0 } } #[inline] - pub fn cl() -> RegSpec { + pub const fn cl() -> RegSpec { RegSpec { bank: RegisterBank::B, num: 1 } } #[inline] - pub fn dl() -> RegSpec { + pub const fn dl() -> RegSpec { RegSpec { bank: RegisterBank::B, num: 2 } } #[inline] - pub fn ah() -> RegSpec { + pub const fn ah() -> RegSpec { RegSpec { bank: RegisterBank::B, num: 4 } } #[inline] - pub fn ch() -> RegSpec { + pub const fn ch() -> RegSpec { RegSpec { bank: RegisterBank::B, num: 5 } } #[inline] - pub fn r8b() -> RegSpec { + pub const fn r8b() -> RegSpec { RegSpec { bank: RegisterBank::rB, num: 8 } } #[inline] - pub fn zmm0() -> RegSpec { + pub const fn zmm0() -> RegSpec { RegSpec { bank: RegisterBank::Z, num: 0 } } #[inline] - pub fn ymm0() -> RegSpec { + pub const fn ymm0() -> RegSpec { RegSpec { bank: RegisterBank::Y, num: 0 } } #[inline] - pub fn xmm0() -> RegSpec { + pub const fn xmm0() -> RegSpec { RegSpec { bank: RegisterBank::X, num: 0 } } #[inline] - pub fn st0() -> RegSpec { + pub const fn st0() -> RegSpec { RegSpec { bank: RegisterBank::ST, num: 0 } } #[inline] - pub fn mm0() -> RegSpec { + pub const fn mm0() -> RegSpec { RegSpec { bank: RegisterBank::MM, num: 0 } } diff --git a/src/protected_mode/mod.rs b/src/protected_mode/mod.rs index b9d626c..29c2750 100644 --- a/src/protected_mode/mod.rs +++ b/src/protected_mode/mod.rs @@ -60,6 +60,9 @@ pub enum ConditionCode { #[allow(non_snake_case)] impl RegSpec { + /// the register `eip`. this register is in the class `eip`, which contains only it. + pub const EIP: RegSpec = RegSpec::eip(); + pub fn num(&self) -> u8 { self.num } @@ -75,7 +78,7 @@ impl RegSpec { } #[inline] - fn st(num: u8) -> RegSpec { + pub fn st(num: u8) -> RegSpec { if num >= 8 { panic!("invalid x87 reg st({})", num); } @@ -86,6 +89,84 @@ impl RegSpec { } } + /// construct a `RegSpec` for xmm reg `num` + #[inline] + pub fn xmm(num: u8) -> RegSpec { + if num >= 32 { + panic!("invalid x86 xmm reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::X + } + } + + /// construct a `RegSpec` for ymm reg `num` + #[inline] + pub fn ymm(num: u8) -> RegSpec { + if num >= 32 { + panic!("invalid x86 ymm reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::Y + } + } + + /// construct a `RegSpec` for zmm reg `num` + #[inline] + pub fn zmm(num: u8) -> RegSpec { + if num >= 32 { + panic!("invalid x86 zmm reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::Z + } + } + + /// construct a `RegSpec` for dword reg `num` + #[inline] + pub fn d(num: u8) -> RegSpec { + if num >= 8 { + panic!("invalid x86 dword reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::D + } + } + + /// construct a `RegSpec` for word reg `num` + #[inline] + pub fn w(num: u8) -> RegSpec { + if num >= 8 { + panic!("invalid x86 word reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::W + } + } + + /// construct a `RegSpec` for non-rex byte reg `num` + #[inline] + pub fn b(num: u8) -> RegSpec { + if num >= 8 { + panic!("invalid x86 byte reg {}", num); + } + + RegSpec { + num, + bank: RegisterBank::B + } + } + #[inline] fn from_parts(num: u8, bank: RegisterBank) -> RegSpec { RegSpec { @@ -95,7 +176,7 @@ impl RegSpec { } #[inline] - pub fn eip() -> RegSpec { + pub const fn eip() -> RegSpec { RegSpec { num: 0, bank: RegisterBank::EIP @@ -103,7 +184,7 @@ impl RegSpec { } #[inline] - pub fn eflags() -> RegSpec { + pub const fn eflags() -> RegSpec { RegSpec { num: 0, bank: RegisterBank::EFlags @@ -111,162 +192,162 @@ impl RegSpec { } #[inline] - pub fn esp() -> RegSpec { + pub const fn esp() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 4 } } #[inline] - pub fn ebp() -> RegSpec { + pub const fn ebp() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 5 } } #[inline] - pub fn cs() -> RegSpec { + pub const fn cs() -> RegSpec { RegSpec { bank: RegisterBank::S, num: 1 } } #[inline] - pub fn ds() -> RegSpec { + pub const fn ds() -> RegSpec { RegSpec { bank: RegisterBank::S, num: 3 } } #[inline] - pub fn es() -> RegSpec { + pub const fn es() -> RegSpec { RegSpec { bank: RegisterBank::S, num: 0 } } #[inline] - pub fn ss() -> RegSpec { + pub const fn ss() -> RegSpec { RegSpec { bank: RegisterBank::S, num: 2 } } #[inline] - pub fn fs() -> RegSpec { + pub const fn fs() -> RegSpec { RegSpec { bank: RegisterBank::S, num: 4 } } #[inline] - pub fn gs() -> RegSpec { + pub const fn gs() -> RegSpec { RegSpec { bank: RegisterBank::S, num: 5 } } #[inline] - pub fn eax() -> RegSpec { + pub const fn eax() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 0 } } #[inline] - pub fn ecx() -> RegSpec { + pub const fn ecx() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 1 } } #[inline] - pub fn edx() -> RegSpec { + pub const fn edx() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 2 } } #[inline] - pub fn ebx() -> RegSpec { + pub const fn ebx() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 3 } } #[inline] - pub fn esi() -> RegSpec { + pub const fn esi() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 6 } } #[inline] - pub fn edi() -> RegSpec { + pub const fn edi() -> RegSpec { RegSpec { bank: RegisterBank::D, num: 7 } } #[inline] - pub fn ax() -> RegSpec { + pub const fn ax() -> RegSpec { RegSpec { bank: RegisterBank::W, num: 0 } } #[inline] - pub fn cx() -> RegSpec { + pub const fn cx() -> RegSpec { RegSpec { bank: RegisterBank::W, num: 1 } } #[inline] - pub fn dx() -> RegSpec { + pub const fn dx() -> RegSpec { RegSpec { bank: RegisterBank::W, num: 2 } } #[inline] - pub fn bx() -> RegSpec { + pub const fn bx() -> RegSpec { RegSpec { bank: RegisterBank::W, num: 3 } } #[inline] - pub fn sp() -> RegSpec { + pub const fn sp() -> RegSpec { RegSpec { bank: RegisterBank::W, num: 4 } } #[inline] - pub fn bp() -> RegSpec { + pub const fn bp() -> RegSpec { RegSpec { bank: RegisterBank::W, num: 5 } } #[inline] - pub fn si() -> RegSpec { + pub const fn si() -> RegSpec { RegSpec { bank: RegisterBank::W, num: 6 } } #[inline] - pub fn di() -> RegSpec { + pub const fn di() -> RegSpec { RegSpec { bank: RegisterBank::W, num: 7 } } #[inline] - pub fn al() -> RegSpec { + pub const fn al() -> RegSpec { RegSpec { bank: RegisterBank::B, num: 0 } } #[inline] - pub fn cl() -> RegSpec { + pub const fn cl() -> RegSpec { RegSpec { bank: RegisterBank::B, num: 1 } } #[inline] - pub fn dl() -> RegSpec { + pub const fn dl() -> RegSpec { RegSpec { bank: RegisterBank::B, num: 2 } } #[inline] - pub fn ah() -> RegSpec { + pub const fn ah() -> RegSpec { RegSpec { bank: RegisterBank::B, num: 4 } } #[inline] - pub fn ch() -> RegSpec { + pub const fn ch() -> RegSpec { RegSpec { bank: RegisterBank::B, num: 5 } } #[inline] - pub fn zmm0() -> RegSpec { + pub const fn zmm0() -> RegSpec { RegSpec { bank: RegisterBank::Z, num: 0 } } #[inline] - pub fn ymm0() -> RegSpec { + pub const fn ymm0() -> RegSpec { RegSpec { bank: RegisterBank::Y, num: 0 } } #[inline] - pub fn xmm0() -> RegSpec { + pub const fn xmm0() -> RegSpec { RegSpec { bank: RegisterBank::X, num: 0 } } #[inline] - pub fn st0() -> RegSpec { + pub const fn st0() -> RegSpec { RegSpec { bank: RegisterBank::ST, num: 0 } } #[inline] - pub fn mm0() -> RegSpec { + pub const fn mm0() -> RegSpec { RegSpec { bank: RegisterBank::MM, num: 0 } } -- cgit v1.1