int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size) {
ks_engine *ks;
ks_err err;
size_t count;
unsigned char *encode = NULL;
size_t size;
char * assembler;
if (outfile == NULL) {
// we compute the output
outfile = fopen("/dev/null", "w");
if (outfile == NULL) {
printf("failed opening /dev/null\n");
abort();
}
}
if (Size < 1) {
return 0;
}
err = ks_open(KS_ARCH_PPC, KS_MODE_PPC32+KS_MODE_BIG_ENDIAN, &ks);
if (err != KS_ERR_OK) {
printf("ERROR: failed on ks_open(), quit error = %u\n", err);
abort();
}
ks_option(ks, KS_OPT_SYNTAX, Data[Size-1]);
assembler = malloc(Size);
memcpy(assembler, Data, Size-1);
//null terminate string
assembler[Size-1] = 0;
if (ks_asm(ks, assembler, 0, &encode, &size, &count) != KS_ERR_OK) {
fprintf(outfile, "ERROR: ks_asm() failed & count = %lu, error = %u\n",
count, ks_errno(ks));
} else {
size_t i;
fprintf(outfile, "%s = ", assembler);
for (i = 0; i < size; i++) {
fprintf(outfile, "%02x ", encode[i]);
}
fprintf(outfile, "\n");
fprintf(outfile, "Compiled: %lu bytes, statements: %lu\n", size, count);
}
free(assembler);
// NOTE: free encode after usage to avoid leaking memory
if (encode != NULL) {
ks_free(encode);
}
// close Keystone instance when done
ks_close(ks);
return 0;
}
static bool unmap_file(struct ko_process *where, void *address_in_module)
{
struct ko_section *ks;
ks = ks_get_by_vaddress(where, (unsigned long)address_in_module);
if (!ks)
return false;
ks_close(where, ks);
return true;
}
int main(int argc, char **argv) {
int ks_arch = KS_ARCH_X86, ks_mode = KS_MODE_64;
char *assembly = ".=x;.=x;1:";
ks_engine *ks;
ks_err err = ks_open(ks_arch, ks_mode, &ks);
if (!err) {
size_t count, size;
unsigned char *insn;
ks_asm(ks, assembly, 0, &insn, &size, &count);
ks_free(insn);
}
ks_close(ks);
}
int main(int argc, char **argv) {
int ks_arch = KS_ARCH_X86, ks_mode = KS_MODE_64;
char *assembly = "{[x*mm0";
ks_engine *ks;
ks_err err = ks_open(ks_arch, ks_mode, &ks);
if (!err) {
size_t count, size;
unsigned char *insn;
if (ks_asm(ks, assembly, 0, &insn, &size, &count))
printf("ERROR: failed on ks_asm() with error = %s, code = %u\n", ks_strerror(ks_errno(ks)), ks_errno(ks));
ks_free(insn);
}
ks_close(ks);
}
int main(int argc, char **argv) {
int ks_arch = KS_ARCH_ARM, ks_mode = KS_MODE_LITTLE_ENDIAN;
char *assembly = "LDr AH1I,=jHGQ,=2,=r,=GQ,=1111111111111112,=ey,=2,=eQ,=rQ',=r,=";
ks_engine *ks;
ks_err err = ks_open(ks_arch, ks_mode, &ks);
if (!err) {
size_t count, size;
unsigned char *insn;
if (ks_asm(ks, assembly, 0, &insn, &size, &count))
printf("ERROR: failed on ks_asm() with error = %s, code = %u\n", ks_strerror(ks_errno(ks)), ks_errno(ks));
ks_free(insn);
}
ks_close(ks);
return 0;
}
int main(int argc, char **argv) {
int ks_arch = KS_ARCH_SYSTEMZ, ks_mode = KS_MODE_LITTLE_ENDIAN;
unsigned char assembly[] = {
'J', '-', '9', 'e', 'e', 0x00,
};
ks_engine *ks;
ks_err err = ks_open(ks_arch, ks_mode, &ks);
if (!err) {
size_t count, size;
unsigned char *insn;
if (ks_asm(ks, (char *)assembly, 0, &insn, &size, &count))
printf("ERROR: failed on ks_asm() with error = %s, code = %u\n", ks_strerror(ks_errno(ks)), ks_errno(ks));
ks_free(insn);
}
ks_close(ks);
return 0;
}
int main(int argc, char **argv) {
int ks_arch = KS_ARCH_X86, ks_mode = KS_MODE_64;
unsigned char assembly[] = {
0x12, ';', 0x12, 'i', ';', '.', 'i', 'R', 'p', ' ',
'e', 'm', 'i', 'R', 'p', ',', 0xca, 0xe9, ',', 'I',
'=', 0x00,
};
ks_engine *ks;
ks_err err = ks_open(ks_arch, ks_mode, &ks);
if (!err) {
size_t count, size;
unsigned char *insn;
if (ks_asm(ks, (char *)assembly, 0, &insn, &size, &count))
printf("ERROR: failed on ks_asm() with error = %s, code = %u\n", ks_strerror(ks_errno(ks)), ks_errno(ks));
ks_free(insn);
}
ks_close(ks);
return 0;
}
static int test_ks(ks_arch arch, int mode, const char *assembly, int syntax)
{
ks_engine *ks;
ks_err err = KS_ERR_ARCH;
size_t count;
unsigned char *encode;
size_t size;
err = ks_open(arch, mode, &ks);
if (err != KS_ERR_OK) {
printf("ERROR: failed on ks_open(), quit\n");
return -1;
}
if (syntax)
ks_option(ks, KS_OPT_SYNTAX, syntax);
if (ks_asm(ks, assembly, 0, &encode, &size, &count)) {
printf("ERROR: failed on ks_asm() with count = %lu, error code = %u\n", count, ks_errno(ks));
} else {
size_t i;
printf("%s = ", assembly);
for (i = 0; i < size; i++) {
printf("%02x ", encode[i]);
}
printf("\n");
printf("Assembled: %lu bytes, %lu statements\n\n", size, count);
}
// NOTE: free encode after usage to avoid leaking memory
ks_free(encode);
// close Keystone instance when done
ks_close(ks);
return 0;
}
static int keystone_assemble(RAsm *a, RAsmOp *ao, const char *str, ks_arch arch, ks_mode mode) {
ks_err err = KS_ERR_ARCH;
bool must_init = false;
size_t count, size;
ut8 *insn = NULL;
if (!ks_arch_supported (arch)) {
return -1;
}
must_init = true; //!oldcur || (a->cur != oldcur || oldbit != a->bits);
oldcur = a->cur;
oldbit = a->bits;
if (must_init) {
if (ks) {
ks_close (ks);
ks = NULL;
}
err = ks_open (arch, mode, &ks);
if (err || !ks) {
eprintf ("Cannot initialize keystone\n");
ks_free (insn);
if (ks) {
ks_close (ks);
ks = NULL;
}
return -1;
}
}
if (!ks) {
ks_free (insn);
if (ks) {
ks_close (ks);
ks = NULL;
}
return -1;
}
if (a->syntax == R_ASM_SYNTAX_ATT) {
ks_option (ks, KS_OPT_SYNTAX, KS_OPT_SYNTAX_ATT);
} else {
ks_option (ks, KS_OPT_SYNTAX, KS_OPT_SYNTAX_NASM);
}
int rc = ks_asm (ks, str, a->pc, &insn, &size, &count);
if (rc) {
eprintf ("ks_asm: %s\n", ks_strerror ((ks_err)ks_errno (ks)));
ks_free (insn);
if (ks) {
ks_close (ks);
ks = NULL;
}
return -1;
}
memcpy (ao->buf, insn, R_MIN (size, sizeof (ao->buf) -1));
ks_free (insn);
if (ks) {
ks_close (ks);
ks = NULL;
}
return size;
}
int main(int argc, char **argv)
{
ks_engine *ks;
ks_err err = KS_ERR_ARCH;
char *mode, *assembly = NULL;
uint64_t start_addr = 0;
char *input = NULL;
size_t count;
unsigned char *insn = NULL;
size_t size;
if (argc == 2) {
// handle code from stdin
#if !defined(WIN32) && !defined(WIN64) && !defined(_WIN32) && !defined(_WIN64)
int flags;
size_t index = 0;
char buf[1024];
mode = argv[1];
if ((flags = fcntl(STDIN_FILENO, F_GETFL, 0)) == -1)
flags = 0;
fcntl(STDIN_FILENO, F_SETFL, flags | O_NONBLOCK);
while(fgets(buf, sizeof(buf), stdin)) {
input = (char*)realloc(input, index + strlen(buf));
if (!input) {
printf("Failed to allocate memory.");
return 1;
}
memcpy(&input[index], buf, strlen(buf));
index += strlen(buf);
}
fcntl(STDIN_FILENO, F_SETFL, flags);
assembly = input;
if (!assembly) {
usage(argv[0]);
return -1;
}
#else // Windows does not handle code from stdin
usage(argv[0]);
return -1;
#endif
} else if (argc == 3) {
// kstool <arch> <assembly>
mode = argv[1];
assembly = argv[2];
} else if (argc == 4) {
// kstool <arch> <assembly> <address>
char *temp;
mode = argv[1];
assembly = argv[2];
start_addr = strtoull(argv[3], &temp, 16);
if (temp == argv[3] || *temp != '\0' || errno == ERANGE) {
printf("ERROR: invalid address argument, quit!\n");
return -2;
}
} else {
usage(argv[0]);
return -1;
}
if (!strcmp(mode, "x16")) {
err = ks_open(KS_ARCH_X86, KS_MODE_16, &ks);
}
if (!strcmp(mode, "x32")) {
err = ks_open(KS_ARCH_X86, KS_MODE_32, &ks);
}
if (!strcmp(mode, "x64")) {
err = ks_open(KS_ARCH_X86, KS_MODE_64, &ks);
}
if (!strcmp(mode, "x16att")) {
err = ks_open(KS_ARCH_X86, KS_MODE_16, &ks);
if (!err) {
ks_option(ks, KS_OPT_SYNTAX, KS_OPT_SYNTAX_ATT);
}
}
if (!strcmp(mode, "x32att")) {
err = ks_open(KS_ARCH_X86, KS_MODE_32, &ks);
if (!err) {
ks_option(ks, KS_OPT_SYNTAX, KS_OPT_SYNTAX_ATT);
}
}
if (!strcmp(mode, "x64att")) {
err = ks_open(KS_ARCH_X86, KS_MODE_64, &ks);
if (!err) {
ks_option(ks, KS_OPT_SYNTAX, KS_OPT_SYNTAX_ATT);
}
}
if (!strcmp(mode, "x16nasm")) {
err = ks_open(KS_ARCH_X86, KS_MODE_16, &ks);
if (!err) {
ks_option(ks, KS_OPT_SYNTAX, KS_OPT_SYNTAX_NASM);
}
}
if (!strcmp(mode, "x32nasm")) {
err = ks_open(KS_ARCH_X86, KS_MODE_32, &ks);
if (!err) {
ks_option(ks, KS_OPT_SYNTAX, KS_OPT_SYNTAX_NASM);
}
}
if (!strcmp(mode, "x64nasm")) {
err = ks_open(KS_ARCH_X86, KS_MODE_64, &ks);
if (!err) {
ks_option(ks, KS_OPT_SYNTAX, KS_OPT_SYNTAX_NASM);
}
}
if (!strcmp(mode, "arm")) {
err = ks_open(KS_ARCH_ARM, KS_MODE_ARM+KS_MODE_LITTLE_ENDIAN, &ks);
}
if (!strcmp(mode, "armbe")) {
err = ks_open(KS_ARCH_ARM, KS_MODE_ARM+KS_MODE_BIG_ENDIAN, &ks);
}
if (!strcmp(mode, "thumb")) {
err = ks_open(KS_ARCH_ARM, KS_MODE_THUMB+KS_MODE_LITTLE_ENDIAN, &ks);
}
if (!strcmp(mode, "thumbbe")) {
err = ks_open(KS_ARCH_ARM, KS_MODE_THUMB+KS_MODE_BIG_ENDIAN, &ks);
}
if (!strcmp(mode, "armv8")) {
err = ks_open(KS_ARCH_ARM, KS_MODE_ARM+KS_MODE_LITTLE_ENDIAN+KS_MODE_V8, &ks);
}
if (!strcmp(mode, "armv8be")) {
err = ks_open(KS_ARCH_ARM, KS_MODE_ARM+KS_MODE_BIG_ENDIAN+KS_MODE_V8, &ks);
}
if (!strcmp(mode, "thumbv8")) {
err = ks_open(KS_ARCH_ARM, KS_MODE_THUMB+KS_MODE_LITTLE_ENDIAN+KS_MODE_V8, &ks);
}
if (!strcmp(mode, "thumbv8be")) {
err = ks_open(KS_ARCH_ARM, KS_MODE_THUMB+KS_MODE_BIG_ENDIAN+KS_MODE_V8, &ks);
}
if (!strcmp(mode, "arm64")) {
err = ks_open(KS_ARCH_ARM64, KS_MODE_LITTLE_ENDIAN, &ks);
}
if (!strcmp(mode, "hex") || !strcmp(mode, "hexagon")) {
err = ks_open(KS_ARCH_HEXAGON, KS_MODE_BIG_ENDIAN, &ks);
}
if (!strcmp(mode, "mips")) {
err = ks_open(KS_ARCH_MIPS, KS_MODE_MIPS32+KS_MODE_LITTLE_ENDIAN, &ks);
}
if (!strcmp(mode, "mipsbe")) {
err = ks_open(KS_ARCH_MIPS, KS_MODE_MIPS32+KS_MODE_BIG_ENDIAN, &ks);
}
if (!strcmp(mode, "mips64")) {
err = ks_open(KS_ARCH_MIPS, KS_MODE_MIPS64+KS_MODE_LITTLE_ENDIAN, &ks);
}
if (!strcmp(mode, "mips64be")) {
err = ks_open(KS_ARCH_MIPS, KS_MODE_MIPS64+KS_MODE_BIG_ENDIAN, &ks);
}
if (!strcmp(mode, "ppc32be")) {
err = ks_open(KS_ARCH_PPC, KS_MODE_PPC32+KS_MODE_BIG_ENDIAN, &ks);
}
if (!strcmp(mode, "ppc64")) {
err = ks_open(KS_ARCH_PPC, KS_MODE_PPC64+KS_MODE_LITTLE_ENDIAN, &ks);
}
if (!strcmp(mode, "ppc64be")) {
err = ks_open(KS_ARCH_PPC, KS_MODE_PPC64+KS_MODE_BIG_ENDIAN, &ks);
}
if (!strcmp(mode, "sparc")) {
err = ks_open(KS_ARCH_SPARC, KS_MODE_SPARC32+KS_MODE_LITTLE_ENDIAN, &ks);
}
if (!strcmp(mode, "sparcbe")) {
err = ks_open(KS_ARCH_SPARC, KS_MODE_SPARC32+KS_MODE_BIG_ENDIAN, &ks);
}
if (!strcmp(mode, "sparc64be")) {
err = ks_open(KS_ARCH_SPARC, KS_MODE_SPARC64+KS_MODE_BIG_ENDIAN, &ks);
}
if (!strcmp(mode, "systemz") || !strcmp(mode, "sysz") || !strcmp(mode, "s390x")) {
err = ks_open(KS_ARCH_SYSTEMZ, KS_MODE_BIG_ENDIAN, &ks);
}
if (!strcmp(mode, "evm")) {
err = ks_open(KS_ARCH_EVM, 0, &ks);
}
if (err) {
printf("ERROR: failed on ks_open()\n");
usage(argv[0]);
return -1;
}
if (ks_asm(ks, assembly, start_addr, &insn, &size, &count)) {
printf("ERROR: failed on ks_asm() with count = %zu, error = '%s' (code = %u)\n", count, ks_strerror(ks_errno(ks)), ks_errno(ks));
} else {
size_t i;
printf("%s = [ ", assembly);
for (i = 0; i < size; i++) {
printf("%02x ", insn[i]);
}
printf("]\n");
//printf("Assembled: %lu bytes, %lu statement(s)\n", size, count);
}
// NOTE: free insn after usage to avoid leaking memory
if (insn != NULL) {
ks_free(insn);
}
// close Keystone instance when done
ks_close(ks);
free(input);
return 0;
}
