fotonmoton/ft_nm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

some refactoring

Browse source
This commit is contained in:
Gregory 2019-07-13 15:10:26 +03:00
parent de74e299cc
commit dd68ed6846
1 changed files with 160 additions and 85 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

153
src/main.c
View file

@ -1,3 +1,15 @@
/* ************************************************************************** */
/* */
/* ::: :::::::: */
/* main.c :+: :+: :+: */
/* +:+ +:+ +:+ */
/* By: gtertysh <marvin@42.fr> +#+ +:+ +#+ */
/* +#+#+#+#+#+ +#+ */
/* Created: 2019/07/13 14:52:27 by gtertysh #+# #+# */
/* Updated: 2019/07/13 14:52:29 by gtertysh ### ########.fr */
/* */
/* ************************************************************************** */
#include "libft.h"
#include <fcntl.h>
#include <sys/stat.h>
@ -5,7 +17,6 @@
#include <mach-o/loader.h>
#include <mach-o/nlist.h>
static void print_addr(void *addr)
{
size_t i;
@ -25,65 +36,81 @@ static void print_addr(void *addr)
}
}
int main(int argc, char **argv)
void open_file(const char *filename, int *fd, char **file)
{
if (argc != 2)
{
ft_putstr("usage\n");
return (1);
}
int fd = open(argv[1], O_RDONLY);
if (fd == -1)
struct stat stat_buff;
*fd = open(filename, O_RDONLY);
if (*fd == -1)
{
ft_putstr("can't open file\n");
return (1);
exit(1);
}
struct stat stat_buff;
stat(argv[1], &stat_buff);
stat(filename, &stat_buff);
if (!S_ISREG(stat_buff.st_mode))
{
ft_putstr("not a regular file\n");
return (1);
close(*fd);
exit(1);
}
char *file;
file = malloc(stat_buff.st_size);
read(fd, file, stat_buff.st_size);
uint32_t magic = *(uint32_t *)file;
if (magic != MH_MAGIC && magic != MH_MAGIC_64)
*file = NULL;
*file = malloc(stat_buff.st_size);
if (!*file)
{
ft_putstr("not a mach-o file\n");
return (1);
ft_putstr("can't allocate memory\n");
close(*fd);
exit(1);
}
read(*fd, *file, stat_buff.st_size);
}
if (magic == MH_MAGIC)
ft_putstr("32 bit mach-o file\n");
void close_file(int fd, char *file)
{
close(fd);
free(file);
}
if (magic == MH_MAGIC_64)
{
struct mach_header_64 *hdr = (struct mach_header_64 *)file;
// ft_putstr("64 bit mach-o file\n");
// ft_putstr("# of cmds: ");
// ft_putnbr(hdr->ncmds);
// ft_putstr("\n");
// ft_putstr("total cmds memory: ");
// ft_putnbr(hdr->sizeofcmds);
// ft_putstr("\n");
void handle_64(char *file)
{
struct mach_header_64 *hdr;
struct load_command *load_cmd;
struct symtab_command *symtab_cmd;
struct segment_command_64 *segment_cmd;
hdr = (struct mach_header_64 *)file;
ft_putstr("64 bit mach-o file\n");
ft_putstr("# of cmds: ");
ft_putnbr(hdr->ncmds);
ft_putstr("\n");
ft_putstr("total cmds memory: ");
ft_putnbr(hdr->sizeofcmds);
ft_putstr("\n");
uint32_t load_cmd_offset = sizeof(struct mach_header_64);
uint32_t i = 0;
while (i < hdr->ncmds)
{
load_cmd = (struct load_command *)(file + load_cmd_offset);
if (load_cmd->cmd == LC_SEGMENT_64)
{
segment_cmd = (struct segment_command_64 *)load_cmd;
ft_putnbr(i);
ft_putchar(' ');
ft_putnbr(segment_cmd->nsects);
ft_putchar(' ');
ft_putstr(segment_cmd->segname);
ft_putchar('\n');
load_cmd_offset += load_cmd->cmdsize;
i++;
continue;
}
if (load_cmd->cmd == LC_SYMTAB)
{
symtab_cmd = (struct symtab_command *)load_cmd;
// ft_putstr("found symbol table load command\n");
// ft_putstr("# of symbols: ");
// ft_putnbr(symtab_cmd->nsyms);
// ft_putstr("\n");
ft_putstr("found symbol table load command\n");
ft_putstr("# of symbols: ");
ft_putnbr(symtab_cmd->nsyms);
ft_putstr("\n");
uint32_t j = 0;
struct nlist_64 *symbol_table = (struct nlist_64 *)
@ -105,6 +132,8 @@ int main(int argc, char **argv)
continue;
}
// some shit herustic should be used
// to determine to print address or not
if (symbol.n_value)
print_addr((void *)symbol.n_value);
else
@ -115,9 +144,14 @@ int main(int argc, char **argv)
if (type == N_ABS)
ft_putchar('A' + offset);
if (type == N_SECT)
{
// lookup in which section symbol is located
ft_putchar('T' + offset);
ft_putchar(' ');
ft_putnbr(symbol.n_sect);
}
ft_putchar(' ');
ft_putstr(name);
ft_putstr("\n");
j++;
@ -126,5 +160,46 @@ int main(int argc, char **argv)
load_cmd_offset += load_cmd->cmdsize;
i++;
}
}
}
void handle_32(char *file)
{
(void)file;
ft_putstr("TODO\n");
}
void hanldle_file(const char *filename)
{
char *file;
int fd;
open_file(filename, &fd, &file);
uint32_t magic;
magic = *(uint32_t *)file;
if (magic == MH_MAGIC_64)
handle_64(file);
else if (magic == MH_MAGIC)
handle_64(file);
else if (magic == MH_CIGAM || magic == MH_CIGAM_64)
ft_putstr("do not support big endian binaries.");
else
ft_putstr("invalid magic number.");
close_file(fd, file);
}
int main(int argc, char **argv)
{
int i;
if (argc == 1)
hanldle_file("a.out");
else
{
i = 1;
while (i < argc)
hanldle_file(argv[i++]);
}
return (0);
}