macho.c 5.21 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
/* macho.c - load Mach-O files. */
/*
 *  GRUB  --  GRand Unified Bootloader
 *  Copyright (C) 2009  Free Software Foundation, Inc.
 *
 *  GRUB 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 of the License, or
 *  (at your option) any later version.
 *
 *  GRUB 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 GRUB.  If not, see <http://www.gnu.org/licenses/>.
 */

/* This Mach-O loader is incomplete and can load only non-relocatable segments.
   This is however enough to boot xnu (otool -l and Mach-O specs for more info).
*/

#include <grub/err.h>
#include <grub/macho.h>
#include <grub/machoload.h>
#include <grub/file.h>
#include <grub/misc.h>
#include <grub/mm.h>
30
#include <grub/i18n.h>
31 32 33
#include <grub/dl.h>

GRUB_MOD_LICENSE ("GPLv3+");
34 35 36 37 38 39

grub_err_t
grub_macho_close (grub_macho_t macho)
{
  grub_file_t file = macho->file;

40 41 42 43 44 45
  if (!macho->uncompressed32)
    grub_free (macho->cmds32);
  grub_free (macho->uncompressed32);
  if (!macho->uncompressed64)
    grub_free (macho->cmds64);
  grub_free (macho->uncompressed64);
46 47 48 49 50 51 52 53 54 55

  grub_free (macho);

  if (file)
    grub_file_close (file);

  return grub_errno;
}

grub_macho_t
56
grub_macho_file (grub_file_t file, const char *filename, int is_64bit)
57 58 59 60 61 62 63 64 65 66 67 68 69 70 71
{
  grub_macho_t macho;
  union grub_macho_filestart filestart;

  macho = grub_malloc (sizeof (*macho));
  if (! macho)
    return 0;

  macho->file = file;
  macho->offset32 = -1;
  macho->offset64 = -1;
  macho->end32 = -1;
  macho->end64 = -1;
  macho->cmds32 = 0;
  macho->cmds64 = 0;
72 73 74 75
  macho->uncompressed32 = 0;
  macho->uncompressed64 = 0;
  macho->compressed32 = 0;
  macho->compressed64 = 0;
76 77 78 79

  if (grub_file_seek (macho->file, 0) == (grub_off_t) -1)
    goto fail;

80
  if (grub_file_read (macho->file, &filestart, sizeof (filestart))
81 82
      != sizeof (filestart))
    {
83 84 85
      if (!grub_errno)
	grub_error (GRUB_ERR_BAD_OS, N_("premature end of file %s"),
		    filename);
86 87 88 89
      goto fail;
    }

  /* Is it a fat file? */
90
  if (filestart.fat.magic == grub_cpu_to_be32_compile_time (GRUB_MACHO_FAT_MAGIC))
91 92 93 94 95 96
    {
      struct grub_macho_fat_arch *archs;
      int i, narchs;

      /* Load architecture description. */
      narchs = grub_be_to_cpu32 (filestart.fat.nfat_arch);
97
      if (grub_file_seek (macho->file, sizeof (struct grub_macho_fat_header))
98 99 100 101 102
	  == (grub_off_t) -1)
	goto fail;
      archs = grub_malloc (sizeof (struct grub_macho_fat_arch) * narchs);
      if (!archs)
	goto fail;
103
      if (grub_file_read (macho->file, archs,
104
			  sizeof (struct grub_macho_fat_arch) * narchs)
105
	  != (grub_ssize_t) sizeof(struct grub_macho_fat_arch) * narchs)
106 107
	{
	  grub_free (archs);
108 109 110
	  if (!grub_errno)
	    grub_error (GRUB_ERR_BAD_OS, N_("premature end of file %s"),
			filename);
111 112 113 114 115
	  goto fail;
	}

      for (i = 0; i < narchs; i++)
	{
116 117 118
	  if ((archs[i].cputype
	       == grub_cpu_to_be32_compile_time (GRUB_MACHO_CPUTYPE_IA32))
	      && !is_64bit)
119 120
	    {
	      macho->offset32 = grub_be_to_cpu32 (archs[i].offset);
121
	      macho->end32 = grub_be_to_cpu32 (archs[i].offset)
122 123
		+ grub_be_to_cpu32 (archs[i].size);
	    }
124 125 126
	  if ((archs[i].cputype
	       == grub_cpu_to_be32_compile_time (GRUB_MACHO_CPUTYPE_AMD64))
	      && is_64bit)
127 128
	    {
	      macho->offset64 = grub_be_to_cpu32 (archs[i].offset);
129
	      macho->end64 = grub_be_to_cpu32 (archs[i].offset)
130 131 132 133 134 135 136
		+ grub_be_to_cpu32 (archs[i].size);
	    }
	}
      grub_free (archs);
    }

  /* Is it a thin 32-bit file? */
137
  if (filestart.thin32.magic == GRUB_MACHO_MAGIC32 && !is_64bit)
138 139 140 141 142 143
    {
      macho->offset32 = 0;
      macho->end32 = grub_file_size (file);
    }

  /* Is it a thin 64-bit file? */
144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159
  if (filestart.thin64.magic == GRUB_MACHO_MAGIC64 && is_64bit)
    {
      macho->offset64 = 0;
      macho->end64 = grub_file_size (file);
    }

  if (grub_memcmp (filestart.lzss.magic, GRUB_MACHO_LZSS_MAGIC,
		   sizeof (filestart.lzss.magic)) == 0 && !is_64bit)
    {
      macho->offset32 = 0;
      macho->end32 = grub_file_size (file);
    }

  /* Is it a thin 64-bit file? */
  if (grub_memcmp (filestart.lzss.magic, GRUB_MACHO_LZSS_MAGIC,
		   sizeof (filestart.lzss.magic)) == 0 && is_64bit)
160 161 162 163 164
    {
      macho->offset64 = 0;
      macho->end64 = grub_file_size (file);
    }

165 166
  grub_macho_parse32 (macho, filename);
  grub_macho_parse64 (macho, filename);
167

168 169 170 171 172 173 174 175 176 177 178 179 180 181
  if (macho->offset32 == -1 && !is_64bit)
    {
      grub_error (GRUB_ERR_BAD_OS,
		  "Mach-O doesn't contain suitable 32-bit architecture");
      goto fail;
    }

  if (macho->offset64 == -1 && is_64bit)
    {
      grub_error (GRUB_ERR_BAD_OS,
		  "Mach-O doesn't contain suitable 64-bit architecture");
      goto fail;
    }

182 183 184
  return macho;

fail:
185
  macho->file = 0;
186 187 188 189 190
  grub_macho_close (macho);
  return 0;
}

grub_macho_t
191
grub_macho_open (const char *name, int is_64bit)
192 193 194 195
{
  grub_file_t file;
  grub_macho_t macho;

196
  file = grub_file_open (name);
197 198 199
  if (! file)
    return 0;

200
  macho = grub_macho_file (file, name, is_64bit);
201 202 203 204 205
  if (! macho)
    grub_file_close (file);

  return macho;
}