* grub-core/loader/machoXX.c: Remove nested functions.

This commit is contained in:
Vladimir 'phcoder' Serbinenko 2013-02-27 20:56:23 +01:00
parent 4eb8b75659
commit 43c918828d
2 changed files with 148 additions and 117 deletions

View file

@ -1,3 +1,7 @@
2013-02-27 Vladimir Serbinenko <phcoder@gmail.com>
* grub-core/loader/machoXX.c: Remove nested functions.
2013-02-27 Colin Watson <cjwatson@ubuntu.com> 2013-02-27 Colin Watson <cjwatson@ubuntu.com>
Remove nested functions from disk and file read hooks. Remove nested functions from disk and file read hooks.

View file

@ -79,7 +79,7 @@ SUFFIX (grub_macho_parse) (grub_macho_t macho, const char *filename)
} }
} }
typedef int NESTED_FUNC_ATTR (*grub_macho_iter_hook_t) typedef int (*grub_macho_iter_hook_t)
(grub_macho_t , struct grub_macho_cmd *, (grub_macho_t , struct grub_macho_cmd *,
void *); void *);
@ -200,46 +200,59 @@ SUFFIX (grub_macho_readfile) (grub_macho_t macho,
return GRUB_ERR_NONE; 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. */ /* Calculate the amount of memory spanned by the segments. */
grub_err_t grub_err_t
SUFFIX (grub_macho_size) (grub_macho_t macho, grub_macho_addr_t *segments_start, SUFFIX (grub_macho_size) (grub_macho_t macho, grub_macho_addr_t *segments_start,
grub_macho_addr_t *segments_end, int flags, grub_macho_addr_t *segments_end, int flags,
const char *filename) const char *filename)
{ {
int nr_phdrs = 0; struct calcsize_ctx ctx = {
.flags = flags,
/* Run through the program headers to calculate the total memory size we .nr_phdrs = 0,
should claim. */ .segments_start = segments_start,
auto int NESTED_FUNC_ATTR calcsize (grub_macho_t _macho, .segments_end = segments_end,
struct grub_macho_cmd *phdr, void *_arg); };
int NESTED_FUNC_ATTR calcsize (grub_macho_t _macho __attribute__ ((unused)),
struct grub_macho_cmd *hdr0,
void *_arg __attribute__ ((unused)))
{
grub_macho_segment_t *hdr = (grub_macho_segment_t *) hdr0;
if (hdr->cmd != GRUB_MACHO_CMD_SEGMENT)
return 0;
if (! hdr->vmsize)
return 0;
if (! hdr->filesize && (flags & GRUB_MACHO_NOBSS))
return 0;
nr_phdrs++;
if (hdr->vmaddr < *segments_start)
*segments_start = hdr->vmaddr;
if (hdr->vmaddr + hdr->vmsize > *segments_end)
*segments_end = hdr->vmaddr + hdr->vmsize;
return 0;
}
*segments_start = (grub_macho_addr_t) -1; *segments_start = (grub_macho_addr_t) -1;
*segments_end = 0; *segments_end = 0;
grub_macho_cmds_iterate (macho, calcsize, 0, filename); grub_macho_cmds_iterate (macho, calcsize, &ctx, filename);
if (nr_phdrs == 0) if (ctx.nr_phdrs == 0)
return grub_error (GRUB_ERR_BAD_OS, "no program headers present"); return grub_error (GRUB_ERR_BAD_OS, "no program headers present");
if (*segments_end < *segments_start) if (*segments_end < *segments_start)
@ -249,109 +262,123 @@ SUFFIX (grub_macho_size) (grub_macho_t macho, grub_macho_addr_t *segments_start,
return GRUB_ERR_NONE; 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'. */ /* Load every loadable segment into memory specified by `_load_hook'. */
grub_err_t grub_err_t
SUFFIX (grub_macho_load) (grub_macho_t macho, const char *filename, SUFFIX (grub_macho_load) (grub_macho_t macho, const char *filename,
char *offset, int flags, int *darwin_version) char *offset, int flags, int *darwin_version)
{ {
auto int NESTED_FUNC_ATTR do_load(grub_macho_t _macho, struct do_load_ctx ctx = {
struct grub_macho_cmd *hdr0, .flags = flags,
void *_arg __attribute__ ((unused))); .offset = offset,
int NESTED_FUNC_ATTR do_load(grub_macho_t _macho, .filename = filename,
struct grub_macho_cmd *hdr0, .darwin_version = darwin_version
void *_arg __attribute__ ((unused))) };
{
grub_macho_segment_t *hdr = (grub_macho_segment_t *) hdr0;
if (hdr->cmd != GRUB_MACHO_CMD_SEGMENT)
return 0;
if (! hdr->filesize && (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 (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, 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"),
filename);
return 1;
}
if (darwin_version)
{
const char *ptr = 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;
*darwin_version = 0;
end += (sizeof ("Darwin Kernel Version ") - 1);
while (ptr < end && grub_isdigit (*ptr))
*darwin_version = (*ptr++ - '0') + *darwin_version * 10;
break;
}
}
}
if (hdr->filesize < hdr->vmsize)
grub_memset (offset + hdr->vmaddr + hdr->filesize,
0, hdr->vmsize - hdr->filesize);
return 0;
}
if (darwin_version) if (darwin_version)
*darwin_version = 0; *darwin_version = 0;
grub_macho_cmds_iterate (macho, do_load, 0, filename); grub_macho_cmds_iterate (macho, do_load, &ctx, filename);
return grub_errno; 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 grub_macho_addr_t
SUFFIX (grub_macho_get_entry_point) (grub_macho_t macho, const char *filename) SUFFIX (grub_macho_get_entry_point) (grub_macho_t macho, const char *filename)
{ {
grub_macho_addr_t entry_point = 0; grub_macho_addr_t entry_point = 0;
auto int NESTED_FUNC_ATTR hook(grub_macho_t _macho, grub_macho_cmds_iterate (macho, find_entry_point, &entry_point, filename);
struct grub_macho_cmd *hdr,
void *_arg __attribute__ ((unused)));
int NESTED_FUNC_ATTR hook(grub_macho_t _macho __attribute__ ((unused)),
struct grub_macho_cmd *hdr,
void *_arg __attribute__ ((unused)))
{
if (hdr->cmd == GRUB_MACHO_CMD_THREAD)
entry_point = ((grub_macho_thread_t *) hdr)->entry_point;
return 0;
}
grub_macho_cmds_iterate (macho, hook, 0, filename);
return entry_point; return entry_point;
} }