use std::borrow::Cow;
use std::fmt;
use std::str;
#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};
#[repr(u32)]
#[non_exhaustive]
#[derive(Clone, Copy, Debug, Eq, Hash, Ord, PartialEq, PartialOrd, Default)]
pub enum CpuFamily {
#[default]
Unknown = 0,
Intel32 = 1,
Amd64 = 2,
Arm32 = 3,
Arm64 = 4,
Ppc32 = 5,
Ppc64 = 6,
Mips32 = 7,
Mips64 = 8,
Arm64_32 = 9,
Wasm32 = 10,
}
impl CpuFamily {
pub fn pointer_size(self) -> Option<usize> {
match self {
CpuFamily::Unknown => None,
CpuFamily::Wasm32 => Some(4),
CpuFamily::Amd64
| CpuFamily::Arm64
| CpuFamily::Ppc64
| CpuFamily::Mips64
| CpuFamily::Arm64_32 => Some(8),
CpuFamily::Intel32 | CpuFamily::Arm32 | CpuFamily::Ppc32 | CpuFamily::Mips32 => Some(4),
}
}
pub fn instruction_alignment(self) -> Option<u64> {
match self {
CpuFamily::Wasm32 => Some(4),
CpuFamily::Arm32 => Some(2),
CpuFamily::Arm64 | CpuFamily::Arm64_32 => Some(4),
CpuFamily::Ppc32 | CpuFamily::Mips32 | CpuFamily::Mips64 => Some(4),
CpuFamily::Ppc64 => Some(8),
CpuFamily::Intel32 | CpuFamily::Amd64 => None,
CpuFamily::Unknown => None,
}
}
pub fn ip_register_name(self) -> Option<&'static str> {
match self {
CpuFamily::Intel32 => Some("eip"),
CpuFamily::Amd64 => Some("rip"),
CpuFamily::Arm32 | CpuFamily::Arm64 | CpuFamily::Arm64_32 => Some("pc"),
CpuFamily::Ppc32 | CpuFamily::Ppc64 => Some("srr0"),
CpuFamily::Mips32 | CpuFamily::Mips64 => Some("pc"),
CpuFamily::Wasm32 => None,
CpuFamily::Unknown => None,
}
}
}
#[derive(Debug)]
pub struct UnknownArchError;
impl fmt::Display for UnknownArchError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "unknown architecture")
}
}
impl std::error::Error for UnknownArchError {}
#[repr(u32)]
#[non_exhaustive]
#[allow(missing_docs)]
#[derive(Clone, Copy, Debug, Eq, Hash, Ord, PartialEq, PartialOrd, Default)]
pub enum Arch {
#[default]
Unknown = 0,
X86 = 101,
X86Unknown = 199,
Amd64 = 201,
Amd64h = 202,
Amd64Unknown = 299,
Arm = 301,
ArmV5 = 302,
ArmV6 = 303,
ArmV6m = 304,
ArmV7 = 305,
ArmV7f = 306,
ArmV7s = 307,
ArmV7k = 308,
ArmV7m = 309,
ArmV7em = 310,
ArmUnknown = 399,
Arm64 = 401,
Arm64V8 = 402,
Arm64e = 403,
Arm64Unknown = 499,
Ppc = 501,
Ppc64 = 601,
Mips = 701,
Mips64 = 801,
Arm64_32 = 901,
Arm64_32V8 = 902,
Arm64_32Unknown = 999,
Wasm32 = 1001,
}
impl Arch {
pub fn from_u32(val: u32) -> Arch {
match val {
0 => Arch::Unknown,
1 | 101 => Arch::X86,
199 => Arch::X86Unknown,
2 | 201 => Arch::Amd64,
3 | 202 => Arch::Amd64h,
299 => Arch::Amd64Unknown,
4 | 301 => Arch::Arm,
5 | 302 => Arch::ArmV5,
6 | 303 => Arch::ArmV6,
7 | 304 => Arch::ArmV6m,
8 | 305 => Arch::ArmV7,
9 | 306 => Arch::ArmV7f,
10 | 307 => Arch::ArmV7s,
11 | 308 => Arch::ArmV7k,
12 | 309 => Arch::ArmV7m,
13 | 310 => Arch::ArmV7em,
399 => Arch::ArmUnknown,
14 | 401 => Arch::Arm64,
15 | 402 => Arch::Arm64V8,
16 | 403 => Arch::Arm64e,
499 => Arch::Arm64Unknown,
17 | 501 => Arch::Ppc,
18 | 601 => Arch::Ppc64,
701 => Arch::Mips,
801 => Arch::Mips64,
901 => Arch::Arm64_32,
902 => Arch::Arm64_32V8,
999 => Arch::Arm64_32Unknown,
1001 => Arch::Wasm32,
_ => Arch::Unknown,
}
}
pub fn cpu_family(self) -> CpuFamily {
match self {
Arch::Unknown => CpuFamily::Unknown,
Arch::X86 | Arch::X86Unknown => CpuFamily::Intel32,
Arch::Amd64 | Arch::Amd64h | Arch::Amd64Unknown => CpuFamily::Amd64,
Arch::Arm64 | Arch::Arm64V8 | Arch::Arm64e | Arch::Arm64Unknown => CpuFamily::Arm64,
Arch::Arm
| Arch::ArmV5
| Arch::ArmV6
| Arch::ArmV6m
| Arch::ArmV7
| Arch::ArmV7f
| Arch::ArmV7s
| Arch::ArmV7k
| Arch::ArmV7m
| Arch::ArmV7em
| Arch::ArmUnknown => CpuFamily::Arm32,
Arch::Ppc => CpuFamily::Ppc32,
Arch::Ppc64 => CpuFamily::Ppc64,
Arch::Mips => CpuFamily::Mips32,
Arch::Mips64 => CpuFamily::Mips64,
Arch::Arm64_32 | Arch::Arm64_32V8 | Arch::Arm64_32Unknown => CpuFamily::Arm64_32,
Arch::Wasm32 => CpuFamily::Wasm32,
}
}
pub fn name(self) -> &'static str {
match self {
Arch::Unknown => "unknown",
Arch::Wasm32 => "wasm32",
Arch::X86 => "x86",
Arch::X86Unknown => "x86_unknown",
Arch::Amd64 => "x86_64",
Arch::Amd64h => "x86_64h",
Arch::Amd64Unknown => "x86_64_unknown",
Arch::Arm64 => "arm64",
Arch::Arm64V8 => "arm64v8",
Arch::Arm64e => "arm64e",
Arch::Arm64Unknown => "arm64_unknown",
Arch::Arm => "arm",
Arch::ArmV5 => "armv5",
Arch::ArmV6 => "armv6",
Arch::ArmV6m => "armv6m",
Arch::ArmV7 => "armv7",
Arch::ArmV7f => "armv7f",
Arch::ArmV7s => "armv7s",
Arch::ArmV7k => "armv7k",
Arch::ArmV7m => "armv7m",
Arch::ArmV7em => "armv7em",
Arch::ArmUnknown => "arm_unknown",
Arch::Ppc => "ppc",
Arch::Ppc64 => "ppc64",
Arch::Mips => "mips",
Arch::Mips64 => "mips64",
Arch::Arm64_32 => "arm64_32",
Arch::Arm64_32V8 => "arm64_32_v8",
Arch::Arm64_32Unknown => "arm64_32_unknown",
}
}
pub fn well_known(self) -> bool {
!matches!(
self,
Arch::Unknown
| Arch::ArmUnknown
| Arch::Arm64Unknown
| Arch::X86Unknown
| Arch::Amd64Unknown
| Arch::Arm64_32Unknown
)
}
}
impl fmt::Display for Arch {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", self.name())
}
}
impl str::FromStr for Arch {
type Err = UnknownArchError;
fn from_str(string: &str) -> Result<Arch, UnknownArchError> {
Ok(match string.to_ascii_lowercase().as_str() {
"unknown" => Arch::Unknown,
"i386" => Arch::X86,
"x86" => Arch::X86,
"x86_unknown" => Arch::X86Unknown,
"x86_64" | "amd64" => Arch::Amd64,
"x86_64h" => Arch::Amd64h,
"x86_64_unknown" => Arch::Amd64Unknown,
"arm64" => Arch::Arm64,
"arm64v8" => Arch::Arm64V8,
"arm64e" => Arch::Arm64e,
"arm64_unknown" => Arch::Arm64Unknown,
"arm" => Arch::Arm,
"armv5" => Arch::ArmV5,
"armv6" => Arch::ArmV6,
"armv6m" => Arch::ArmV6m,
"armv7" => Arch::ArmV7,
"armv7f" => Arch::ArmV7f,
"armv7s" => Arch::ArmV7s,
"armv7k" => Arch::ArmV7k,
"armv7m" => Arch::ArmV7m,
"armv7em" => Arch::ArmV7em,
"arm_unknown" => Arch::ArmUnknown,
"ppc" => Arch::Ppc,
"ppc64" => Arch::Ppc64,
"mips" => Arch::Mips,
"mips64" => Arch::Mips64,
"arm64_32" => Arch::Arm64_32,
"arm64_32_v8" => Arch::Arm64_32V8,
"arm64_32_unknown" => Arch::Arm64_32Unknown,
"x86-64" => Arch::Amd64,
"arm-64" => Arch::Arm64,
"wasm32" => Arch::Wasm32,
_ => return Err(UnknownArchError),
})
}
}
#[derive(Debug)]
pub struct UnknownLanguageError;
impl fmt::Display for UnknownLanguageError {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "unknown language")
}
}
impl std::error::Error for UnknownLanguageError {}
#[repr(u32)]
#[non_exhaustive]
#[allow(missing_docs)]
#[derive(Clone, Copy, Debug, Eq, Hash, Ord, PartialEq, PartialOrd, Default)]
pub enum Language {
#[default]
Unknown = 0,
C = 1,
Cpp = 2,
D = 3,
Go = 4,
ObjC = 5,
ObjCpp = 6,
Rust = 7,
Swift = 8,
CSharp = 9,
VisualBasic = 10,
FSharp = 11,
}
impl Language {
pub fn from_u32(val: u32) -> Language {
match val {
0 => Self::Unknown,
1 => Self::C,
2 => Self::Cpp,
3 => Self::D,
4 => Self::Go,
5 => Self::ObjC,
6 => Self::ObjCpp,
7 => Self::Rust,
8 => Self::Swift,
9 => Self::CSharp,
10 => Self::VisualBasic,
11 => Self::FSharp,
_ => Self::Unknown,
}
}
pub fn name(self) -> &'static str {
match self {
Language::Unknown => "unknown",
Language::C => "c",
Language::Cpp => "cpp",
Language::D => "d",
Language::Go => "go",
Language::ObjC => "objc",
Language::ObjCpp => "objcpp",
Language::Rust => "rust",
Language::Swift => "swift",
Language::CSharp => "csharp",
Language::VisualBasic => "visualbasic",
Language::FSharp => "fsharp",
}
}
}
impl fmt::Display for Language {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
let formatted = match *self {
Language::Unknown => "unknown",
Language::C => "C",
Language::Cpp => "C++",
Language::D => "D",
Language::Go => "Go",
Language::ObjC => "Objective-C",
Language::ObjCpp => "Objective-C++",
Language::Rust => "Rust",
Language::Swift => "Swift",
Language::CSharp => "C#",
Language::VisualBasic => "Visual Basic",
Language::FSharp => "F#",
};
f.write_str(formatted)
}
}
impl str::FromStr for Language {
type Err = UnknownLanguageError;
fn from_str(string: &str) -> Result<Language, UnknownLanguageError> {
Ok(match string {
"unknown" => Language::Unknown,
"c" => Language::C,
"cpp" => Language::Cpp,
"d" => Language::D,
"go" => Language::Go,
"objc" => Language::ObjC,
"objcpp" => Language::ObjCpp,
"rust" => Language::Rust,
"swift" => Language::Swift,
"csharp" => Language::CSharp,
"visualbasic" => Language::VisualBasic,
"fsharp" => Language::FSharp,
_ => return Err(UnknownLanguageError),
})
}
}
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[derive(Clone, Copy, Debug, Eq, Hash, Ord, PartialEq, PartialOrd, Default)]
pub enum NameMangling {
Mangled,
Unmangled,
#[default]
Unknown,
}
#[derive(Clone, Debug, Eq, Hash, PartialEq)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
pub struct Name<'a> {
string: Cow<'a, str>,
lang: Language,
#[cfg_attr(feature = "serde", serde(default))]
mangling: NameMangling,
}
impl<'a> Name<'a> {
#[inline]
pub fn new<S>(string: S, mangling: NameMangling, lang: Language) -> Self
where
S: Into<Cow<'a, str>>,
{
Name {
string: string.into(),
lang,
mangling,
}
}
pub fn as_str(&self) -> &str {
&self.string
}
pub fn set_language(&mut self, language: Language) -> &mut Self {
self.lang = language;
self
}
pub fn language(&self) -> Language {
self.lang
}
pub fn set_mangling(&mut self, mangling: NameMangling) -> &mut Self {
self.mangling = mangling;
self
}
pub fn mangling(&self) -> NameMangling {
self.mangling
}
pub fn into_cow(self) -> Cow<'a, str> {
self.string
}
pub fn into_string(self) -> String {
self.string.into_owned()
}
}
impl AsRef<str> for Name<'_> {
fn as_ref(&self) -> &str {
self.as_str()
}
}
impl From<Name<'_>> for String {
fn from(name: Name) -> Self {
name.string.into()
}
}
impl<'a, S> From<S> for Name<'a>
where
S: Into<Cow<'a, str>>,
{
fn from(string: S) -> Self {
Self::new(string, NameMangling::Unknown, Language::Unknown)
}
}
impl fmt::Display for Name<'_> {
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
write!(f, "{}", self.as_str())?;
if f.alternate() && self.lang != Language::Unknown {
write!(f, " [{}]", self.lang)?;
}
Ok(())
}
}
macro_rules! impl_eq {
($lhs:ty, $rhs: ty) => {
#[allow(clippy::extra_unused_lifetimes)]
impl<'a, 'b> PartialEq<$rhs> for $lhs {
#[inline]
fn eq(&self, other: &$rhs) -> bool {
PartialEq::eq(&self.string, other)
}
}
#[allow(clippy::extra_unused_lifetimes)]
impl<'a, 'b> PartialEq<$lhs> for $rhs {
#[inline]
fn eq(&self, other: &$lhs) -> bool {
PartialEq::eq(self, &other.string)
}
}
};
}
impl_eq! { Name<'a>, str }
impl_eq! { Name<'a>, &'b str }
impl_eq! { Name<'a>, String }
impl_eq! { Name<'a>, std::borrow::Cow<'b, str> }
#[cfg(feature = "serde")]
mod derive_serde {
macro_rules! impl_str_serde {
($type:ty) => {
impl ::serde::ser::Serialize for $type {
fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
where
S: ::serde::ser::Serializer,
{
serializer.serialize_str(self.name())
}
}
impl<'de> ::serde::de::Deserialize<'de> for $type {
fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
where
D: ::serde::de::Deserializer<'de>,
{
<::std::borrow::Cow<str>>::deserialize(deserializer)?
.parse()
.map_err(::serde::de::Error::custom)
}
}
};
}
impl_str_serde!(super::Arch);
impl_str_serde!(super::Language);
}