Support compilation of individual arch

Makefile

@@ -19,12 +19,26 @@ INSTALL_LIBRARY ?= install -m0755
 
 LIBNAME = capstone
 
+CAPSTONE_ARCHS = arm x86 mips aarch64
+
 LIBOBJ =
 LIBOBJ += cs.o utils.o SStream.o MCInstrDesc.o MCRegisterInfo.o
-LIBOBJ += arch/Mips/MipsDisassembler.o arch/Mips/MipsInstPrinter.o arch/Mips/mapping.o
+ifneq (,$(findstring arm,$(CAPSTONE_ARCHS)))
-LIBOBJ += arch/AArch64/AArch64BaseInfo.o arch/AArch64/AArch64Disassembler.o arch/AArch64/AArch64InstPrinter.o arch/AArch64/mapping.o
+	LIBOBJ += arch/ARM/ARMDisassembler.o arch/ARM/ARMInstPrinter.o arch/ARM/mapping.o
-LIBOBJ += arch/ARM/ARMDisassembler.o arch/ARM/ARMInstPrinter.o arch/ARM/mapping.o
+	CFLAGS += -DCS_SUPPORT_ARM
-LIBOBJ += arch/X86/X86DisassemblerDecoder.o arch/X86/X86Disassembler.o arch/X86/X86IntelInstPrinter.o arch/X86/X86ATTInstPrinter.o arch/X86/mapping.o
+endif
+ifneq (,$(findstring x86,$(CAPSTONE_ARCHS)))
+	LIBOBJ += arch/X86/X86DisassemblerDecoder.o arch/X86/X86Disassembler.o arch/X86/X86IntelInstPrinter.o arch/X86/X86ATTInstPrinter.o arch/X86/mapping.o
+	CFLAGS += -DCS_SUPPORT_X86
+endif
+ifneq (,$(findstring mips,$(CAPSTONE_ARCHS)))
+	LIBOBJ += arch/Mips/MipsDisassembler.o arch/Mips/MipsInstPrinter.o arch/Mips/mapping.o
+	CFLAGS += -DCS_SUPPORT_AARCH64
+endif
+ifneq (,$(findstring aarch64,$(CAPSTONE_ARCHS)))
+	LIBOBJ += arch/AArch64/AArch64BaseInfo.o arch/AArch64/AArch64Disassembler.o arch/AArch64/AArch64InstPrinter.o arch/AArch64/mapping.o
+	CFLAGS += -DCS_SUPPORT_MIPS
+endif
 LIBOBJ += MCInst.o
 
 EXT = so

arch.h

@@ -0,0 +1,21 @@
+#ifndef __ARCH_H__
+#define __ARCH_H__
+
+#define MAX_ARCH 32
+
+void (*init_arch[MAX_ARCH]) (cs_struct *);
+
+#ifdef CS_SUPPORT_X86
+#include "arch/X86/include.h"
+#endif
+#ifdef CS_SUPPORT_ARM
+#include "arch/ARM/include.h"
+#endif
+#ifdef CS_SUPPORT_AARCH64
+#include "arch/AArch64/include.h"
+#endif
+#ifdef CS_SUPPORT_MIPS
+#include "arch/Mips/include.h"
+#endif
+
+#endif

arch/AArch64/include.h

@@ -0,0 +1,26 @@
+#ifndef __ARM64_INCLUDE_H__
+#define __ARM64_INCLUDE_H__
+
+#include "AArch64Disassembler.h"
+#include "AArch64InstPrinter.h"
+#include "mapping.h"
+
+void init_arm64(cs_struct *ud) {
+  MCRegisterInfo *mri = malloc(sizeof(*mri));
+
+  AArch64_init(mri);
+  ud->printer = AArch64_printInst;
+  ud->printer_info = mri;
+  ud->getinsn_info = mri;
+  ud->disasm = AArch64_getInstruction;
+  ud->reg_name = AArch64_reg_name;
+  ud->insn_id = AArch64_get_insn_id;
+  ud->insn_name = AArch64_insn_name;
+  ud->post_printer = AArch64_post_printer;
+}
+
+void __attribute__ ((constructor)) __init_arm64__() {
+  init_arch[CS_ARCH_ARM64] = init_arm64;
+}
+
+#endif

arch/ARM/include.h

@@ -0,0 +1,30 @@
+#ifndef __ARM_INCLUDE_H__
+#define __ARM_INCLUDE_H__
+
+#include "ARMDisassembler.h"
+#include "ARMInstPrinter.h"
+#include "mapping.h"
+
+void init_arm(cs_struct *ud) {
+  MCRegisterInfo *mri = malloc(sizeof(*mri));
+
+  ARM_init(mri);
+
+  ud->printer = ARM_printInst;
+  ud->printer_info = mri;
+  ud->reg_name = ARM_reg_name;
+  ud->insn_id = ARM_get_insn_id;
+  ud->insn_name = ARM_insn_name;
+  ud->post_printer = ARM_post_printer;
+
+  if (ud->mode & CS_MODE_THUMB)
+    ud->disasm = Thumb_getInstruction;
+  else
+    ud->disasm = ARM_getInstruction;
+}
+
+void __attribute__ ((constructor)) __init_arm__() {
+  init_arch[CS_ARCH_ARM] = init_arm;
+}
+
+#endif

arch/Mips/include.h

@@ -0,0 +1,29 @@
+#ifndef __MIPS_INCLUDE_H__
+#define __MIPS_INCLUDE_H__
+
+#include "MipsDisassembler.h"
+#include "MipsInstPrinter.h"
+#include "mapping.h"
+
+void init_mips(cs_struct *ud) {
+  MCRegisterInfo *mri = malloc(sizeof(*mri));
+
+  Mips_init(mri);
+  ud->printer = Mips_printInst;
+  ud->printer_info = mri;
+  ud->getinsn_info = mri;
+  ud->reg_name = Mips_reg_name;
+  ud->insn_id = Mips_get_insn_id;
+  ud->insn_name = Mips_insn_name;
+
+  if (ud->mode & CS_MODE_32)
+    ud->disasm = Mips_getInstruction;
+  else
+    ud->disasm = Mips64_getInstruction;
+}
+
+void __attribute__ ((constructor)) __init_mips__() {
+  init_arch[CS_ARCH_MIPS] = init_mips;
+}
+
+#endif

arch/X86/include.h

@@ -0,0 +1,23 @@
+#ifndef __X86_INCLUDE_H__
+#define __X86_INCLUDE_H__
+
+#include "X86Disassembler.h"
+#include "X86InstPrinter.h"
+#include "mapping.h"
+
+void init_x86(cs_struct *ud) {
+  // by default, we use Intel syntax
+  ud->printer = X86_Intel_printInst;
+  ud->printer_info = NULL;
+  ud->disasm = X86_getInstruction;
+  ud->reg_name = X86_reg_name;
+  ud->insn_id = X86_get_insn_id;
+  ud->insn_name = X86_insn_name;
+  ud->post_printer = X86_post_printer;
+}
+
+void __attribute__ ((constructor)) __init_x86__() {
+  init_arch[CS_ARCH_X86] = init_x86;
+}
+
+#endif

cs.c

@@ -10,25 +10,9 @@
 
 #include "MCRegisterInfo.h"
 
-#include "arch/X86/X86Disassembler.h"
+#include "arch.h"
-#include "arch/X86/X86InstPrinter.h"
-#include "arch/X86/mapping.h"
-
-#include "arch/ARM/ARMDisassembler.h"
-#include "arch/ARM/ARMInstPrinter.h"
-#include "arch/ARM/mapping.h"
-
-#include "arch/Mips/MipsDisassembler.h"
-#include "arch/Mips/MipsInstPrinter.h"
-#include "arch/Mips/mapping.h"
-
-#include "arch/AArch64/AArch64Disassembler.h"
-#include "arch/AArch64/AArch64InstPrinter.h"
-#include "arch/AArch64/mapping.h"
-
 #include "utils.h"
 
-
 void cs_version(int *major, int *minor)
 {
 	*major = CS_API_MAJOR;
@@ -62,71 +46,7 @@ cs_err cs_open(cs_arch arch, cs_mode mode, csh *handle)
   ud->reg_name = NULL;
   ud->detail = CS_OPT_ON;	// by default break instruction into details
 
-	switch (ud->arch) {
+  init_arch[ud->arch](ud);
-		case CS_ARCH_X86:
-			// by default, we use Intel syntax
-			ud->printer = X86_Intel_printInst;
-			ud->printer_info = NULL;
-			ud->disasm = X86_getInstruction;
-			ud->reg_name = X86_reg_name;
-			ud->insn_id = X86_get_insn_id;
-			ud->insn_name = X86_insn_name;
-			ud->post_printer = X86_post_printer;
-			break;
-		case CS_ARCH_ARM: {
-					MCRegisterInfo *mri = malloc(sizeof(*mri));
-
-					ARM_init(mri);
-
-					ud->printer = ARM_printInst;
-					ud->printer_info = mri;
-					ud->reg_name = ARM_reg_name;
-					ud->insn_id = ARM_get_insn_id;
-					ud->insn_name = ARM_insn_name;
-					ud->post_printer = ARM_post_printer;
-
-					if (ud->mode & CS_MODE_THUMB)
-						ud->disasm = Thumb_getInstruction;
-					else
-						ud->disasm = ARM_getInstruction;
-					break;
-				}
-		case CS_ARCH_MIPS: {
-				   MCRegisterInfo *mri = malloc(sizeof(*mri));
-
-				   Mips_init(mri);
-				   ud->printer = Mips_printInst;
-				   ud->printer_info = mri;
-				   ud->getinsn_info = mri;
-				   ud->reg_name = Mips_reg_name;
-				   ud->insn_id = Mips_get_insn_id;
-				   ud->insn_name = Mips_insn_name;
-
-				   if (ud->mode & CS_MODE_32)
-					   ud->disasm = Mips_getInstruction;
-				   else
-					   ud->disasm = Mips64_getInstruction;
-
-				   break;
-			}
-		case CS_ARCH_ARM64: {
-					MCRegisterInfo *mri = malloc(sizeof(*mri));
-
-					AArch64_init(mri);
-					ud->printer = AArch64_printInst;
-					ud->printer_info = mri;
-					ud->getinsn_info = mri;
-					ud->disasm = AArch64_getInstruction;
-					ud->reg_name = AArch64_reg_name;
-					ud->insn_id = AArch64_get_insn_id;
-					ud->insn_name = AArch64_insn_name;
-					ud->post_printer = AArch64_post_printer;
-					break;
-			}
-		default:	// unsupported architecture
-			free(ud);
-			return CS_ERR_ARCH;
-	}
 
   *handle = (uintptr_t)ud;
 
@@ -226,6 +146,7 @@ cs_err cs_option(csh ud, cs_opt_type type, size_t value)
           handle->errnum = CS_ERR_OPTION;
           return CS_ERR_OPTION;
 
+#ifdef CS_SUPPORT_X86
         case CS_ARCH_X86:
           switch(value) {
             default:
@@ -242,6 +163,7 @@ cs_err cs_option(csh ud, cs_opt_type type, size_t value)
               break;
           }
           break;
+#endif
       }
       break;
 
@@ -251,6 +173,7 @@ cs_err cs_option(csh ud, cs_opt_type type, size_t value)
         default:
           // only selected archs care about CS_OPT_SYNTAX
           break;
+#ifdef CS_SUPPORT_ARM
         case CS_ARCH_ARM:
           if (value & CS_MODE_THUMB)
             handle->disasm = Thumb_getInstruction;
@@ -259,6 +182,8 @@ cs_err cs_option(csh ud, cs_opt_type type, size_t value)
 
           handle->mode = value;
           break;
+#endif
+#ifdef CS_SUPPORT_AARCH64
         case CS_ARCH_MIPS:
           if (value & CS_MODE_32)
             handle->disasm = Mips_getInstruction;
@@ -267,6 +192,7 @@ cs_err cs_option(csh ud, cs_opt_type type, size_t value)
 
           handle->mode = value;
           break;
+#endif
       }
       break;
   }