===================================================================
@@ -0,0 +1,205 @@
+/* Very simple endian-ness layer for LTO object file handling
+ Copyright 2010 Free Software Foundation, Inc.
+
+This file is part of GCC.
+
+GCC is free software; you can redistribute it and/or modify it under
+the terms of the GNU General Public License as published by the Free
+Software Foundation; either version 3, or (at your option) any later
+version.
+
+GCC is distributed in the hope that it will be useful, but WITHOUT ANY
+WARRANTY; without even the implied warranty of MERCHANTABILITY or
+FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+for more details.
+
+You should have received a copy of the GNU General Public License
+along with GCC; see the file COPYING3. If not see
+<http://www.gnu.org/licenses/>. */
+
+/* This header file provides a simple way to handle object files in
+ another endian-ness than the host machine. This is necesarry to
+ enable cross-compilation with LTO enabled. Targets that use the
+ ELF binary object format do not need this (libelf already handles
+ endian-ness) but for COFF and Mach-O the functions in this header
+ are used in the minimal binary object reader/writer.
+
+ For all functions in this header, the user is responsible for
+ making sure that the memory accesses are valid. */
+
+#ifndef GCC_LTO_ENDIAN_H
+#define GCC_LTO_ENDIAN_H
+
+#include <stdint.h>
+#include <inttypes.h>
+
+static inline uint16_t
+get_uint16_le (const unsigned char *ptr)
+{
+ return ptr[0] | (ptr[1] << 8);
+}
+
+static inline uint32_t
+get_uint32_le (const unsigned char *ptr)
+{
+ return ptr[0] | (ptr[1] << 8) | (ptr[2] << 16) | (ptr[3] << 24);
+}
+
+static inline uint64_t
+get_uint64_le (const unsigned char *ptr_)
+{
+#define ptr (uint64_t) ptr_
+ return ptr[0] | (ptr[1] << 8) | (ptr[2] << 16) | (ptr[3] << 24)
+ | (ptr[4] << 32) | (ptr[5] << 40) | (ptr[6] << 48) | (ptr[7] << 56);
+#undef ptr
+}
+
+static inline uint16_t
+get_uint16_be (const unsigned char *ptr)
+{
+ return ptr[1] | (ptr[2] << 8);
+}
+
+static inline uint32_t
+get_uint32_be (const unsigned char *ptr)
+{
+ return ptr[3] | (ptr[2] << 8) | (ptr[1] << 16) | (ptr[0] << 24);
+}
+
+static inline uint64_t
+get_uint64_be (const unsigned char *ptr_)
+{
+#define ptr (uint64_t) ptr_
+ return ptr[7] | (ptr[6] << 8) | (ptr[5] << 16) | (ptr[4] << 24)
+ | (ptr[3] << 32) | (ptr[2] << 40) | (ptr[1] << 48) | (ptr[0] << 56);
+#undef ptr
+}
+
+static inline void
+put_uint16_le (unsigned char *ptr, uint16_t data)
+{
+ ptr[0] = data & 0xff;
+ ptr[1] = (data >> 8) & 0xff;
+}
+
+static inline void
+put_uint32_le (unsigned char *ptr, uint32_t data)
+{
+ ptr[0] = data & 0xff;
+ ptr[1] = (data >> 8) & 0xff;
+ ptr[2] = (data >> 16) & 0xff;
+ ptr[3] = (data >> 24) & 0xff;
+}
+
+static inline void
+put_uint64_le (unsigned char *ptr, uint64_t data)
+{
+ ptr[0] = data & 0xff;
+ ptr[1] = (data >> 8) & 0xff;
+ ptr[2] = (data >> 16) & 0xff;
+ ptr[3] = (data >> 24) & 0xff;
+ ptr[4] = (data >> 32) & 0xff;
+ ptr[5] = (data >> 40) & 0xff;
+ ptr[6] = (data >> 48) & 0xff;
+ ptr[7] = (data >> 56) & 0xff;
+}
+
+static inline void
+put_uint16_be (unsigned char *ptr, uint16_t data)
+{
+ ptr[1] = data & 0xff;
+ ptr[0] = (data >> 8) & 0xff;
+}
+
+static inline void
+put_uint32_be (unsigned char *ptr, uint32_t data)
+{
+ ptr[3] = data & 0xff;
+ ptr[2] = (data >> 8) & 0xff;
+ ptr[1] = (data >> 16) & 0xff;
+ ptr[0] = (data >> 24) & 0xff;
+}
+
+static inline void
+put_uint64_be (unsigned char *ptr, uint64_t data)
+{
+ ptr[7] = data & 0xff;
+ ptr[6] = (data >> 8) & 0xff;
+ ptr[5] = (data >> 16) & 0xff;
+ ptr[4] = (data >> 24) & 0xff;
+ ptr[3] = (data >> 32) & 0xff;
+ ptr[2] = (data >> 40) & 0xff;
+ ptr[1] = (data >> 48) & 0xff;
+ ptr[0] = (data >> 56) & 0xff;
+}
+
+static inline void
+get_string (unsigned char *ptr, char *dest, size_t len)
+{
+ memcpy (dest, ptr, len);
+}
+
+static inline void
+put_string (unsigned char *ptr, char *src, size_t len)
+{
+ memcpy (ptr, src, len);
+}
+
+/* Use the target macro BYTES_BIG_ENDIAN to choose. */
+
+static inline uint16_t
+get_uint16 (const unsigned char *ptr)
+{
+ if (BYTES_BIG_ENDIAN)
+ return get_uint16_be (ptr);
+ else
+ return get_uint16_le (ptr);
+}
+
+static inline uint32_t
+get_uint32 (const unsigned char *ptr)
+{
+ if (BYTES_BIG_ENDIAN)
+ return get_uint32_be (ptr);
+ else
+ return get_uint32_le (ptr);
+}
+
+static inline uint64_t
+get_uint64 (const unsigned char *ptr)
+{
+ if (BYTES_BIG_ENDIAN)
+ return get_uint64_be (ptr);
+ else
+ return get_uint64_le (ptr);
+}
+
+static inline void
+put_uint16 (unsigned char *ptr, uint16_t data)
+{
+ if (BYTES_BIG_ENDIAN)
+ put_uint16_be (ptr, data);
+ else
+ put_uint16_le (ptr, data);
+}
+
+static inline void
+put_uint32 (unsigned char *ptr, uint32_t data)
+{
+ if (BYTES_BIG_ENDIAN)
+ put_uint32_be (ptr, data);
+ else
+ put_uint32_le (ptr, data);
+}
+
+static inline void
+put_uint64 (unsigned char *ptr, uint64_t data)
+{
+ if (BYTES_BIG_ENDIAN)
+ put_uint64_be (ptr, data);
+ else
+ put_uint64_le (ptr, data);
+}
+
+#endif /* GCC_LTO_ENDIAN_H */
+
===================================================================
@@ -0,0 +1,969 @@
+/* LTO routines for Mach-O object files.
+ Copyright 2010 Free Software Foundation, Inc.
+ Contributed by Steven Bosscher.
+
+This file is part of GCC.
+
+GCC is free software; you can redistribute it and/or modify it under
+the terms of the GNU General Public License as published by the Free
+Software Foundation; either version 3, or (at your option) any later
+version.
+
+GCC is distributed in the hope that it will be useful, but WITHOUT ANY
+WARRANTY; without even the implied warranty of MERCHANTABILITY or
+FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+for more details.
+
+You should have received a copy of the GNU General Public License
+along with GCC; see the file COPYING3. If not see
+<http://www.gnu.org/licenses/>. */
+
+#include "config.h"
+#include "system.h"
+#include "coretypes.h"
+#include "toplev.h"
+#include "lto.h"
+#include "tm.h"
+#include "libiberty.h"
+#include "lto-streamer.h"
+#include "lto/lto-endian.h"
+#include "lto/lto-macho.h"
+
+/* Rather than implementing a libmacho to match libelf, or attempting to
+ integrate libbfd into GCC, this file is a self-contained (and very
+ minimal) Mach-O format object file reader/writer. The generated files
+ will contain a Mach-O header, a number of Mach-O load commands an
+ section headers, the section data itself, and a trailing string table
+ for section names. */
+
+/* This needs to be kept in sync with darwin.c. Better yet, lto-macho.c
+ and lto-macho.h should be moved to config/, and likewise for lto-coff.*
+ and lto-elf.*. */
+
+/* Segment name for LTO sections. */
+#define LTO_SEGMENT_NAME "__GNU_LTO"
+
+/* Section name for LTO section names section. */
+#define LTO_NAMES_SECTION "__section_names"
+
+/* Handle opening elf files on hosts, such as Windows, that may use
+ text file handling that will break binary access. */
+#ifndef O_BINARY
+# define O_BINARY 0
+#endif
+
+/* Cached object file header. We use a header_64 for this, since all
+ the fields we need are in there, in the same position as header_32. */
+mach_o_header_64 cached_mach_o_header;
+uint32_t cached_mach_o_magic;
+
+/* The current output file. */
+static lto_file *current_out_file;
+
+
+/* Is this a 32-bits or 64-bits Mach-O object file? */
+static int
+mach_o_word_size (void)
+{
+ gcc_assert (cached_mach_o_magic != 0);
+ return (cached_mach_o_magic == MACH_O_MH_MAGIC_64
+ || cached_mach_o_magic == MACH_O_MH_CIGAM_64) ? 64 : 32;
+}
+
+/* Sets the current output file to FILE. Returns the old output file or
+ NULL. */
+
+lto_file *
+lto_set_current_out_file (lto_file *file)
+{
+ lto_file *old_file = current_out_file;
+ current_out_file = file;
+ return old_file;
+}
+
+
+/* Returns the current output file. */
+
+lto_file *
+lto_get_current_out_file (void)
+{
+ return current_out_file;
+}
+
+/* Mach-O section structure constructor. */
+
+static lto_mach_o_section
+mach_o_new_section (lto_mach_o_file *mach_o_file, const char *name)
+{
+ lto_mach_o_section ptr;
+
+ /* FIXME We could allocate these things on an obstack. */
+ ptr = XCNEW (struct lto_mach_o_section_d);
+ if (name)
+ {
+ if (strncmp (name, LTO_SECTION_NAME_PREFIX,
+ strlen(LTO_SECTION_NAME_PREFIX)) != 0)
+ sorry ("not implemented: Mach-O writer for non-LTO sections");
+ ptr->name = xstrdup (name);
+ }
+
+ VEC_safe_push (lto_mach_o_section, heap, mach_o_file->section_vec, ptr);
+
+ return ptr;
+}
+
+/* Mach-O section data block structure constructor. */
+
+static lto_mach_o_data
+mach_o_new_data (lto_mach_o_section sec)
+{
+ lto_mach_o_data ptr, *chain_ptr_ptr;
+
+ /* FIXME We could allocate these things on an obstack. */
+ ptr = XCNEW (struct lto_mach_o_data_d);
+
+ chain_ptr_ptr = &sec->data_chain;
+ while (*chain_ptr_ptr)
+ chain_ptr_ptr = &(*chain_ptr_ptr)->next;
+ *chain_ptr_ptr = ptr;
+
+ return ptr;
+}
+
+/* Initialize FILE, an LTO file object for FILENAME. Offset is the
+ offset into FILE where the object is located (e.g. in an archive). */
+
+static void
+lto_file_init (lto_file *file, const char *filename, off_t offset)
+{
+ file->filename = filename;
+ file->offset = offset;
+}
+
+/* Returns a hash code for P. */
+
+static hashval_t
+hash_name (const void *p)
+{
+ const struct lto_section_slot *s = (const struct lto_section_slot *) p;
+ return (hashval_t) htab_hash_string (s->name);
+}
+
+/* Returns nonzero if P1 and P2 are equal. */
+
+static int
+eq_name (const void *p1, const void *p2)
+{
+ const struct lto_section_slot *s1 =
+ (const struct lto_section_slot *) p1;
+ const struct lto_section_slot *s2 =
+ (const struct lto_section_slot *) p2;
+
+ return strcmp (s1->name, s2->name) == 0;
+}
+
+/* Build a hash table whose key is the section names and whose data is
+ the start and size of each section in the .o file. */
+
+htab_t
+lto_obj_build_section_table (lto_file *lto_file)
+{
+ lto_mach_o_file *mach_o_file = (lto_mach_o_file *)lto_file;
+ lto_mach_o_section sec;
+ htab_t section_hash_table;
+ off_t strtab_offs;
+ ssize_t strtab_size;
+ char *strtab = NULL;
+ int i;
+
+ section_hash_table = htab_create (37, hash_name, eq_name, free);
+
+ /* Seek the string table. */
+ /* FIXME The segment name should be in darwin.h, but can we include it
+ here in this file? */
+ for (i = 0;
+ VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ {
+ if (strncmp (sec->u.section.segname, "__GNU_LTO", 16) != 0)
+ continue;
+ if (strncmp (sec->u.section.sectname, "__section_names", 16) == 0)
+ break;
+ }
+ if (! sec)
+ {
+ error ("invalid Mach-O LTO object file: no __section_names section found");
+ goto done;
+ }
+ mach_o_file->section_names_section = sec;
+
+ if (mach_o_word_size () == 64)
+ {
+ strtab_offs = (off_t) get_uint32 (&sec->u.section_64.offset[0]);
+ strtab_size = (size_t) get_uint64 (&sec->u.section_64.size[0]);
+ }
+ else
+ {
+ strtab_offs = (off_t) get_uint32 (&sec->u.section_32.offset[0]);
+ strtab_size = (size_t) get_uint32 (&sec->u.section_32.size[0]);
+ }
+
+ /* Seek to start of string table. */
+ if (strtab_offs != lseek (mach_o_file->fd,
+ mach_o_file->base.offset + strtab_offs,
+ SEEK_SET))
+ {
+ error ("altered or invalid Mach-O object file");
+ goto done;
+ }
+
+ strtab = XNEWVEC (char, strtab_size);
+ if (read (mach_o_file->fd, strtab, strtab_size) != strtab_size)
+ {
+ error ("invalid Mach-O LTO object file __section_names section");
+ goto done;
+ }
+
+ /* Scan sections looking at names. */
+ for (i = 0;
+ VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ {
+ struct lto_section_slot s_slot;
+ void **slot;
+ char *new_name;
+ unsigned long stringoffset;
+ char name[17];
+
+ /* Ignore non-LTO sections. Also ignore the __section_names section
+ which does not need renaming. */
+ if (strncmp (sec->u.section.segname, "__GNU_LTO", 16) != 0)
+ continue;
+ if (sec == mach_o_file->section_names_section)
+ continue;
+
+ /* Try to extract the offset of the real name for this section from
+ __section_names. */
+ memcpy (&name[0], sec->u.section.sectname, 16);
+ name[16] = '\0';
+ if (name[0] != '_' || name[1] != '_'
+ || sscanf (&name[2], "%08lX", &stringoffset) != 1
+ || strtab_size < (ssize_t) stringoffset)
+ {
+ error ("invalid Mach-O LTO section name string: %s", name);
+ continue;
+ }
+
+ new_name = XNEWVEC (char, strlen (strtab + stringoffset) + 1);
+ strcpy (new_name, strtab + stringoffset);
+ s_slot.name = new_name;
+ slot = htab_find_slot (section_hash_table, &s_slot, INSERT);
+ if (*slot == NULL)
+ {
+ struct lto_section_slot *new_slot = XNEW (struct lto_section_slot);
+
+ new_slot->name = new_name;
+ if (mach_o_word_size() == 64)
+ {
+ new_slot->start =
+ (intptr_t) get_uint32 (&sec->u.section_64.offset[0]);
+ new_slot->len =
+ (size_t) get_uint64 (&sec->u.section_64.size[0]);
+ }
+ else
+ {
+ new_slot->start =
+ (intptr_t) get_uint32 (&sec->u.section_32.offset[0]);
+ new_slot->len =
+ (size_t) get_uint32 (&sec->u.section_32.size[0]);
+ }
+
+ *slot = new_slot;
+ }
+ else
+ {
+ error ("two or more sections for %s:", new_name);
+ goto done;
+ }
+ }
+
+ done:
+ if (strtab)
+ free (strtab);
+ return section_hash_table;
+}
+
+
+/* Begin a new Mach-O section named NAME in the current output file. */
+
+void
+lto_obj_begin_section (const char *name)
+{
+ lto_mach_o_file *file;
+
+ if (strncmp (name, LTO_SECTION_NAME_PREFIX,
+ strlen(LTO_SECTION_NAME_PREFIX)) != 0)
+ sorry ("not implemented: Mach-O writer for non-LTO sections");
+
+ /* Grab the current output file and do some basic assertion checking. */
+ file = (lto_mach_o_file *) lto_get_current_out_file (),
+ gcc_assert (file && file->writable && !file->scn);
+
+ /* Create a new section. */
+ file->scn = mach_o_new_section (file, name);
+ if (!file->scn)
+ fatal_error ("could not create a new Mach-O section: %m");
+}
+
+
+/* Append DATA of length LEN to the current output section. BASE is a pointer
+ to the output page containing DATA. It is freed once the output file has
+ been written. */
+
+void
+lto_obj_append_data (const void *data, size_t len, void *block)
+{
+ lto_mach_o_file *file;
+ lto_mach_o_data mach_o_data;
+ struct lto_char_ptr_base *base = (struct lto_char_ptr_base *) block;
+
+ /* Grab the current output file and do some basic assertion checking. */
+ file = (lto_mach_o_file *) lto_get_current_out_file ();
+ gcc_assert (file);
+ gcc_assert (file->scn);
+
+ mach_o_data = mach_o_new_data (file->scn);
+ if (!mach_o_data)
+ fatal_error ("could not append data to Mach-O section: %m");
+
+ mach_o_data->d_buf = CONST_CAST (void *, data);
+ mach_o_data->d_size = len;
+
+ /* Chain all data blocks (from all sections) on one singly-linked
+ list for freeing en masse after the file is closed. */
+ base->ptr = (char *)file->data;
+ file->data = base;
+}
+
+
+/* End the current output section. This just does some assertion checking
+ and sets the current output file's scn member to NULL. */
+
+void
+lto_obj_end_section (void)
+{
+ lto_mach_o_file *file;
+
+ /* Grab the current output file and validate some basic assertions. */
+ file = (lto_mach_o_file *) lto_get_current_out_file ();
+ gcc_assert (file);
+ gcc_assert (file->scn);
+
+ file->scn = NULL;
+}
+
+
+/* Read a Mach-O header from MACH_O_FILE and validate it.
+ The file descriptor in MACH_O_FILE points at the start of the file.
+ If cached_mach_o_header is uninitialized, caches the results.
+ On succes, returns true and moves file pointer to the start of the
+ load commands. On failure, returns false. */
+
+static bool
+validate_mach_o_header (lto_mach_o_file *mach_o_file)
+{
+ ssize_t i, n;
+ unsigned char magic[4];
+ uint32_t cputype;
+ off_t startpos;
+
+ /* Known header magics for validation, as an array. */
+ static const unsigned int mach_o_known_formats[] = {
+ MACH_O_MH_MAGIC,
+ MACH_O_MH_CIGAM,
+ MACH_O_MH_MAGIC_64,
+ MACH_O_MH_CIGAM_64,
+ };
+#define MACH_O_NUM_KNOWN_FORMATS \
+ ((ssize_t) ARRAY_SIZE (mach_o_known_formats))
+
+ startpos = lseek (mach_o_file->fd, 0, SEEK_CUR);
+ if (read (mach_o_file->fd, &magic, sizeof (magic)) != 4
+ || lseek (mach_o_file->fd, -4, SEEK_CUR) != startpos)
+ {
+ error ("cannot read file %s", mach_o_file->base.filename);
+ return false;
+ }
+
+ for (i = 0; i < MACH_O_NUM_KNOWN_FORMATS; ++i)
+ if (get_uint32 (&magic[0]) == mach_o_known_formats[i])
+ break;
+ if (i == MACH_O_NUM_KNOWN_FORMATS)
+ goto not_for_target;
+
+ /* Check the endian-ness. */
+ if (BYTES_BIG_ENDIAN && magic[0] != 0xfe)
+ goto not_for_target;
+
+ /* Set or check cached magic number. */
+ if (cached_mach_o_magic == 0)
+ cached_mach_o_magic = get_uint32 (&magic[0]);
+ else if (cached_mach_o_magic != get_uint32 (&magic[0]))
+ goto not_for_target;
+
+ n = mach_o_word_size () == 64
+ ? sizeof (mach_o_header_64) : sizeof (mach_o_header_32);
+ if (read (mach_o_file->fd, &mach_o_file->u.header, n) != n)
+ goto not_for_target;
+
+ /* Is this a supported CPU? */
+ /* ??? Would be nice to validate the exact target architecture. */
+ cputype = get_uint32 (&mach_o_file->u.header.cputype[0]);
+ if (cputype == MACH_O_CPU_TYPE_I386
+ || cputype == MACH_O_CPU_TYPE_POWERPC)
+ {
+ if (mach_o_word_size () != 32)
+ goto not_for_target;
+ }
+ else if (cputype == MACH_O_CPU_TYPE_X86_64
+ || cputype == MACH_O_CPU_TYPE_POWERPC_64)
+ {
+ if (mach_o_word_size () != 64)
+ goto not_for_target;
+ }
+
+ /* Is this an MH_OBJECT file? */
+ if (get_uint32 (&mach_o_file->u.header.filetype[0]) != MACH_O_MH_OBJECT)
+ error ("Mach-O file %s is not an MH_OBJECT file",
+ mach_o_file->base.filename);
+
+ /* Save the header for future use. */
+ memcpy (&cached_mach_o_header, &mach_o_file->u.header,
+ sizeof (cached_mach_o_header));
+
+ return true;
+
+ not_for_target:
+ error ("file %s is not a Mach-O object file for target",
+ mach_o_file->base.filename);
+ return false;
+}
+
+
+/* Read a Mach-O LC_SEGMENT command (32 bits) from MACH_O_FILE and
+ validate it.
+ The file descriptor in MACH_O_FILE points at the start of the load
+ command. On sucess, returns true and advances the file pointer
+ past the end of the load command. On failure, returns false. */
+
+static bool
+validate_mach_o_segment_command_32 (lto_mach_o_file *mach_o_file)
+{
+ mach_o_segment_command_32 seg_cmd_32;
+ unsigned int i;
+ ssize_t n;
+ off_t startpos;
+
+ /* Fields we're interested in. */
+ uint32_t cmd;
+ uint32_t cmdsize;
+ uint32_t nsects;
+
+ startpos = lseek (mach_o_file->fd, 0, SEEK_CUR);
+
+ n = sizeof (mach_o_segment_command_32);
+ if (read (mach_o_file->fd, (void *) &seg_cmd_32, n) != n)
+ goto fail;
+
+ cmd = get_uint32 (&seg_cmd_32.cmd[0]);
+ cmdsize = get_uint32 (&seg_cmd_32.cmdsize[0]);
+ nsects = get_uint32 (&seg_cmd_32.nsects[0]);
+ gcc_assert (cmd == MACH_O_LC_SEGMENT);
+
+ /* Validate section table entries. */
+ for (i = 0; i < nsects; i++)
+ {
+ mach_o_section_32 sec_32;
+ lto_mach_o_section ltosec;
+
+ n = sizeof (mach_o_section_32);
+ if (read (mach_o_file->fd, &sec_32, n) != n)
+ goto fail;
+
+ /* ??? Perform some checks. */
+
+ /* Looks ok, so record its details. We don't read the
+ string table or set up names yet; we'll do that when
+ we build the hash table. */
+ ltosec = mach_o_new_section (mach_o_file, NULL);
+ memcpy (<osec->u.section_32, &sec_32, sizeof (sec_32));
+ }
+
+ if (lseek (mach_o_file->fd, 0, SEEK_CUR) != startpos + cmdsize)
+ goto fail;
+
+ return true;
+
+ fail:
+ error ("could not read LC_SEGMENT command in Mach-O file %s",
+ mach_o_file->base.filename);
+ return false;
+}
+
+
+/* Read a Mach-O LC_SEGMENT_64 command from MACH_O_FILE and validate it.
+ The file descriptor in MACH_O_FILE points at the start of the load
+ command. On sucess, returns true and advances the file pointer
+ past the end of the load command. On failure, returns false. */
+
+static bool
+validate_mach_o_segment_command_64 (lto_mach_o_file *mach_o_file)
+{
+ mach_o_segment_command_64 seg_cmd_64;
+ unsigned int i;
+ ssize_t n;
+ off_t startpos;
+
+ /* Fields we're interested in. */
+ uint32_t cmd;
+ uint32_t cmdsize;
+ uint32_t nsects;
+
+ startpos = lseek (mach_o_file->fd, 0, SEEK_CUR);
+
+ n = sizeof (mach_o_segment_command_64);
+ if (read (mach_o_file->fd, (void *) &seg_cmd_64, n) != n)
+ goto fail;
+
+ cmd = get_uint32 (&seg_cmd_64.cmd[0]);
+ cmdsize = get_uint32 (&seg_cmd_64.cmdsize[0]);
+ nsects = get_uint32 (&seg_cmd_64.nsects[0]);
+ gcc_assert (cmd == MACH_O_LC_SEGMENT_64);
+
+ /* Validate section table entries. */
+ for (i = 0; i < nsects; i++)
+ {
+ mach_o_section_64 sec_64;
+ lto_mach_o_section ltosec;
+
+ n = sizeof (mach_o_section_64);
+ if (read (mach_o_file->fd, &sec_64, n) != n)
+ goto fail;
+
+ /* ??? Perform some checks. */
+
+ /* Looks ok, so record its details. We don't read the
+ string table or set up names yet; we'll do that when
+ we build the hash table. */
+ ltosec = mach_o_new_section (mach_o_file, NULL);
+ memcpy (<osec->u.section_64, &sec_64, sizeof (sec_64));
+ }
+
+ if (lseek (mach_o_file->fd, 0, SEEK_CUR) != startpos + cmdsize)
+ goto fail;
+
+ return true;
+
+ fail:
+ error ("could not read LC_SEGMENT_64 command in Mach-O file %s",
+ mach_o_file->base.filename);
+ return false;
+}
+
+/* Read a Mach-O load commands from MACH_O_FILE and validate it.
+ The file descriptor in MACH_O_FILE points at the start of the load
+ command. On sucess, returns true and advances the file pointer
+ past the end of the load command. On failure, returns false. */
+
+static bool
+validate_mach_o_load_command (lto_mach_o_file *mach_o_file)
+{
+ mach_o_load_command load_command;
+ uint32_t cmd;
+ uint32_t cmdsize;
+ ssize_t n;
+
+ n = sizeof (load_command);
+ if (read (mach_o_file->fd, &load_command, n) != n)
+ {
+ error ("could not read load commands in Mach-O file %s",
+ mach_o_file->base.filename);
+ return false;
+ }
+ lseek (mach_o_file->fd, -1 * (off_t) sizeof (load_command), SEEK_CUR);
+
+ cmd = get_uint32 (&load_command.cmd[0]);
+ cmdsize = get_uint32 (&load_command.cmdsize[0]);
+ switch (cmd)
+ {
+ case MACH_O_LC_SEGMENT:
+ return validate_mach_o_segment_command_32 (mach_o_file);
+ case MACH_O_LC_SEGMENT_64:
+ return validate_mach_o_segment_command_64 (mach_o_file);
+
+ default:
+ /* Just skip over it. */
+ lseek (mach_o_file->fd, cmdsize, SEEK_CUR);
+ return true;
+ }
+}
+
+/* Validate's MACH_O_FILE's executable header and, if cached_mach_o_header is
+ uninitialized, caches the results. Also records the section header string
+ table's section index. Returns true on success, false on failure. */
+
+static bool
+validate_file (lto_mach_o_file *mach_o_file)
+{
+ uint32_t i, ncmds;
+
+ /* Read and sanity check the raw header. */
+ if (! validate_mach_o_header (mach_o_file))
+ return false;
+
+ ncmds = get_uint32 (&mach_o_file->u.header.ncmds[0]);
+ for (i = 0; i < ncmds; ++i)
+ if (! validate_mach_o_load_command (mach_o_file))
+ return false;
+
+ return true;
+}
+
+/* Initialize MACH_O_FILE's executable header using cached data from previously
+ read files. */
+
+static void
+init_mach_o_header (lto_mach_o_file *mach_o_file)
+{
+ gcc_assert (cached_mach_o_magic != 0);
+ memcpy (&mach_o_file->u.header,
+ &cached_mach_o_header,
+ sizeof (mach_o_file->u.header));
+ put_uint32 (&mach_o_file->u.header.ncmds[0], 0);
+ put_uint32 (&mach_o_file->u.header.sizeofcmds[0], 0);
+}
+
+/* Open Mach-O file FILENAME. If WRITABLE is true, the file is opened for write
+ and, if necessary, created. Otherwise, the file is opened for reading.
+ Returns the opened file. */
+
+lto_file *
+lto_obj_file_open (const char *filename, bool writable)
+{
+ lto_mach_o_file *mach_o_file;
+ lto_file *result = NULL;
+ off_t offset;
+ const char *offset_p;
+ char *fname;
+ struct stat statbuf;
+
+ offset_p = strchr (filename, '@');
+ if (!offset_p)
+ {
+ fname = xstrdup (filename);
+ offset = 0;
+ }
+ else
+ {
+ /* The file started with '@' is a file containing command line
+ options. Stop if it doesn't exist. */
+ if (offset_p == filename)
+ fatal_error ("command line option file '%s' does not exist",
+ filename);
+
+ fname = (char *) xmalloc (offset_p - filename + 1);
+ memcpy (fname, filename, offset_p - filename);
+ fname[offset_p - filename] = '\0';
+ offset_p += 3; /* skip the @0x */
+ offset = lto_parse_hex (offset_p);
+ }
+
+ /* Set up. */
+ mach_o_file = XCNEW (lto_mach_o_file);
+ result = (lto_file *) mach_o_file;
+ lto_file_init (result, fname, offset);
+ mach_o_file->fd = -1;
+ mach_o_file->writable = writable;
+
+ /* Open the file. */
+ mach_o_file->fd = open (fname,
+ O_BINARY | (writable ? O_WRONLY | O_CREAT | O_TRUNC : O_RDONLY), 0666);
+
+ if (mach_o_file->fd == -1)
+ {
+ error ("could not open file %s", fname);
+ goto fail;
+ }
+
+ if (stat (fname, &statbuf) < 0)
+ {
+ error ("could not stat file %s", fname);
+ goto fail;
+ }
+
+ mach_o_file->file_size = statbuf.st_size;
+
+ /* If the object is in an archive, get it out. */
+ if (offset != 0)
+ {
+ char ar_tail[12];
+ int size;
+
+ /* Surely not? */
+ gcc_assert (!writable);
+
+ /* Seek to offset, or error. */
+ if (lseek (mach_o_file->fd, offset, SEEK_SET) != (ssize_t) offset)
+ {
+ error ("could not find archive member @0x%lx", (long) offset);
+ goto fail;
+ }
+
+ /* Now seek back 12 chars and read the tail of the AR header to
+ find the length of the member file. */
+ if (lseek (mach_o_file->fd, -12, SEEK_CUR) < 0
+ || read (mach_o_file->fd, ar_tail, 12) != 12
+ || lseek (mach_o_file->fd, 0, SEEK_CUR) != (ssize_t) offset
+ || ar_tail[10] != '`' || ar_tail[11] != '\n')
+ {
+ error ("could not find archive header @0x%lx", (long) offset);
+ goto fail;
+ }
+
+ ar_tail[11] = 0;
+ if (sscanf (ar_tail, "%d", &size) != 1)
+ {
+ error ("invalid archive header @0x%lx", (long) offset);
+ goto fail;
+ }
+ mach_o_file->file_size = size;
+ }
+
+ if (writable)
+ {
+ init_mach_o_header (mach_o_file);
+ }
+ else
+ if (! validate_file (mach_o_file))
+ goto fail;
+
+ return result;
+
+ fail:
+ if (result)
+ lto_obj_file_close (result);
+ return NULL;
+}
+
+
+/* Write the data in MACH_O_FILE to a real Mach-O binary object.
+ We write a header, a segment load command, and section data. */
+
+static bool
+mach_o_write_object_file (lto_mach_o_file *mach_o_file)
+{
+ lto_mach_o_section sec, snsec;
+ lto_mach_o_data snsec_data;
+ ssize_t hdrsize, cmdsize, secsize;
+ size_t num_sections, snsec_size, total_sec_size;
+ unsigned int sec_offs, strtab_offs;
+ int i;
+ bool write_err = false;
+
+ /* The number of sections we will write is the number of sections added by
+ the streamer, plus 1 for the section names section. */
+ num_sections = VEC_length (lto_mach_o_section, mach_o_file->section_vec) + 1;
+
+ /* Calculate the size of the basic data structures on disk. */
+ if (mach_o_word_size () == 64)
+ {
+ hdrsize = sizeof (mach_o_header_64);
+ secsize = sizeof (mach_o_section_64);
+ cmdsize = sizeof (mach_o_segment_command_64) + num_sections * secsize;
+ }
+ else
+ {
+ hdrsize = sizeof (mach_o_header_32);
+ secsize = sizeof (mach_o_section_32);
+ cmdsize = sizeof (mach_o_segment_command_32) + num_sections * secsize;
+ }
+
+ /* Allocate the section names section. */
+ snsec_size = 0;
+ for (i = 0;
+ VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ snsec_size += strlen (sec->name) + 1;
+ snsec = mach_o_new_section (mach_o_file, NULL);
+ snsec->name = LTO_NAMES_SECTION;
+ snsec_data = mach_o_new_data (snsec);
+ snsec_data->d_buf = XCNEWVEC (char, snsec_size);
+ snsec_data->d_size = snsec_size;
+
+ /* Position all the sections, and fill out their headers. */
+ sec_offs = hdrsize + cmdsize;
+ strtab_offs = 0;
+ total_sec_size = 0;
+ for (i = 0;
+ VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ {
+ lto_mach_o_data data;
+ size_t data_size;
+ /* Put the section and segment names. Add the section name to the
+ section names section (unless, of course, this *is* the section
+ names section). */
+ if (sec == snsec)
+ snprintf (sec->u.section.sectname, 16, "%s", LTO_NAMES_SECTION);
+ else
+ {
+ sprintf (sec->u.section.sectname, "__%08X", strtab_offs);
+ memcpy ((char *) snsec_data->d_buf + strtab_offs, sec->name, strlen (sec->name));
+ }
+ memcpy (&sec->u.section.segname[0],
+ LTO_SEGMENT_NAME, strlen (LTO_SEGMENT_NAME));
+
+ /* Add layout and attributes. */
+ for (data = sec->data_chain, data_size = 0; data; data = data->next)
+ data_size += data->d_size;
+ if (mach_o_word_size () == 64)
+ {
+ put_uint64 (&sec->u.section_64.addr[0], total_sec_size);
+ put_uint64 (&sec->u.section_64.size[0], data_size);
+ put_uint32 (&sec->u.section_64.offset[0], sec_offs);
+ put_uint32 (&sec->u.section_64.flags[0], MACH_O_S_ATTR_DEBUG);
+ }
+ else
+ {
+ put_uint32 (&sec->u.section_64.addr[0], total_sec_size);
+ put_uint32 (&sec->u.section_32.size[0], data_size);
+ put_uint32 (&sec->u.section_32.offset[0], sec_offs);
+ put_uint32 (&sec->u.section_32.flags[0], MACH_O_S_ATTR_DEBUG);
+ }
+
+ sec_offs += data_size;
+ total_sec_size += data_size;
+ strtab_offs += strlen (sec->name) + 1;
+ }
+
+ /* We can write the data now. As there's no way to indicate an error return
+ from this hook, error handling is limited to not wasting our time doing
+ any more writes in the event that any one fails. */
+
+ /* Write the header. */
+ put_uint32 (&mach_o_file->u.header.ncmds[0], 1);
+ put_uint32 (&mach_o_file->u.header.sizeofcmds[0], cmdsize);
+ write_err = (write (mach_o_file->fd,
+ &mach_o_file->u.header, hdrsize) != hdrsize);
+ /* Write the segment load command. */
+ if (mach_o_word_size () == 64)
+ {
+ mach_o_segment_command_64 lc;
+ ssize_t lc_size = sizeof (lc);
+ memset (&lc, 0, lc_size);
+ put_uint32 (&lc.cmd[0], MACH_O_LC_SEGMENT_64);
+ put_uint32 (&lc.cmdsize[0], cmdsize);
+ put_uint64 (&lc.fileoff[0], hdrsize + cmdsize);
+ put_uint64 (&lc.filesize[0], total_sec_size);
+ put_uint32 (&lc.nsects[0], num_sections);
+ write_err = (write (mach_o_file->fd, &lc, lc_size) != lc_size);
+ }
+ else
+ {
+ mach_o_segment_command_32 lc;
+ ssize_t lc_size = sizeof (lc);
+ memset (&lc, 0, lc_size);
+ put_uint32 (&lc.cmd[0], MACH_O_LC_SEGMENT);
+ put_uint32 (&lc.cmdsize[0], cmdsize);
+ put_uint32 (&lc.fileoff[0], hdrsize + cmdsize);
+ put_uint32 (&lc.filesize[0], total_sec_size);
+ put_uint32 (&lc.nsects[0], num_sections);
+ write_err = (write (mach_o_file->fd, &lc, lc_size) != lc_size);
+ }
+ for (i = 0;
+ !write_err
+ && VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ write_err = (write (mach_o_file->fd,
+ &sec->u.section, secsize) != secsize);
+
+ gcc_assert (lseek (mach_o_file->fd, 0, SEEK_CUR) == hdrsize + cmdsize);
+
+ /* Write the section data. */
+ for (i = 0;
+ !write_err
+ && VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ {
+ lto_mach_o_data data;
+
+ for (data = sec->data_chain; data; data = data->next)
+ {
+ if (!write_err)
+ write_err = (write (mach_o_file->fd, data->d_buf, data->d_size)
+ != data->d_size);
+ else
+ break;
+ }
+ }
+
+ return !write_err;
+}
+
+/* Close Mach-O file FILE and clean up any associated data structures. If FILE
+ was opened for writing, the file's Mach-O data is written at this time. Any
+ cached data buffers are freed. */
+
+void
+lto_obj_file_close (lto_file *file)
+{
+ lto_mach_o_file *mach_o_file = (lto_mach_o_file *) file;
+ struct lto_char_ptr_base *cur, *tmp;
+ lto_mach_o_section sec;
+ bool write_err = false;
+ int i;
+
+ /* If this file is open for writing, write a Mach-O object file. */
+ if (mach_o_file->writable)
+ {
+ if (! mach_o_write_object_file (mach_o_file))
+ fatal_error ("cannot write Mach-O object file");
+ }
+
+ /* Close the file, we're done. */
+ if (mach_o_file->fd != -1)
+ close (mach_o_file->fd);
+
+ /* Free any data buffers. */
+ cur = mach_o_file->data;
+ while (cur)
+ {
+ tmp = cur;
+ cur = (struct lto_char_ptr_base *) cur->ptr;
+ free (tmp);
+ }
+
+ /* Free any sections and their data chains. */
+ for (i = 0;
+ VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ {
+ lto_mach_o_data curdata, nextdata;
+ curdata = sec->data_chain;
+ while (curdata)
+ {
+ nextdata = curdata->next;
+ free (curdata);
+ curdata = nextdata;
+ }
+ free (sec);
+ }
+ VEC_free (lto_mach_o_section, heap, mach_o_file->section_vec);
+
+ free (file);
+
+ /* If there was an error, mention it. */
+ if (write_err)
+ error ("I/O error writing Mach-O output file");
+}
+
===================================================================
@@ -0,0 +1,969 @@
+/* LTO routines for Mach-O object files.
+ Copyright 2010 Free Software Foundation, Inc.
+ Contributed by Steven Bosscher.
+
+This file is part of GCC.
+
+GCC is free software; you can redistribute it and/or modify it under
+the terms of the GNU General Public License as published by the Free
+Software Foundation; either version 3, or (at your option) any later
+version.
+
+GCC is distributed in the hope that it will be useful, but WITHOUT ANY
+WARRANTY; without even the implied warranty of MERCHANTABILITY or
+FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
+for more details.
+
+You should have received a copy of the GNU General Public License
+along with GCC; see the file COPYING3. If not see
+<http://www.gnu.org/licenses/>. */
+
+#include "config.h"
+#include "system.h"
+#include "coretypes.h"
+#include "toplev.h"
+#include "lto.h"
+#include "tm.h"
+#include "libiberty.h"
+#include "lto-streamer.h"
+#include "lto/lto-endian.h"
+#include "lto/lto-macho.h"
+
+/* Rather than implementing a libmacho to match libelf, or attempting to
+ integrate libbfd into GCC, this file is a self-contained (and very
+ minimal) Mach-O format object file reader/writer. The generated files
+ will contain a Mach-O header, a number of Mach-O load commands an
+ section headers, the section data itself, and a trailing string table
+ for section names. */
+
+/* This needs to be kept in sync with darwin.c. Better yet, lto-macho.c
+ and lto-macho.h should be moved to config/, and likewise for lto-coff.*
+ and lto-elf.*. */
+
+/* Segment name for LTO sections. */
+#define LTO_SEGMENT_NAME "__GNU_LTO"
+
+/* Section name for LTO section names section. */
+#define LTO_NAMES_SECTION "__section_names"
+
+/* Handle opening elf files on hosts, such as Windows, that may use
+ text file handling that will break binary access. */
+#ifndef O_BINARY
+# define O_BINARY 0
+#endif
+
+/* Cached object file header. We use a header_64 for this, since all
+ the fields we need are in there, in the same position as header_32. */
+mach_o_header_64 cached_mach_o_header;
+uint32_t cached_mach_o_magic;
+
+/* The current output file. */
+static lto_file *current_out_file;
+
+
+/* Is this a 32-bits or 64-bits Mach-O object file? */
+static int
+mach_o_word_size (void)
+{
+ gcc_assert (cached_mach_o_magic != 0);
+ return (cached_mach_o_magic == MACH_O_MH_MAGIC_64
+ || cached_mach_o_magic == MACH_O_MH_CIGAM_64) ? 64 : 32;
+}
+
+/* Sets the current output file to FILE. Returns the old output file or
+ NULL. */
+
+lto_file *
+lto_set_current_out_file (lto_file *file)
+{
+ lto_file *old_file = current_out_file;
+ current_out_file = file;
+ return old_file;
+}
+
+
+/* Returns the current output file. */
+
+lto_file *
+lto_get_current_out_file (void)
+{
+ return current_out_file;
+}
+
+/* Mach-O section structure constructor. */
+
+static lto_mach_o_section
+mach_o_new_section (lto_mach_o_file *mach_o_file, const char *name)
+{
+ lto_mach_o_section ptr;
+
+ /* FIXME We could allocate these things on an obstack. */
+ ptr = XCNEW (struct lto_mach_o_section_d);
+ if (name)
+ {
+ if (strncmp (name, LTO_SECTION_NAME_PREFIX,
+ strlen(LTO_SECTION_NAME_PREFIX)) != 0)
+ sorry ("not implemented: Mach-O writer for non-LTO sections");
+ ptr->name = xstrdup (name);
+ }
+
+ VEC_safe_push (lto_mach_o_section, heap, mach_o_file->section_vec, ptr);
+
+ return ptr;
+}
+
+/* Mach-O section data block structure constructor. */
+
+static lto_mach_o_data
+mach_o_new_data (lto_mach_o_section sec)
+{
+ lto_mach_o_data ptr, *chain_ptr_ptr;
+
+ /* FIXME We could allocate these things on an obstack. */
+ ptr = XCNEW (struct lto_mach_o_data_d);
+
+ chain_ptr_ptr = &sec->data_chain;
+ while (*chain_ptr_ptr)
+ chain_ptr_ptr = &(*chain_ptr_ptr)->next;
+ *chain_ptr_ptr = ptr;
+
+ return ptr;
+}
+
+/* Initialize FILE, an LTO file object for FILENAME. Offset is the
+ offset into FILE where the object is located (e.g. in an archive). */
+
+static void
+lto_file_init (lto_file *file, const char *filename, off_t offset)
+{
+ file->filename = filename;
+ file->offset = offset;
+}
+
+/* Returns a hash code for P. */
+
+static hashval_t
+hash_name (const void *p)
+{
+ const struct lto_section_slot *s = (const struct lto_section_slot *) p;
+ return (hashval_t) htab_hash_string (s->name);
+}
+
+/* Returns nonzero if P1 and P2 are equal. */
+
+static int
+eq_name (const void *p1, const void *p2)
+{
+ const struct lto_section_slot *s1 =
+ (const struct lto_section_slot *) p1;
+ const struct lto_section_slot *s2 =
+ (const struct lto_section_slot *) p2;
+
+ return strcmp (s1->name, s2->name) == 0;
+}
+
+/* Build a hash table whose key is the section names and whose data is
+ the start and size of each section in the .o file. */
+
+htab_t
+lto_obj_build_section_table (lto_file *lto_file)
+{
+ lto_mach_o_file *mach_o_file = (lto_mach_o_file *)lto_file;
+ lto_mach_o_section sec;
+ htab_t section_hash_table;
+ off_t strtab_offs;
+ ssize_t strtab_size;
+ char *strtab = NULL;
+ int i;
+
+ section_hash_table = htab_create (37, hash_name, eq_name, free);
+
+ /* Seek the string table. */
+ /* FIXME The segment name should be in darwin.h, but can we include it
+ here in this file? */
+ for (i = 0;
+ VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ {
+ if (strncmp (sec->u.section.segname, "__GNU_LTO", 16) != 0)
+ continue;
+ if (strncmp (sec->u.section.sectname, "__section_names", 16) == 0)
+ break;
+ }
+ if (! sec)
+ {
+ error ("invalid Mach-O LTO object file: no __section_names section found");
+ goto done;
+ }
+ mach_o_file->section_names_section = sec;
+
+ if (mach_o_word_size () == 64)
+ {
+ strtab_offs = (off_t) get_uint32 (&sec->u.section_64.offset[0]);
+ strtab_size = (size_t) get_uint64 (&sec->u.section_64.size[0]);
+ }
+ else
+ {
+ strtab_offs = (off_t) get_uint32 (&sec->u.section_32.offset[0]);
+ strtab_size = (size_t) get_uint32 (&sec->u.section_32.size[0]);
+ }
+
+ /* Seek to start of string table. */
+ if (strtab_offs != lseek (mach_o_file->fd,
+ mach_o_file->base.offset + strtab_offs,
+ SEEK_SET))
+ {
+ error ("altered or invalid Mach-O object file");
+ goto done;
+ }
+
+ strtab = XNEWVEC (char, strtab_size);
+ if (read (mach_o_file->fd, strtab, strtab_size) != strtab_size)
+ {
+ error ("invalid Mach-O LTO object file __section_names section");
+ goto done;
+ }
+
+ /* Scan sections looking at names. */
+ for (i = 0;
+ VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ {
+ struct lto_section_slot s_slot;
+ void **slot;
+ char *new_name;
+ unsigned long stringoffset;
+ char name[17];
+
+ /* Ignore non-LTO sections. Also ignore the __section_names section
+ which does not need renaming. */
+ if (strncmp (sec->u.section.segname, "__GNU_LTO", 16) != 0)
+ continue;
+ if (sec == mach_o_file->section_names_section)
+ continue;
+
+ /* Try to extract the offset of the real name for this section from
+ __section_names. */
+ memcpy (&name[0], sec->u.section.sectname, 16);
+ name[16] = '\0';
+ if (name[0] != '_' || name[1] != '_'
+ || sscanf (&name[2], "%08lX", &stringoffset) != 1
+ || strtab_size < (ssize_t) stringoffset)
+ {
+ error ("invalid Mach-O LTO section name string: %s", name);
+ continue;
+ }
+
+ new_name = XNEWVEC (char, strlen (strtab + stringoffset) + 1);
+ strcpy (new_name, strtab + stringoffset);
+ s_slot.name = new_name;
+ slot = htab_find_slot (section_hash_table, &s_slot, INSERT);
+ if (*slot == NULL)
+ {
+ struct lto_section_slot *new_slot = XNEW (struct lto_section_slot);
+
+ new_slot->name = new_name;
+ if (mach_o_word_size() == 64)
+ {
+ new_slot->start =
+ (intptr_t) get_uint32 (&sec->u.section_64.offset[0]);
+ new_slot->len =
+ (size_t) get_uint64 (&sec->u.section_64.size[0]);
+ }
+ else
+ {
+ new_slot->start =
+ (intptr_t) get_uint32 (&sec->u.section_32.offset[0]);
+ new_slot->len =
+ (size_t) get_uint32 (&sec->u.section_32.size[0]);
+ }
+
+ *slot = new_slot;
+ }
+ else
+ {
+ error ("two or more sections for %s:", new_name);
+ goto done;
+ }
+ }
+
+ done:
+ if (strtab)
+ free (strtab);
+ return section_hash_table;
+}
+
+
+/* Begin a new Mach-O section named NAME in the current output file. */
+
+void
+lto_obj_begin_section (const char *name)
+{
+ lto_mach_o_file *file;
+
+ if (strncmp (name, LTO_SECTION_NAME_PREFIX,
+ strlen(LTO_SECTION_NAME_PREFIX)) != 0)
+ sorry ("not implemented: Mach-O writer for non-LTO sections");
+
+ /* Grab the current output file and do some basic assertion checking. */
+ file = (lto_mach_o_file *) lto_get_current_out_file (),
+ gcc_assert (file && file->writable && !file->scn);
+
+ /* Create a new section. */
+ file->scn = mach_o_new_section (file, name);
+ if (!file->scn)
+ fatal_error ("could not create a new Mach-O section: %m");
+}
+
+
+/* Append DATA of length LEN to the current output section. BASE is a pointer
+ to the output page containing DATA. It is freed once the output file has
+ been written. */
+
+void
+lto_obj_append_data (const void *data, size_t len, void *block)
+{
+ lto_mach_o_file *file;
+ lto_mach_o_data mach_o_data;
+ struct lto_char_ptr_base *base = (struct lto_char_ptr_base *) block;
+
+ /* Grab the current output file and do some basic assertion checking. */
+ file = (lto_mach_o_file *) lto_get_current_out_file ();
+ gcc_assert (file);
+ gcc_assert (file->scn);
+
+ mach_o_data = mach_o_new_data (file->scn);
+ if (!mach_o_data)
+ fatal_error ("could not append data to Mach-O section: %m");
+
+ mach_o_data->d_buf = CONST_CAST (void *, data);
+ mach_o_data->d_size = len;
+
+ /* Chain all data blocks (from all sections) on one singly-linked
+ list for freeing en masse after the file is closed. */
+ base->ptr = (char *)file->data;
+ file->data = base;
+}
+
+
+/* End the current output section. This just does some assertion checking
+ and sets the current output file's scn member to NULL. */
+
+void
+lto_obj_end_section (void)
+{
+ lto_mach_o_file *file;
+
+ /* Grab the current output file and validate some basic assertions. */
+ file = (lto_mach_o_file *) lto_get_current_out_file ();
+ gcc_assert (file);
+ gcc_assert (file->scn);
+
+ file->scn = NULL;
+}
+
+
+/* Read a Mach-O header from MACH_O_FILE and validate it.
+ The file descriptor in MACH_O_FILE points at the start of the file.
+ If cached_mach_o_header is uninitialized, caches the results.
+ On succes, returns true and moves file pointer to the start of the
+ load commands. On failure, returns false. */
+
+static bool
+validate_mach_o_header (lto_mach_o_file *mach_o_file)
+{
+ ssize_t i, n;
+ unsigned char magic[4];
+ uint32_t cputype;
+ off_t startpos;
+
+ /* Known header magics for validation, as an array. */
+ static const unsigned int mach_o_known_formats[] = {
+ MACH_O_MH_MAGIC,
+ MACH_O_MH_CIGAM,
+ MACH_O_MH_MAGIC_64,
+ MACH_O_MH_CIGAM_64,
+ };
+#define MACH_O_NUM_KNOWN_FORMATS \
+ ((ssize_t) ARRAY_SIZE (mach_o_known_formats))
+
+ startpos = lseek (mach_o_file->fd, 0, SEEK_CUR);
+ if (read (mach_o_file->fd, &magic, sizeof (magic)) != 4
+ || lseek (mach_o_file->fd, -4, SEEK_CUR) != startpos)
+ {
+ error ("cannot read file %s", mach_o_file->base.filename);
+ return false;
+ }
+
+ for (i = 0; i < MACH_O_NUM_KNOWN_FORMATS; ++i)
+ if (get_uint32 (&magic[0]) == mach_o_known_formats[i])
+ break;
+ if (i == MACH_O_NUM_KNOWN_FORMATS)
+ goto not_for_target;
+
+ /* Check the endian-ness. */
+ if (BYTES_BIG_ENDIAN && magic[0] != 0xfe)
+ goto not_for_target;
+
+ /* Set or check cached magic number. */
+ if (cached_mach_o_magic == 0)
+ cached_mach_o_magic = get_uint32 (&magic[0]);
+ else if (cached_mach_o_magic != get_uint32 (&magic[0]))
+ goto not_for_target;
+
+ n = mach_o_word_size () == 64
+ ? sizeof (mach_o_header_64) : sizeof (mach_o_header_32);
+ if (read (mach_o_file->fd, &mach_o_file->u.header, n) != n)
+ goto not_for_target;
+
+ /* Is this a supported CPU? */
+ /* ??? Would be nice to validate the exact target architecture. */
+ cputype = get_uint32 (&mach_o_file->u.header.cputype[0]);
+ if (cputype == MACH_O_CPU_TYPE_I386
+ || cputype == MACH_O_CPU_TYPE_POWERPC)
+ {
+ if (mach_o_word_size () != 32)
+ goto not_for_target;
+ }
+ else if (cputype == MACH_O_CPU_TYPE_X86_64
+ || cputype == MACH_O_CPU_TYPE_POWERPC_64)
+ {
+ if (mach_o_word_size () != 64)
+ goto not_for_target;
+ }
+
+ /* Is this an MH_OBJECT file? */
+ if (get_uint32 (&mach_o_file->u.header.filetype[0]) != MACH_O_MH_OBJECT)
+ error ("Mach-O file %s is not an MH_OBJECT file",
+ mach_o_file->base.filename);
+
+ /* Save the header for future use. */
+ memcpy (&cached_mach_o_header, &mach_o_file->u.header,
+ sizeof (cached_mach_o_header));
+
+ return true;
+
+ not_for_target:
+ error ("file %s is not a Mach-O object file for target",
+ mach_o_file->base.filename);
+ return false;
+}
+
+
+/* Read a Mach-O LC_SEGMENT command (32 bits) from MACH_O_FILE and
+ validate it.
+ The file descriptor in MACH_O_FILE points at the start of the load
+ command. On sucess, returns true and advances the file pointer
+ past the end of the load command. On failure, returns false. */
+
+static bool
+validate_mach_o_segment_command_32 (lto_mach_o_file *mach_o_file)
+{
+ mach_o_segment_command_32 seg_cmd_32;
+ unsigned int i;
+ ssize_t n;
+ off_t startpos;
+
+ /* Fields we're interested in. */
+ uint32_t cmd;
+ uint32_t cmdsize;
+ uint32_t nsects;
+
+ startpos = lseek (mach_o_file->fd, 0, SEEK_CUR);
+
+ n = sizeof (mach_o_segment_command_32);
+ if (read (mach_o_file->fd, (void *) &seg_cmd_32, n) != n)
+ goto fail;
+
+ cmd = get_uint32 (&seg_cmd_32.cmd[0]);
+ cmdsize = get_uint32 (&seg_cmd_32.cmdsize[0]);
+ nsects = get_uint32 (&seg_cmd_32.nsects[0]);
+ gcc_assert (cmd == MACH_O_LC_SEGMENT);
+
+ /* Validate section table entries. */
+ for (i = 0; i < nsects; i++)
+ {
+ mach_o_section_32 sec_32;
+ lto_mach_o_section ltosec;
+
+ n = sizeof (mach_o_section_32);
+ if (read (mach_o_file->fd, &sec_32, n) != n)
+ goto fail;
+
+ /* ??? Perform some checks. */
+
+ /* Looks ok, so record its details. We don't read the
+ string table or set up names yet; we'll do that when
+ we build the hash table. */
+ ltosec = mach_o_new_section (mach_o_file, NULL);
+ memcpy (<osec->u.section_32, &sec_32, sizeof (sec_32));
+ }
+
+ if (lseek (mach_o_file->fd, 0, SEEK_CUR) != startpos + cmdsize)
+ goto fail;
+
+ return true;
+
+ fail:
+ error ("could not read LC_SEGMENT command in Mach-O file %s",
+ mach_o_file->base.filename);
+ return false;
+}
+
+
+/* Read a Mach-O LC_SEGMENT_64 command from MACH_O_FILE and validate it.
+ The file descriptor in MACH_O_FILE points at the start of the load
+ command. On sucess, returns true and advances the file pointer
+ past the end of the load command. On failure, returns false. */
+
+static bool
+validate_mach_o_segment_command_64 (lto_mach_o_file *mach_o_file)
+{
+ mach_o_segment_command_64 seg_cmd_64;
+ unsigned int i;
+ ssize_t n;
+ off_t startpos;
+
+ /* Fields we're interested in. */
+ uint32_t cmd;
+ uint32_t cmdsize;
+ uint32_t nsects;
+
+ startpos = lseek (mach_o_file->fd, 0, SEEK_CUR);
+
+ n = sizeof (mach_o_segment_command_64);
+ if (read (mach_o_file->fd, (void *) &seg_cmd_64, n) != n)
+ goto fail;
+
+ cmd = get_uint32 (&seg_cmd_64.cmd[0]);
+ cmdsize = get_uint32 (&seg_cmd_64.cmdsize[0]);
+ nsects = get_uint32 (&seg_cmd_64.nsects[0]);
+ gcc_assert (cmd == MACH_O_LC_SEGMENT_64);
+
+ /* Validate section table entries. */
+ for (i = 0; i < nsects; i++)
+ {
+ mach_o_section_64 sec_64;
+ lto_mach_o_section ltosec;
+
+ n = sizeof (mach_o_section_64);
+ if (read (mach_o_file->fd, &sec_64, n) != n)
+ goto fail;
+
+ /* ??? Perform some checks. */
+
+ /* Looks ok, so record its details. We don't read the
+ string table or set up names yet; we'll do that when
+ we build the hash table. */
+ ltosec = mach_o_new_section (mach_o_file, NULL);
+ memcpy (<osec->u.section_64, &sec_64, sizeof (sec_64));
+ }
+
+ if (lseek (mach_o_file->fd, 0, SEEK_CUR) != startpos + cmdsize)
+ goto fail;
+
+ return true;
+
+ fail:
+ error ("could not read LC_SEGMENT_64 command in Mach-O file %s",
+ mach_o_file->base.filename);
+ return false;
+}
+
+/* Read a Mach-O load commands from MACH_O_FILE and validate it.
+ The file descriptor in MACH_O_FILE points at the start of the load
+ command. On sucess, returns true and advances the file pointer
+ past the end of the load command. On failure, returns false. */
+
+static bool
+validate_mach_o_load_command (lto_mach_o_file *mach_o_file)
+{
+ mach_o_load_command load_command;
+ uint32_t cmd;
+ uint32_t cmdsize;
+ ssize_t n;
+
+ n = sizeof (load_command);
+ if (read (mach_o_file->fd, &load_command, n) != n)
+ {
+ error ("could not read load commands in Mach-O file %s",
+ mach_o_file->base.filename);
+ return false;
+ }
+ lseek (mach_o_file->fd, -1 * (off_t) sizeof (load_command), SEEK_CUR);
+
+ cmd = get_uint32 (&load_command.cmd[0]);
+ cmdsize = get_uint32 (&load_command.cmdsize[0]);
+ switch (cmd)
+ {
+ case MACH_O_LC_SEGMENT:
+ return validate_mach_o_segment_command_32 (mach_o_file);
+ case MACH_O_LC_SEGMENT_64:
+ return validate_mach_o_segment_command_64 (mach_o_file);
+
+ default:
+ /* Just skip over it. */
+ lseek (mach_o_file->fd, cmdsize, SEEK_CUR);
+ return true;
+ }
+}
+
+/* Validate's MACH_O_FILE's executable header and, if cached_mach_o_header is
+ uninitialized, caches the results. Also records the section header string
+ table's section index. Returns true on success, false on failure. */
+
+static bool
+validate_file (lto_mach_o_file *mach_o_file)
+{
+ uint32_t i, ncmds;
+
+ /* Read and sanity check the raw header. */
+ if (! validate_mach_o_header (mach_o_file))
+ return false;
+
+ ncmds = get_uint32 (&mach_o_file->u.header.ncmds[0]);
+ for (i = 0; i < ncmds; ++i)
+ if (! validate_mach_o_load_command (mach_o_file))
+ return false;
+
+ return true;
+}
+
+/* Initialize MACH_O_FILE's executable header using cached data from previously
+ read files. */
+
+static void
+init_mach_o_header (lto_mach_o_file *mach_o_file)
+{
+ gcc_assert (cached_mach_o_magic != 0);
+ memcpy (&mach_o_file->u.header,
+ &cached_mach_o_header,
+ sizeof (mach_o_file->u.header));
+ put_uint32 (&mach_o_file->u.header.ncmds[0], 0);
+ put_uint32 (&mach_o_file->u.header.sizeofcmds[0], 0);
+}
+
+/* Open Mach-O file FILENAME. If WRITABLE is true, the file is opened for write
+ and, if necessary, created. Otherwise, the file is opened for reading.
+ Returns the opened file. */
+
+lto_file *
+lto_obj_file_open (const char *filename, bool writable)
+{
+ lto_mach_o_file *mach_o_file;
+ lto_file *result = NULL;
+ off_t offset;
+ const char *offset_p;
+ char *fname;
+ struct stat statbuf;
+
+ offset_p = strchr (filename, '@');
+ if (!offset_p)
+ {
+ fname = xstrdup (filename);
+ offset = 0;
+ }
+ else
+ {
+ /* The file started with '@' is a file containing command line
+ options. Stop if it doesn't exist. */
+ if (offset_p == filename)
+ fatal_error ("command line option file '%s' does not exist",
+ filename);
+
+ fname = (char *) xmalloc (offset_p - filename + 1);
+ memcpy (fname, filename, offset_p - filename);
+ fname[offset_p - filename] = '\0';
+ offset_p += 3; /* skip the @0x */
+ offset = lto_parse_hex (offset_p);
+ }
+
+ /* Set up. */
+ mach_o_file = XCNEW (lto_mach_o_file);
+ result = (lto_file *) mach_o_file;
+ lto_file_init (result, fname, offset);
+ mach_o_file->fd = -1;
+ mach_o_file->writable = writable;
+
+ /* Open the file. */
+ mach_o_file->fd = open (fname,
+ O_BINARY | (writable ? O_WRONLY | O_CREAT | O_TRUNC : O_RDONLY), 0666);
+
+ if (mach_o_file->fd == -1)
+ {
+ error ("could not open file %s", fname);
+ goto fail;
+ }
+
+ if (stat (fname, &statbuf) < 0)
+ {
+ error ("could not stat file %s", fname);
+ goto fail;
+ }
+
+ mach_o_file->file_size = statbuf.st_size;
+
+ /* If the object is in an archive, get it out. */
+ if (offset != 0)
+ {
+ char ar_tail[12];
+ int size;
+
+ /* Surely not? */
+ gcc_assert (!writable);
+
+ /* Seek to offset, or error. */
+ if (lseek (mach_o_file->fd, offset, SEEK_SET) != (ssize_t) offset)
+ {
+ error ("could not find archive member @0x%lx", (long) offset);
+ goto fail;
+ }
+
+ /* Now seek back 12 chars and read the tail of the AR header to
+ find the length of the member file. */
+ if (lseek (mach_o_file->fd, -12, SEEK_CUR) < 0
+ || read (mach_o_file->fd, ar_tail, 12) != 12
+ || lseek (mach_o_file->fd, 0, SEEK_CUR) != (ssize_t) offset
+ || ar_tail[10] != '`' || ar_tail[11] != '\n')
+ {
+ error ("could not find archive header @0x%lx", (long) offset);
+ goto fail;
+ }
+
+ ar_tail[11] = 0;
+ if (sscanf (ar_tail, "%d", &size) != 1)
+ {
+ error ("invalid archive header @0x%lx", (long) offset);
+ goto fail;
+ }
+ mach_o_file->file_size = size;
+ }
+
+ if (writable)
+ {
+ init_mach_o_header (mach_o_file);
+ }
+ else
+ if (! validate_file (mach_o_file))
+ goto fail;
+
+ return result;
+
+ fail:
+ if (result)
+ lto_obj_file_close (result);
+ return NULL;
+}
+
+
+/* Write the data in MACH_O_FILE to a real Mach-O binary object.
+ We write a header, a segment load command, and section data. */
+
+static bool
+mach_o_write_object_file (lto_mach_o_file *mach_o_file)
+{
+ lto_mach_o_section sec, snsec;
+ lto_mach_o_data snsec_data;
+ ssize_t hdrsize, cmdsize, secsize;
+ size_t num_sections, snsec_size, total_sec_size;
+ unsigned int sec_offs, strtab_offs;
+ int i;
+ bool write_err = false;
+
+ /* The number of sections we will write is the number of sections added by
+ the streamer, plus 1 for the section names section. */
+ num_sections = VEC_length (lto_mach_o_section, mach_o_file->section_vec) + 1;
+
+ /* Calculate the size of the basic data structures on disk. */
+ if (mach_o_word_size () == 64)
+ {
+ hdrsize = sizeof (mach_o_header_64);
+ secsize = sizeof (mach_o_section_64);
+ cmdsize = sizeof (mach_o_segment_command_64) + num_sections * secsize;
+ }
+ else
+ {
+ hdrsize = sizeof (mach_o_header_32);
+ secsize = sizeof (mach_o_section_32);
+ cmdsize = sizeof (mach_o_segment_command_32) + num_sections * secsize;
+ }
+
+ /* Allocate the section names section. */
+ snsec_size = 0;
+ for (i = 0;
+ VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ snsec_size += strlen (sec->name) + 1;
+ snsec = mach_o_new_section (mach_o_file, NULL);
+ snsec->name = LTO_NAMES_SECTION;
+ snsec_data = mach_o_new_data (snsec);
+ snsec_data->d_buf = XCNEWVEC (char, snsec_size);
+ snsec_data->d_size = snsec_size;
+
+ /* Position all the sections, and fill out their headers. */
+ sec_offs = hdrsize + cmdsize;
+ strtab_offs = 0;
+ total_sec_size = 0;
+ for (i = 0;
+ VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ {
+ lto_mach_o_data data;
+ size_t data_size;
+ /* Put the section and segment names. Add the section name to the
+ section names section (unless, of course, this *is* the section
+ names section). */
+ if (sec == snsec)
+ snprintf (sec->u.section.sectname, 16, "%s", LTO_NAMES_SECTION);
+ else
+ {
+ sprintf (sec->u.section.sectname, "__%08X", strtab_offs);
+ memcpy ((char *) snsec_data->d_buf + strtab_offs, sec->name, strlen (sec->name));
+ }
+ memcpy (&sec->u.section.segname[0],
+ LTO_SEGMENT_NAME, strlen (LTO_SEGMENT_NAME));
+
+ /* Add layout and attributes. */
+ for (data = sec->data_chain, data_size = 0; data; data = data->next)
+ data_size += data->d_size;
+ if (mach_o_word_size () == 64)
+ {
+ put_uint64 (&sec->u.section_64.addr[0], total_sec_size);
+ put_uint64 (&sec->u.section_64.size[0], data_size);
+ put_uint32 (&sec->u.section_64.offset[0], sec_offs);
+ put_uint32 (&sec->u.section_64.flags[0], MACH_O_S_ATTR_DEBUG);
+ }
+ else
+ {
+ put_uint32 (&sec->u.section_64.addr[0], total_sec_size);
+ put_uint32 (&sec->u.section_32.size[0], data_size);
+ put_uint32 (&sec->u.section_32.offset[0], sec_offs);
+ put_uint32 (&sec->u.section_32.flags[0], MACH_O_S_ATTR_DEBUG);
+ }
+
+ sec_offs += data_size;
+ total_sec_size += data_size;
+ strtab_offs += strlen (sec->name) + 1;
+ }
+
+ /* We can write the data now. As there's no way to indicate an error return
+ from this hook, error handling is limited to not wasting our time doing
+ any more writes in the event that any one fails. */
+
+ /* Write the header. */
+ put_uint32 (&mach_o_file->u.header.ncmds[0], 1);
+ put_uint32 (&mach_o_file->u.header.sizeofcmds[0], cmdsize);
+ write_err = (write (mach_o_file->fd,
+ &mach_o_file->u.header, hdrsize) != hdrsize);
+ /* Write the segment load command. */
+ if (mach_o_word_size () == 64)
+ {
+ mach_o_segment_command_64 lc;
+ ssize_t lc_size = sizeof (lc);
+ memset (&lc, 0, lc_size);
+ put_uint32 (&lc.cmd[0], MACH_O_LC_SEGMENT_64);
+ put_uint32 (&lc.cmdsize[0], cmdsize);
+ put_uint64 (&lc.fileoff[0], hdrsize + cmdsize);
+ put_uint64 (&lc.filesize[0], total_sec_size);
+ put_uint32 (&lc.nsects[0], num_sections);
+ write_err = (write (mach_o_file->fd, &lc, lc_size) != lc_size);
+ }
+ else
+ {
+ mach_o_segment_command_32 lc;
+ ssize_t lc_size = sizeof (lc);
+ memset (&lc, 0, lc_size);
+ put_uint32 (&lc.cmd[0], MACH_O_LC_SEGMENT);
+ put_uint32 (&lc.cmdsize[0], cmdsize);
+ put_uint32 (&lc.fileoff[0], hdrsize + cmdsize);
+ put_uint32 (&lc.filesize[0], total_sec_size);
+ put_uint32 (&lc.nsects[0], num_sections);
+ write_err = (write (mach_o_file->fd, &lc, lc_size) != lc_size);
+ }
+ for (i = 0;
+ !write_err
+ && VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ write_err = (write (mach_o_file->fd,
+ &sec->u.section, secsize) != secsize);
+
+ gcc_assert (lseek (mach_o_file->fd, 0, SEEK_CUR) == hdrsize + cmdsize);
+
+ /* Write the section data. */
+ for (i = 0;
+ !write_err
+ && VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ {
+ lto_mach_o_data data;
+
+ for (data = sec->data_chain; data; data = data->next)
+ {
+ if (!write_err)
+ write_err = (write (mach_o_file->fd, data->d_buf, data->d_size)
+ != data->d_size);
+ else
+ break;
+ }
+ }
+
+ return !write_err;
+}
+
+/* Close Mach-O file FILE and clean up any associated data structures. If FILE
+ was opened for writing, the file's Mach-O data is written at this time. Any
+ cached data buffers are freed. */
+
+void
+lto_obj_file_close (lto_file *file)
+{
+ lto_mach_o_file *mach_o_file = (lto_mach_o_file *) file;
+ struct lto_char_ptr_base *cur, *tmp;
+ lto_mach_o_section sec;
+ bool write_err = false;
+ int i;
+
+ /* If this file is open for writing, write a Mach-O object file. */
+ if (mach_o_file->writable)
+ {
+ if (! mach_o_write_object_file (mach_o_file))
+ fatal_error ("cannot write Mach-O object file");
+ }
+
+ /* Close the file, we're done. */
+ if (mach_o_file->fd != -1)
+ close (mach_o_file->fd);
+
+ /* Free any data buffers. */
+ cur = mach_o_file->data;
+ while (cur)
+ {
+ tmp = cur;
+ cur = (struct lto_char_ptr_base *) cur->ptr;
+ free (tmp);
+ }
+
+ /* Free any sections and their data chains. */
+ for (i = 0;
+ VEC_iterate (lto_mach_o_section, mach_o_file->section_vec, i, sec);
+ i++)
+ {
+ lto_mach_o_data curdata, nextdata;
+ curdata = sec->data_chain;
+ while (curdata)
+ {
+ nextdata = curdata->next;
+ free (curdata);
+ curdata = nextdata;
+ }
+ free (sec);
+ }
+ VEC_free (lto_mach_o_section, heap, mach_o_file->section_vec);
+
+ free (file);
+
+ /* If there was an error, mention it. */
+ if (write_err)
+ error ("I/O error writing Mach-O output file");
+}
+