如何优化Mach-O 64位二进制解析器的代码组织？

针对你开发Mach-O 64位解析器时遇到的问题，最简洁高效的方案是使用静态查找表，将每个加载命令的数值与对应的可读名称关联起来，无需编写大量if-else分支。

具体实现步骤： #

1. 定义查找表结构

首先创建一个结构体，用于存储加载命令的数值和对应的名称：

#include <mach-o/loader.h>

typedef struct {
    uint32_t cmd;
    const char* name;
} LoadCommandName;

2. 填充加载命令映射表

把所有需要识别的Mach-O加载命令（比如LC_SEGMENT_64、LC_SYMTAB等）加入这个数组：

static const LoadCommandName load_command_names[] = {
    { LC_SEGMENT_64, "LC_SEGMENT_64" },
    { LC_SYMTAB, "LC_SYMTAB" },
    { LC_DYSYMTAB, "LC_DYSYMTAB" },
    { LC_LOAD_DYLIB, "LC_LOAD_DYLIB" },
    { LC_UUID, "LC_UUID" },
    { LC_VERSION_MIN_MACOSX, "LC_VERSION_MIN_MACOSX" },
    { LC_MAIN, "LC_MAIN" },
    // 可继续添加其他需要识别的加载命令
    { 0, NULL } // 结束标记
};

3. 编写查找函数

实现一个辅助函数，根据命令数值查找对应的可读名称：

const char* get_load_command_name(uint32_t cmd) {
    for (int i = 0; load_command_names[i].name != NULL; i++) {
        if (load_command_names[i].cmd == cmd) {
            return load_command_names[i].name;
        }
    }
    return "Unknown Load Command";
}

4. 替换原代码中的if-else分支

在你的HexPrinter函数中，把原来的if-else块替换为调用这个查找函数：

// 替换之前的if-else部分
const char* cmd_name = get_load_command_name(buffer);
printf("\t%s\n", cmd_name);

优化后的完整代码示例 #

整合以上修改并修复原代码中的若干bug（字节移位错误、文件打开模式、地址递增时机等）后的代码如下：

#include <errno.h>
#include <mach-o/loader.h>
#include <mach/machine.h>
#include <stdint.h>
#include <stdio.h>
#include <string.h>

#define BUFFER_SIZE 4
#define ERROR(msg) fprintf(stderr, "ERROR: %s | %s\n", msg, strerror(errno));

typedef struct {
    uint32_t cmd;
    const char* name;
} LoadCommandName;

static const LoadCommandName load_command_names[] = {
    { LC_SEGMENT_64, "LC_SEGMENT_64" },
    { LC_SYMTAB, "LC_SYMTAB" },
    { LC_DYSYMTAB, "LC_DYSYMTAB" },
    { LC_LOAD_DYLIB, "LC_LOAD_DYLIB" },
    { LC_UUID, "LC_UUID" },
    { LC_VERSION_MIN_MACOSX, "LC_VERSION_MIN_MACOSX" },
    { LC_MAIN, "LC_MAIN" },
    { 0, NULL }
};

const char* get_load_command_name(uint32_t cmd) {
    for (int i = 0; load_command_names[i].name != NULL; i++) {
        if (load_command_names[i].cmd == cmd) {
            return load_command_names[i].name;
        }
    }
    return "Unknown Load Command";
}

void HexPrinter(uint32_t buffer, FILE *binary) {
    uint64_t mem_addr = 0x10000000;
    fread(&buffer, 1, BUFFER_SIZE, binary);
    if (buffer != MH_MAGIC_64) {
        ERROR("Not a 64-bit Mach-O binary");
        return;
    } else {
        printf("0x%llx: %02x %02x %02x %02x\n", mem_addr, 
               (buffer & 0xFF), ((buffer >> 8) & 0xFF), 
               ((buffer >> 16) & 0xFF), ((buffer >> 24) & 0xFF));
    }

    mem_addr += BUFFER_SIZE; // 修正：第一次读取后先递增地址

    while ((fread(&buffer, 1, BUFFER_SIZE, binary)) == BUFFER_SIZE) {
        printf("0x%llx: %02x %02x %02x %02x\t", mem_addr, 
               (buffer & 0xFF), ((buffer >> 8) & 0xFF),
               ((buffer >> 16) & 0xFF), ((buffer >> 24) & 0xFF));

        // 使用查找函数获取命令名称
        const char* cmd_name = get_load_command_name(buffer);
        printf("%s\n", cmd_name);

        mem_addr += BUFFER_SIZE;
    }

    if (ferror(binary)) {
        ERROR("Error reading file");
        fclose(binary);
        return;
    }
}

int main(int argc, char *argv[]) { // 修正参数声明错误
    FILE *binary;
    char *pathname = argv[1];

    uint32_t buffer;

    if (!argv[1]) {
        ERROR("Usage: ./nibBrev <pathname>");
        return (-1);
    }

    binary = fopen(pathname, "rb"); // 修正：以二进制模式打开文件
    if (!binary) {
        ERROR("Couldn't open file");
        return (-1);
    }

    HexPrinter(buffer, binary);

    fclose(binary);
    return 0;
}

额外注意事项 #

• 字节顺序修正：原代码中((buffer >> 9) & 0xFF)是错误的，每个字节占8位，正确移位应为>>8。
• 二进制模式打开文件：使用"rb"而非"r"打开，避免Windows系统下的换行符转换问题。
• 地址递增时机：第一次读取魔数后需先递增地址，否则会出现地址重复。
• 字节序兼容：若需支持不同字节序的Mach-O文件，需使用swap32等函数转换字节序后再进行查找。

内容的提问来源于stack exchange，提问作者fig