Add rudimentary support for parsing Metal libraries.

Author: maleadt

src/MachO/MachO.jl

@@ -32,6 +32,7 @@ include("MachOSymbol.jl")
 
 # These aren't complete implementations of the API
 include("MachOFat.jl")
+include("MetalLibrary.jl")
 
 
 end # module MachO

src/MachO/MachOHeader.jl

@@ -44,6 +44,8 @@ function macho_header_type(magic::UInt32)
         return MachOHeader64{MachOHandle}
     elseif magic in (FAT_MAGIC, FAT_CIGAM, FAT_MAGIC_64, FAT_CIGAM_64, FAT_MAGIC_METAL, FAT_CIGAM_METAL)
         return MachOFatHeader{MachOHandle}
+    elseif magic in (METALLIB_MAGIC,)
+        return MetallibHeader{MachOHandle}
     else
         throw(MagicMismatch("Invalid Magic ($(string(magic, base=16)))!"))
     end
@@ -58,7 +60,7 @@ Mach-O header.
 function macho_is64bit(magic::UInt32)
     if magic in (MH_MAGIC_64, MH_CIGAM_64, FAT_MAGIC_64, FAT_CIGAM_64)
         return true
-    elseif magic in (MH_MAGIC, MH_CIGAM, FAT_MAGIC, FAT_CIGAM, FAT_MAGIC_METAL, FAT_CIGAM_METAL)
+    elseif magic in (MH_MAGIC, MH_CIGAM, FAT_MAGIC, FAT_CIGAM, FAT_MAGIC_METAL, FAT_CIGAM_METAL, METALLIB_MAGIC)
         return false
     else
         throw(MagicMismatch("Invalid Magic ($(string(magic, base=16)))!"))
@@ -74,7 +76,7 @@ Mach-O header.
 function macho_endianness(magic::UInt32)
     if magic in (MH_CIGAM, MH_CIGAM_64, FAT_CIGAM, FAT_CIGAM_METAL)
         return :BigEndian
-    elseif magic in (MH_MAGIC, MH_MAGIC_64, FAT_MAGIC, FAT_MAGIC_METAL)
+    elseif magic in (MH_MAGIC, MH_MAGIC_64, FAT_MAGIC, FAT_MAGIC_METAL, METALLIB_MAGIC)
         return :LittleEndian
     else
         throw(MagicMismatch("Invalid Magic ($(string(magic, base=16)))!"))

src/MachO/MetalLibrary.jl

@@ -0,0 +1,58 @@
+@enum METALLIB_TARGET::UInt16 begin
+    METALLIB_TARGET_MACOS = 0x8001
+    METALLIB_TARGET_IOS = 0x0001
+end
+
+@enum METALLIB_FILETYPE::UInt8 begin
+    METALLIB_FILETYPE_EXECUTABLE = 0x00
+    METALLIB_FILETYPE_COREIMAGE = 0x01
+    METALLIB_FILETYPE_DYNAMIC = 0x02
+    METALLIB_FILETYPE_SYMBOLCOMPANION = 0x03
+end
+
+@enum METALLIB_TARGET_OS::UInt8 begin
+    METALLIB_TARGET_OS_UNKNOWN = 0x00
+    METALLIB_TARGET_OS_MACOS = 0x81
+    METALLIB_TARGET_OS_IOS = 0x82
+    METALLIB_TARGET_OS_TVOS = 0x83
+    METALLIB_TARGET_OS_WATCHOS = 0x84
+    METALLIB_TARGET_OS_BRIDGEOS = 0x85
+    METALLIB_TARGET_OS_MACCATALYST = 0x86
+    METALLIB_TARGET_OS_IOS_SIMULATOR = 0x87
+    METALLIB_TARGET_OS_TVOS_SIMULATOR = 0x88
+    METALLIB_TARGET_OS_WATCHOS_SIMULATOR = 0x89
+end
+
+@io struct MetallibHeader{H <: ObjectHandle} <: MachOHeader{H}
+    magic::UInt32
+    target::METALLIB_TARGET
+    ver_major::UInt16
+    ver_minor::UInt16
+    filetype::METALLIB_FILETYPE
+    target_os::METALLIB_TARGET_OS
+    os_major::UInt16
+    os_minor::UInt16
+    size::UInt64
+    funclist_offset::UInt64
+    funclist_size::UInt64
+    public_md_offset::UInt64
+    public_md_size::UInt64
+    private_md_offset::UInt64
+    private_md_size::UInt64
+    bitcode_offset::UInt64
+    bitcode_size::UInt64
+end
+
+function show(io::IO, header::MetallibHeader)
+    println(io, "Metallib Header")
+    println(io, "  Target ", header.target)
+    println(io, "  Version ", header.ver_major, ".", header.ver_minor)
+    println(io, "  File Type ", header.filetype)
+    println(io, "  Target OS ", header.target_os)
+    println(io, "  OS Version ", header.os_major, ".", header.os_minor)
+    println(io, "  Size ", Base.format_bytes(header.size))
+    println(io, "  Function List ", Base.format_bytes(header.funclist_size), " at 0x", string(header.funclist_offset; base=16))
+    println(io, "  Public Metadata ", Base.format_bytes(header.public_md_size), " at 0x", string(header.public_md_offset; base=16))
+    println(io, "  Private Metadata ", Base.format_bytes(header.private_md_size), " at 0x", string(header.private_md_offset; base=16))
+    println(io, "  Bitcode ", Base.format_bytes(header.bitcode_size), " at 0x", string(header.bitcode_offset; base=16))
+end

src/MachO/constants.jl

@@ -14,6 +14,8 @@
 
     const FAT_MAGIC_METAL = 0xCBFEBABE
     const FAT_CIGAM_METAL = bswap(FAT_MAGIC_METAL)
+
+    const METALLIB_MAGIC = 0x424c544d   # "MTLB" in ascii
 end
 
 const CPU_ARCH_MASK     = 0xff000000