384 lines
9.8 KiB
C
384 lines
9.8 KiB
C
|
|
#include <grub/file.h>
|
|
#include <grub/mm.h>
|
|
#include <grub/misc.h>
|
|
#include <grub/i18n.h>
|
|
|
|
#define min(a,b) (((a) < (b)) ? (a) : (b))
|
|
|
|
static int
|
|
SUFFIX (grub_macho_contains_macho) (grub_macho_t macho)
|
|
{
|
|
return macho->offsetXX != -1;
|
|
}
|
|
|
|
void
|
|
SUFFIX (grub_macho_parse) (grub_macho_t macho, const char *filename)
|
|
{
|
|
union {
|
|
struct grub_macho_lzss_header lzss;
|
|
grub_macho_header_t macho;
|
|
} head;
|
|
|
|
/* Is there any candidate at all? */
|
|
if (macho->offsetXX == -1)
|
|
return;
|
|
|
|
/* Read header and check magic. */
|
|
if (grub_file_seek (macho->file, macho->offsetXX) == (grub_off_t) -1
|
|
|| grub_file_read (macho->file, &head, sizeof (head))
|
|
!= sizeof (head))
|
|
{
|
|
if (!grub_errno)
|
|
grub_error (GRUB_ERR_BAD_OS, N_("premature end of file %s"),
|
|
filename);
|
|
macho->offsetXX = -1;
|
|
return;
|
|
}
|
|
if (grub_memcmp (head.lzss.magic, GRUB_MACHO_LZSS_MAGIC,
|
|
sizeof (head.lzss.magic)) == 0)
|
|
{
|
|
macho->compressed_sizeXX = grub_be_to_cpu32 (head.lzss.compressed_size);
|
|
macho->uncompressed_sizeXX
|
|
= grub_be_to_cpu32 (head.lzss.uncompressed_size);
|
|
if (macho->uncompressed_sizeXX < sizeof (head.macho))
|
|
{
|
|
grub_error (GRUB_ERR_BAD_OS, N_("premature end of file %s"),
|
|
filename);
|
|
macho->offsetXX = -1;
|
|
return;
|
|
}
|
|
/* Skip header check. */
|
|
macho->compressedXX = 1;
|
|
return;
|
|
}
|
|
|
|
if (head.macho.magic != GRUB_MACHO_MAGIC)
|
|
{
|
|
grub_error (GRUB_ERR_BAD_OS, "invalid Mach-O header");
|
|
macho->offsetXX = -1;
|
|
return;
|
|
}
|
|
|
|
/* Read commands. */
|
|
macho->ncmdsXX = head.macho.ncmds;
|
|
macho->cmdsizeXX = head.macho.sizeofcmds;
|
|
macho->cmdsXX = grub_malloc (macho->cmdsizeXX);
|
|
if (! macho->cmdsXX)
|
|
return;
|
|
if (grub_file_seek (macho->file, macho->offsetXX
|
|
+ sizeof (grub_macho_header_t)) == (grub_off_t) -1
|
|
|| grub_file_read (macho->file, macho->cmdsXX,
|
|
(grub_size_t) macho->cmdsizeXX)
|
|
!= (grub_ssize_t) macho->cmdsizeXX)
|
|
{
|
|
if (!grub_errno)
|
|
grub_error (GRUB_ERR_BAD_OS, N_("premature end of file %s"),
|
|
filename);
|
|
macho->offsetXX = -1;
|
|
}
|
|
}
|
|
|
|
typedef int (*grub_macho_iter_hook_t)
|
|
(grub_macho_t , struct grub_macho_cmd *,
|
|
void *);
|
|
|
|
static grub_err_t
|
|
grub_macho_cmds_iterate (grub_macho_t macho,
|
|
grub_macho_iter_hook_t hook,
|
|
void *hook_arg,
|
|
const char *filename)
|
|
{
|
|
grub_uint8_t *hdrs;
|
|
int i;
|
|
|
|
if (macho->compressedXX && !macho->uncompressedXX)
|
|
{
|
|
grub_uint8_t *tmp;
|
|
grub_macho_header_t *head;
|
|
macho->uncompressedXX = grub_malloc (macho->uncompressed_sizeXX);
|
|
if (!macho->uncompressedXX)
|
|
return grub_errno;
|
|
tmp = grub_malloc (macho->compressed_sizeXX);
|
|
if (!tmp)
|
|
{
|
|
grub_free (macho->uncompressedXX);
|
|
macho->uncompressedXX = 0;
|
|
return grub_errno;
|
|
}
|
|
if (grub_file_seek (macho->file, macho->offsetXX
|
|
+ GRUB_MACHO_LZSS_OFFSET) == (grub_off_t) -1
|
|
|| grub_file_read (macho->file, tmp,
|
|
(grub_size_t) macho->compressed_sizeXX)
|
|
!= (grub_ssize_t) macho->compressed_sizeXX)
|
|
{
|
|
if (!grub_errno)
|
|
grub_error (GRUB_ERR_BAD_OS, N_("premature end of file %s"),
|
|
filename);
|
|
grub_free (tmp);
|
|
grub_free (macho->uncompressedXX);
|
|
macho->uncompressedXX = 0;
|
|
macho->offsetXX = -1;
|
|
return grub_errno;
|
|
}
|
|
if (grub_decompress_lzss (macho->uncompressedXX,
|
|
macho->uncompressedXX
|
|
+ macho->uncompressed_sizeXX,
|
|
tmp, tmp + macho->compressed_sizeXX)
|
|
!= macho->uncompressed_sizeXX)
|
|
{
|
|
if (!grub_errno)
|
|
grub_error (GRUB_ERR_BAD_OS, N_("premature end of file %s"),
|
|
filename);
|
|
grub_free (tmp);
|
|
grub_free (macho->uncompressedXX);
|
|
macho->uncompressedXX = 0;
|
|
macho->offsetXX = -1;
|
|
return grub_errno;
|
|
}
|
|
grub_free (tmp);
|
|
head = (grub_macho_header_t *) macho->uncompressedXX;
|
|
macho->ncmdsXX = head->ncmds;
|
|
macho->cmdsizeXX = head->sizeofcmds;
|
|
macho->cmdsXX = macho->uncompressedXX + sizeof (grub_macho_header_t);
|
|
if (sizeof (grub_macho_header_t) + macho->cmdsizeXX
|
|
>= macho->uncompressed_sizeXX)
|
|
{
|
|
grub_error (GRUB_ERR_BAD_OS, N_("premature end of file %s"),
|
|
filename);
|
|
grub_free (macho->uncompressedXX);
|
|
macho->uncompressedXX = 0;
|
|
macho->offsetXX = -1;
|
|
return grub_errno;
|
|
}
|
|
}
|
|
|
|
if (! macho->cmdsXX)
|
|
return grub_error (GRUB_ERR_BAD_OS, "couldn't find Mach-O commands");
|
|
hdrs = macho->cmdsXX;
|
|
for (i = 0; i < macho->ncmdsXX; i++)
|
|
{
|
|
struct grub_macho_cmd *hdr = (struct grub_macho_cmd *) hdrs;
|
|
if (hook (macho, hdr, hook_arg))
|
|
break;
|
|
hdrs += hdr->cmdsize;
|
|
}
|
|
|
|
return grub_errno;
|
|
}
|
|
|
|
grub_size_t
|
|
SUFFIX (grub_macho_filesize) (grub_macho_t macho)
|
|
{
|
|
if (SUFFIX (grub_macho_contains_macho) (macho))
|
|
return macho->endXX - macho->offsetXX;
|
|
return 0;
|
|
}
|
|
|
|
grub_err_t
|
|
SUFFIX (grub_macho_readfile) (grub_macho_t macho,
|
|
const char *filename,
|
|
void *dest)
|
|
{
|
|
grub_ssize_t read;
|
|
if (! SUFFIX (grub_macho_contains_macho) (macho))
|
|
return grub_error (GRUB_ERR_BAD_OS,
|
|
"couldn't read architecture-specific part");
|
|
|
|
if (grub_file_seek (macho->file, macho->offsetXX) == (grub_off_t) -1)
|
|
return grub_errno;
|
|
|
|
read = grub_file_read (macho->file, dest,
|
|
macho->endXX - macho->offsetXX);
|
|
if (read != (grub_ssize_t) (macho->endXX - macho->offsetXX))
|
|
{
|
|
if (!grub_errno)
|
|
grub_error (GRUB_ERR_BAD_OS, N_("premature end of file %s"),
|
|
filename);
|
|
return grub_errno;
|
|
}
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
struct calcsize_ctx
|
|
{
|
|
int flags;
|
|
int nr_phdrs;
|
|
grub_macho_addr_t *segments_start;
|
|
grub_macho_addr_t *segments_end;
|
|
};
|
|
|
|
/* Run through the program headers to calculate the total memory size we
|
|
should claim. */
|
|
static int
|
|
calcsize (grub_macho_t _macho __attribute__ ((unused)),
|
|
struct grub_macho_cmd *hdr0,
|
|
void *_arg)
|
|
{
|
|
grub_macho_segment_t *hdr = (grub_macho_segment_t *) hdr0;
|
|
struct calcsize_ctx *ctx = _arg;
|
|
if (hdr->cmd != GRUB_MACHO_CMD_SEGMENT)
|
|
return 0;
|
|
|
|
if (! hdr->vmsize)
|
|
return 0;
|
|
|
|
if (! hdr->filesize && (ctx->flags & GRUB_MACHO_NOBSS))
|
|
return 0;
|
|
|
|
ctx->nr_phdrs++;
|
|
if (hdr->vmaddr < *ctx->segments_start)
|
|
*ctx->segments_start = hdr->vmaddr;
|
|
if (hdr->vmaddr + hdr->vmsize > *ctx->segments_end)
|
|
*ctx->segments_end = hdr->vmaddr + hdr->vmsize;
|
|
return 0;
|
|
}
|
|
|
|
/* Calculate the amount of memory spanned by the segments. */
|
|
grub_err_t
|
|
SUFFIX (grub_macho_size) (grub_macho_t macho, grub_macho_addr_t *segments_start,
|
|
grub_macho_addr_t *segments_end, int flags,
|
|
const char *filename)
|
|
{
|
|
struct calcsize_ctx ctx = {
|
|
.flags = flags,
|
|
.nr_phdrs = 0,
|
|
.segments_start = segments_start,
|
|
.segments_end = segments_end,
|
|
};
|
|
|
|
*segments_start = (grub_macho_addr_t) -1;
|
|
*segments_end = 0;
|
|
|
|
grub_macho_cmds_iterate (macho, calcsize, &ctx, filename);
|
|
|
|
if (ctx.nr_phdrs == 0)
|
|
return grub_error (GRUB_ERR_BAD_OS, "no program headers present");
|
|
|
|
if (*segments_end < *segments_start)
|
|
/* Very bad addresses. */
|
|
return grub_error (GRUB_ERR_BAD_OS, "bad program header load addresses");
|
|
|
|
return GRUB_ERR_NONE;
|
|
}
|
|
|
|
struct do_load_ctx
|
|
{
|
|
int flags;
|
|
char *offset;
|
|
const char *filename;
|
|
int *darwin_version;
|
|
};
|
|
|
|
static int
|
|
do_load(grub_macho_t _macho,
|
|
struct grub_macho_cmd *hdr0,
|
|
void *_arg)
|
|
{
|
|
grub_macho_segment_t *hdr = (grub_macho_segment_t *) hdr0;
|
|
struct do_load_ctx *ctx = _arg;
|
|
|
|
if (hdr->cmd != GRUB_MACHO_CMD_SEGMENT)
|
|
return 0;
|
|
|
|
if (! hdr->filesize && (ctx->flags & GRUB_MACHO_NOBSS))
|
|
return 0;
|
|
if (! hdr->vmsize)
|
|
return 0;
|
|
|
|
if (hdr->filesize)
|
|
{
|
|
grub_ssize_t read, toread = min (hdr->filesize, hdr->vmsize);
|
|
if (_macho->uncompressedXX)
|
|
{
|
|
if (hdr->fileoff + (grub_size_t) toread
|
|
> _macho->uncompressed_sizeXX)
|
|
read = -1;
|
|
else
|
|
{
|
|
read = toread;
|
|
grub_memcpy (ctx->offset + hdr->vmaddr,
|
|
_macho->uncompressedXX + hdr->fileoff, read);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (grub_file_seek (_macho->file, hdr->fileoff
|
|
+ _macho->offsetXX) == (grub_off_t) -1)
|
|
return 1;
|
|
read = grub_file_read (_macho->file, ctx->offset + hdr->vmaddr,
|
|
toread);
|
|
}
|
|
|
|
if (read != toread)
|
|
{
|
|
/* XXX How can we free memory from `load_hook'? */
|
|
if (!grub_errno)
|
|
grub_error (GRUB_ERR_BAD_OS, N_("premature end of file %s"),
|
|
ctx->filename);
|
|
|
|
return 1;
|
|
}
|
|
if (ctx->darwin_version)
|
|
{
|
|
const char *ptr = ctx->offset + hdr->vmaddr;
|
|
const char *end = ptr + min (hdr->filesize, hdr->vmsize)
|
|
- (sizeof ("Darwin Kernel Version ") - 1);
|
|
for (; ptr < end; ptr++)
|
|
if (grub_memcmp (ptr, "Darwin Kernel Version ",
|
|
sizeof ("Darwin Kernel Version ") - 1) == 0)
|
|
{
|
|
ptr += sizeof ("Darwin Kernel Version ") - 1;
|
|
*ctx->darwin_version = 0;
|
|
end += (sizeof ("Darwin Kernel Version ") - 1);
|
|
while (ptr < end && grub_isdigit (*ptr))
|
|
*ctx->darwin_version = (*ptr++ - '0') + *ctx->darwin_version * 10;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (hdr->filesize < hdr->vmsize)
|
|
grub_memset (ctx->offset + hdr->vmaddr + hdr->filesize,
|
|
0, hdr->vmsize - hdr->filesize);
|
|
return 0;
|
|
}
|
|
|
|
/* Load every loadable segment into memory specified by `_load_hook'. */
|
|
grub_err_t
|
|
SUFFIX (grub_macho_load) (grub_macho_t macho, const char *filename,
|
|
char *offset, int flags, int *darwin_version)
|
|
{
|
|
struct do_load_ctx ctx = {
|
|
.flags = flags,
|
|
.offset = offset,
|
|
.filename = filename,
|
|
.darwin_version = darwin_version
|
|
};
|
|
|
|
if (darwin_version)
|
|
*darwin_version = 0;
|
|
|
|
grub_macho_cmds_iterate (macho, do_load, &ctx, filename);
|
|
|
|
return grub_errno;
|
|
}
|
|
|
|
static int
|
|
find_entry_point (grub_macho_t _macho __attribute__ ((unused)),
|
|
struct grub_macho_cmd *hdr,
|
|
void *_arg)
|
|
{
|
|
grub_macho_addr_t *entry_point = _arg;
|
|
if (hdr->cmd == GRUB_MACHO_CMD_THREAD)
|
|
*entry_point = ((grub_macho_thread_t *) hdr)->entry_point;
|
|
return 0;
|
|
}
|
|
|
|
grub_macho_addr_t
|
|
SUFFIX (grub_macho_get_entry_point) (grub_macho_t macho, const char *filename)
|
|
{
|
|
grub_macho_addr_t entry_point = 0;
|
|
grub_macho_cmds_iterate (macho, find_entry_point, &entry_point, filename);
|
|
return entry_point;
|
|
}
|