Merge pull request #41 from staticfloat/sf/platform_extraction

Add `Platform(oh::ObjectHandle)`

Author: staticfloat

src/Abstract/ObjectHandle.jl

@@ -44,6 +44,7 @@ where `oh <: COFFHandle`).
   - unpack()
 
 ### Format-specific properties
+  - *Platform()*
   - *endianness()*
   - *is64bit()*
   - *isrelocatable()*
@@ -189,6 +190,12 @@ for f in [:skip, :seekstart, :eof, :read, :readuntil, :readbytes, :write]
     @eval $(f)(oh::H, args...) where {H<:ObjectHandle} = $(f)(iostream(oh), args...)
 end
 
+"""
+    Platform(oh::ObjectHandle)
+
+Returns a `Platform` object representing the binary platform this object is built for.
+"""
+@mustimplement Platform(oh::ObjectHandle)
 
 """
     endianness(oh::ObjectHandle)

src/COFF/COFF.jl

@@ -1,6 +1,7 @@
 module COFF
 
-using StructIO
+using StructIO, Base.BinaryPlatforms
+import Base.BinaryPlatforms: Platform
 
 # Bring in ObjectFile definitions
 using ObjectFile

src/COFF/COFFHandle.jl

@@ -73,6 +73,7 @@ iostream(oh::COFFHandle) = oh.io
 
 ## Format-specific properties:
 header(oh::COFFHandle) = oh.header
+Platform(oh::COFFHandle) = Platform(coff_machine_to_arch(oh.header.Machine), "windows")
 endianness(oh::COFFHandle) = coff_header_endianness(header(oh))
 is64bit(oh::COFFHandle) = coff_header_is64bit(header(oh))
 isrelocatable(oh::COFFHandle) = isrelocatable(header(oh))

src/COFF/constants.jl

@@ -24,6 +24,20 @@
     const IMAGE_FILE_MACHINE_WCEMIPSV2      = 0x169     #   MIPS little-endian WCE v2
 end
 
+function coff_machine_to_arch(machine::UInt16)
+    if machine ∈ (IMAGE_FILE_MACHINE_I386,)
+        return "i686"
+    elseif machine ∈ (IMAGE_FILE_MACHINE_AMD64, IMAGE_FILE_MACHINE_IA64)
+        return "x86_64"
+    elseif machine ∈ (IMAGE_FILE_MACHINE_ARM, IMAGE_FILE_MACHINE_ARMNT, IMAGE_FILE_MACHINE_THUMB)
+        return "armv7l"
+    elseif machine ∈ (IMAGE_FILE_MACHINE_ARM64,)
+        return "aarch64"
+    elseif machine ∈ (IMAGE_FILE_MACHINE_POWERPC,)
+        return "ppc64le"
+    end
+end
+
 # # # Characteristics
 
 @constants IMAGE_FILE_CHARACTERISTICS "IMAGE_FILE_" begin

src/ELF/ELF.jl

@@ -1,6 +1,7 @@
 module ELF
 
-using StructIO
+using StructIO, Base.BinaryPlatforms
+import Base.BinaryPlatforms: Platform
 
 # Bring in ObjectFile definitions
 using ObjectFile

src/ELF/ELFHandle.jl

@@ -64,9 +64,15 @@ end
 startaddr(oh::ELFHandle) = oh.start
 iostream(oh::ELFHandle) = oh.io
 
+# We don't try to inspect dynamic libraries to figure out if this is a glibc or musl dynamic object
+function strip_libc_tag(p::Platform)
+    delete!(tags(p), "libc")
+    return p
+end
 
 ## Format-specific properties:
 header(oh::ELFHandle) = oh.header
+Platform(oh::ELFHandle) = strip_libc_tag(Platform(elf_machine_to_arch(oh.header.e_machine), "linux"))
 endianness(oh::ELFHandle) = elf_internal_endianness(oh.ei)
 is64bit(oh::ELFHandle) = elf_internal_is64bit(oh.ei)
 isrelocatable(oh::ELFHandle) = header(oh).e_type == ET_REL

src/ELF/ELFHeader.jl

@@ -56,6 +56,20 @@ function machinetype(e_machine)
     return string("Unknown (0x",string(e_machine, base=16),")")
 end
 
+function elf_machine_to_arch(machine::UInt16)
+    if machine ∈ (EM_386,)
+        return "i686"
+    elseif machine ∈ (EM_IA_64, EM_X86_64)
+        return "x86_64"
+    elseif machine ∈ (EM_ARM,)
+        return "armv7l"
+    elseif machine ∈ (EM_AARCH64,)
+        return "aarch64"
+    elseif machine ∈ (EM_PPC64,)
+        return "ppc64le"
+    end
+end
+
 
 function show(io::IO, header::ELFHeader)
     println(io, "ELF Header")

src/MachO/MachO.jl

@@ -1,6 +1,7 @@
 module MachO
 
-using StructIO
+using StructIO, Base.BinaryPlatforms
+import Base.BinaryPlatforms: Platform
 
 # Bring in ObjectFile definitions
 using ObjectFile

src/MachO/MachOHandle.jl

@@ -46,6 +46,7 @@ iostream(oh::AbstractMachOHandle) = oh.io
 ## Format-specific properties:
 header(oh::AbstractMachOHandle) = oh.header
 endianness(oh::AbstractMachOHandle) = macho_endianness(header(oh).magic)
+Platform(oh::MachOHandle) = Platform(macho_cpu_to_arch(header(oh).cputype), "macos")
 is64bit(oh::MachOHandle) = macho_is64bit(header(oh).magic)
 isrelocatable(oh::MachOHandle) = header(oh).filetype == MH_OBJECT
 isexecutable(oh::MachOHandle) = header(oh).filetype == MH_EXECUTE

src/MachO/constants.jl

@@ -48,6 +48,21 @@ const CPU_ARCH_ABI64_32 = 0x02000000
     const CPU_TYPE_ARM64_32     = CPU_TYPE_ARM | CPU_ARCH_ABI64_32
 end
 
+function macho_cpu_to_arch(cputype::UInt32)
+    if cputype ∈ (CPU_TYPE_X86,)
+        return "i686"
+    elseif cputype ∈ (CPU_TYPE_X86_64,)
+        return "x86_64"
+    elseif cputype ∈ (CPU_TYPE_ARM,)
+        return "armv7l"
+    elseif cputype ∈ (CPU_TYPE_POWERPC64,)
+        return "ppc64le"
+    elseif cputype ∈ (CPU_TYPE_ARM64,)
+        return "aarch64"
+    end
+end
+
+
 # TODO subtype constants
 @constants NLISTTYPES "N_" begin
     const N_STAB         = 0xe0