Subversion Repositories openrisc_me

/* 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 (&ltosec->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 (&ltosec->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;