1 files changed, 99 insertions, 0 deletions
diff --git a/third_party/rust/goblin/src/pe/characteristic.rs b/third_party/rust/goblin/src/pe/characteristic.rs
new file mode 100644
index 0000000000..f5c1fe4e5d
--- /dev/null
+++ b/third_party/rust/goblin/src/pe/characteristic.rs
@@ -0,0 +1,99 @@
+/*
+type characteristic =
+    | IMAGE_FILE_RELOCS_STRIPPED
+    | IMAGE_FILE_EXECUTABLE_IMAGE
+    | IMAGE_FILE_LINE_NUMS_STRIPPED
+    | IMAGE_FILE_LOCAL_SYMS_STRIPPED
+    | IMAGE_FILE_AGGRESSIVE_WS_TRIM
+    | IMAGE_FILE_LARGE_ADDRESS_AWARE
+    | RESERVED
+    | IMAGE_FILE_BYTES_REVERSED_LO
+    | IMAGE_FILE_32BIT_MACHINE
+    | IMAGE_FILE_DEBUG_STRIPPED
+    | IMAGE_FILE_REMOVABLE_RUN_FROM_SWAP
+    | IMAGE_FILE_NET_RUN_FROM_SWAP
+    | IMAGE_FILE_SYSTEM
+    | IMAGE_FILE_DLL
+    | IMAGE_FILE_UP_SYSTEM_ONLY
+    | IMAGE_FILE_BYTES_REVERSED_HI
+    | UNKNOWN of int
+
+let get_characteristic =
+  function
+  | 0x0001 -> IMAGE_FILE_RELOCS_STRIPPED
+  | 0x0002 -> IMAGE_FILE_EXECUTABLE_IMAGE
+  | 0x0004 -> IMAGE_FILE_LINE_NUMS_STRIPPED
+  | 0x0008 -> IMAGE_FILE_LOCAL_SYMS_STRIPPED
+  | 0x0010 -> IMAGE_FILE_AGGRESSIVE_WS_TRIM
+  | 0x0020 -> IMAGE_FILE_LARGE_ADDRESS_AWARE
+  | 0x0040 -> RESERVED
+  | 0x0080 -> IMAGE_FILE_BYTES_REVERSED_LO
+  | 0x0100 -> IMAGE_FILE_32BIT_MACHINE
+  | 0x0200 -> IMAGE_FILE_DEBUG_STRIPPED
+  | 0x0400 -> IMAGE_FILE_REMOVABLE_RUN_FROM_SWAP
+  | 0x0800 -> IMAGE_FILE_NET_RUN_FROM_SWAP
+  | 0x1000 -> IMAGE_FILE_SYSTEM
+  | 0x2000 -> IMAGE_FILE_DLL
+  | 0x4000 -> IMAGE_FILE_UP_SYSTEM_ONLY
+  | 0x8000 -> IMAGE_FILE_BYTES_REVERSED_HI
+  | x -> UNKNOWN x
+
+let characteristic_to_string =
+  function
+  | IMAGE_FILE_RELOCS_STRIPPED -> "IMAGE_FILE_RELOCS_STRIPPED"
+  | IMAGE_FILE_EXECUTABLE_IMAGE -> "IMAGE_FILE_EXECUTABLE_IMAGE"
+  | IMAGE_FILE_LINE_NUMS_STRIPPED -> "IMAGE_FILE_LINE_NUMS_STRIPPED"
+  | IMAGE_FILE_LOCAL_SYMS_STRIPPED -> "IMAGE_FILE_LOCAL_SYMS_STRIPPED"
+  | IMAGE_FILE_AGGRESSIVE_WS_TRIM -> "IMAGE_FILE_AGGRESSIVE_WS_TRIM"
+  | IMAGE_FILE_LARGE_ADDRESS_AWARE -> "IMAGE_FILE_LARGE_ADDRESS_AWARE"
+  | RESERVED -> "RESERVED"
+  | IMAGE_FILE_BYTES_REVERSED_LO -> "IMAGE_FILE_BYTES_REVERSED_LO"
+  | IMAGE_FILE_32BIT_MACHINE -> "IMAGE_FILE_32BIT_MACHINE"
+  | IMAGE_FILE_DEBUG_STRIPPED -> "IMAGE_FILE_DEBUG_STRIPPED"
+  | IMAGE_FILE_REMOVABLE_RUN_FROM_SWAP -> "IMAGE_FILE_REMOVABLE_RUN_FROM_SWAP"
+  | IMAGE_FILE_NET_RUN_FROM_SWAP -> "IMAGE_FILE_NET_RUN_FROM_SWAP"
+  | IMAGE_FILE_SYSTEM -> "IMAGE_FILE_SYSTEM"
+  | IMAGE_FILE_DLL -> "IMAGE_FILE_DLL"
+  | IMAGE_FILE_UP_SYSTEM_ONLY -> "IMAGE_FILE_UP_SYSTEM_ONLY"
+  | IMAGE_FILE_BYTES_REVERSED_HI -> "IMAGE_FILE_BYTES_REVERSED_HI"
+  | UNKNOWN x -> Printf.sprintf "UNKNOWN_CHARACTERISTIC 0x%x" x
+
+let is_dll characteristics =
+  let characteristic = characteristic_to_int IMAGE_FILE_DLL in
+  characteristics land characteristic = characteristic
+
+let has characteristic characteristics =
+  let characteristic = characteristic_to_int characteristic in
+  characteristics land characteristic = characteristic
+
+(* TODO: this is a mad hack *)
+let show_type characteristics =
+  if (has IMAGE_FILE_DLL characteristics) then "DLL"
+  else if (has IMAGE_FILE_EXECUTABLE_IMAGE characteristics) then "EXE"
+  else "MANY"                   (* print all *)
+ */
+
+pub const IMAGE_FILE_RELOCS_STRIPPED: u16 = 0x0001;
+pub const IMAGE_FILE_EXECUTABLE_IMAGE: u16 = 0x0002;
+pub const IMAGE_FILE_LINE_NUMS_STRIPPED: u16 = 0x0004;
+pub const IMAGE_FILE_LOCAL_SYMS_STRIPPED: u16 = 0x0008;
+pub const IMAGE_FILE_AGGRESSIVE_WS_TRIM: u16 = 0x0010;
+pub const IMAGE_FILE_LARGE_ADDRESS_AWARE: u16 = 0x0020;
+pub const RESERVED: u16 = 0x0040;
+pub const IMAGE_FILE_BYTES_REVERSED_LO: u16 = 0x0080;
+pub const IMAGE_FILE_32BIT_MACHINE: u16 = 0x0100;
+pub const IMAGE_FILE_DEBUG_STRIPPED: u16 = 0x0200;
+pub const IMAGE_FILE_REMOVABLE_RUN_FROM_SWAP: u16 = 0x0400;
+pub const IMAGE_FILE_NET_RUN_FROM_SWAP: u16 = 0x0800;
+pub const IMAGE_FILE_SYSTEM: u16 = 0x1000;
+pub const IMAGE_FILE_DLL: u16 = 0x2000;
+pub const IMAGE_FILE_UP_SYSTEM_ONLY: u16 = 0x4000;
+pub const IMAGE_FILE_BYTES_REVERSED_HI: u16 = 0x8000;
+
+pub fn is_dll(characteristics: u16) -> bool {
+    characteristics & IMAGE_FILE_DLL == IMAGE_FILE_DLL
+}
+
+pub fn is_exe(characteristics: u16) -> bool {
+    characteristics & IMAGE_FILE_EXECUTABLE_IMAGE == IMAGE_FILE_EXECUTABLE_IMAGE
+}