162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0
262306a36Sopenharmony_ci#include <linux/types.h>
362306a36Sopenharmony_ci#include <string.h>
462306a36Sopenharmony_ci
562306a36Sopenharmony_ci#include "debug.h"
662306a36Sopenharmony_ci#include "tests/tests.h"
762306a36Sopenharmony_ci#include "arch-tests.h"
862306a36Sopenharmony_ci#include "../../../../arch/x86/include/asm/insn.h"
962306a36Sopenharmony_ci
1062306a36Sopenharmony_ci#include "intel-pt-decoder/intel-pt-insn-decoder.h"
1162306a36Sopenharmony_ci
1262306a36Sopenharmony_cistruct test_data {
1362306a36Sopenharmony_ci	u8 data[MAX_INSN_SIZE];
1462306a36Sopenharmony_ci	int expected_length;
1562306a36Sopenharmony_ci	int expected_rel;
1662306a36Sopenharmony_ci	const char *expected_op_str;
1762306a36Sopenharmony_ci	const char *expected_branch_str;
1862306a36Sopenharmony_ci	const char *asm_rep;
1962306a36Sopenharmony_ci};
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_ciconst struct test_data test_data_32[] = {
2262306a36Sopenharmony_ci#include "insn-x86-dat-32.c"
2362306a36Sopenharmony_ci	{{0x0f, 0x01, 0xee}, 3, 0, NULL, NULL, "0f 01 ee             \trdpkru"},
2462306a36Sopenharmony_ci	{{0x0f, 0x01, 0xef}, 3, 0, NULL, NULL, "0f 01 ef             \twrpkru"},
2562306a36Sopenharmony_ci	{{0}, 0, 0, NULL, NULL, NULL},
2662306a36Sopenharmony_ci};
2762306a36Sopenharmony_ci
2862306a36Sopenharmony_ciconst struct test_data test_data_64[] = {
2962306a36Sopenharmony_ci#include "insn-x86-dat-64.c"
3062306a36Sopenharmony_ci	{{0x0f, 0x01, 0xee}, 3, 0, NULL, NULL, "0f 01 ee             \trdpkru"},
3162306a36Sopenharmony_ci	{{0x0f, 0x01, 0xef}, 3, 0, NULL, NULL, "0f 01 ef             \twrpkru"},
3262306a36Sopenharmony_ci	{{0xf2, 0x0f, 0x01, 0xca}, 4, 0, "erets", "indirect", "f2 0f 01 ca  \terets"},
3362306a36Sopenharmony_ci	{{0xf3, 0x0f, 0x01, 0xca}, 4, 0, "eretu", "indirect", "f3 0f 01 ca  \teretu"},
3462306a36Sopenharmony_ci	{{0}, 0, 0, NULL, NULL, NULL},
3562306a36Sopenharmony_ci};
3662306a36Sopenharmony_ci
3762306a36Sopenharmony_cistatic int get_op(const char *op_str)
3862306a36Sopenharmony_ci{
3962306a36Sopenharmony_ci	struct val_data {
4062306a36Sopenharmony_ci		const char *name;
4162306a36Sopenharmony_ci		int val;
4262306a36Sopenharmony_ci	} vals[] = {
4362306a36Sopenharmony_ci		{"other",   INTEL_PT_OP_OTHER},
4462306a36Sopenharmony_ci		{"call",    INTEL_PT_OP_CALL},
4562306a36Sopenharmony_ci		{"ret",     INTEL_PT_OP_RET},
4662306a36Sopenharmony_ci		{"jcc",     INTEL_PT_OP_JCC},
4762306a36Sopenharmony_ci		{"jmp",     INTEL_PT_OP_JMP},
4862306a36Sopenharmony_ci		{"loop",    INTEL_PT_OP_LOOP},
4962306a36Sopenharmony_ci		{"iret",    INTEL_PT_OP_IRET},
5062306a36Sopenharmony_ci		{"int",     INTEL_PT_OP_INT},
5162306a36Sopenharmony_ci		{"syscall", INTEL_PT_OP_SYSCALL},
5262306a36Sopenharmony_ci		{"sysret",  INTEL_PT_OP_SYSRET},
5362306a36Sopenharmony_ci		{"vmentry",  INTEL_PT_OP_VMENTRY},
5462306a36Sopenharmony_ci		{"erets",   INTEL_PT_OP_ERETS},
5562306a36Sopenharmony_ci		{"eretu",   INTEL_PT_OP_ERETU},
5662306a36Sopenharmony_ci		{NULL, 0},
5762306a36Sopenharmony_ci	};
5862306a36Sopenharmony_ci	struct val_data *val;
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_ci	if (!op_str || !strlen(op_str))
6162306a36Sopenharmony_ci		return 0;
6262306a36Sopenharmony_ci
6362306a36Sopenharmony_ci	for (val = vals; val->name; val++) {
6462306a36Sopenharmony_ci		if (!strcmp(val->name, op_str))
6562306a36Sopenharmony_ci			return val->val;
6662306a36Sopenharmony_ci	}
6762306a36Sopenharmony_ci
6862306a36Sopenharmony_ci	pr_debug("Failed to get op\n");
6962306a36Sopenharmony_ci
7062306a36Sopenharmony_ci	return -1;
7162306a36Sopenharmony_ci}
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_cistatic int get_branch(const char *branch_str)
7462306a36Sopenharmony_ci{
7562306a36Sopenharmony_ci	struct val_data {
7662306a36Sopenharmony_ci		const char *name;
7762306a36Sopenharmony_ci		int val;
7862306a36Sopenharmony_ci	} vals[] = {
7962306a36Sopenharmony_ci		{"no_branch",     INTEL_PT_BR_NO_BRANCH},
8062306a36Sopenharmony_ci		{"indirect",      INTEL_PT_BR_INDIRECT},
8162306a36Sopenharmony_ci		{"conditional",   INTEL_PT_BR_CONDITIONAL},
8262306a36Sopenharmony_ci		{"unconditional", INTEL_PT_BR_UNCONDITIONAL},
8362306a36Sopenharmony_ci		{NULL, 0},
8462306a36Sopenharmony_ci	};
8562306a36Sopenharmony_ci	struct val_data *val;
8662306a36Sopenharmony_ci
8762306a36Sopenharmony_ci	if (!branch_str || !strlen(branch_str))
8862306a36Sopenharmony_ci		return 0;
8962306a36Sopenharmony_ci
9062306a36Sopenharmony_ci	for (val = vals; val->name; val++) {
9162306a36Sopenharmony_ci		if (!strcmp(val->name, branch_str))
9262306a36Sopenharmony_ci			return val->val;
9362306a36Sopenharmony_ci	}
9462306a36Sopenharmony_ci
9562306a36Sopenharmony_ci	pr_debug("Failed to get branch\n");
9662306a36Sopenharmony_ci
9762306a36Sopenharmony_ci	return -1;
9862306a36Sopenharmony_ci}
9962306a36Sopenharmony_ci
10062306a36Sopenharmony_cistatic int test_data_item(const struct test_data *dat, int x86_64)
10162306a36Sopenharmony_ci{
10262306a36Sopenharmony_ci	struct intel_pt_insn intel_pt_insn;
10362306a36Sopenharmony_ci	int op, branch, ret;
10462306a36Sopenharmony_ci	struct insn insn;
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ci	ret = insn_decode(&insn, dat->data, MAX_INSN_SIZE,
10762306a36Sopenharmony_ci			  x86_64 ? INSN_MODE_64 : INSN_MODE_32);
10862306a36Sopenharmony_ci	if (ret < 0) {
10962306a36Sopenharmony_ci		pr_debug("Failed to decode: %s\n", dat->asm_rep);
11062306a36Sopenharmony_ci		return -1;
11162306a36Sopenharmony_ci	}
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_ci	if (insn.length != dat->expected_length) {
11462306a36Sopenharmony_ci		pr_debug("Failed to decode length (%d vs expected %d): %s\n",
11562306a36Sopenharmony_ci			 insn.length, dat->expected_length, dat->asm_rep);
11662306a36Sopenharmony_ci		return -1;
11762306a36Sopenharmony_ci	}
11862306a36Sopenharmony_ci
11962306a36Sopenharmony_ci	op = get_op(dat->expected_op_str);
12062306a36Sopenharmony_ci	branch = get_branch(dat->expected_branch_str);
12162306a36Sopenharmony_ci
12262306a36Sopenharmony_ci	if (intel_pt_get_insn(dat->data, MAX_INSN_SIZE, x86_64, &intel_pt_insn)) {
12362306a36Sopenharmony_ci		pr_debug("Intel PT failed to decode: %s\n", dat->asm_rep);
12462306a36Sopenharmony_ci		return -1;
12562306a36Sopenharmony_ci	}
12662306a36Sopenharmony_ci
12762306a36Sopenharmony_ci	if ((int)intel_pt_insn.op != op) {
12862306a36Sopenharmony_ci		pr_debug("Failed to decode 'op' value (%d vs expected %d): %s\n",
12962306a36Sopenharmony_ci			 intel_pt_insn.op, op, dat->asm_rep);
13062306a36Sopenharmony_ci		return -1;
13162306a36Sopenharmony_ci	}
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_ci	if ((int)intel_pt_insn.branch != branch) {
13462306a36Sopenharmony_ci		pr_debug("Failed to decode 'branch' value (%d vs expected %d): %s\n",
13562306a36Sopenharmony_ci			 intel_pt_insn.branch, branch, dat->asm_rep);
13662306a36Sopenharmony_ci		return -1;
13762306a36Sopenharmony_ci	}
13862306a36Sopenharmony_ci
13962306a36Sopenharmony_ci	if (intel_pt_insn.rel != dat->expected_rel) {
14062306a36Sopenharmony_ci		pr_debug("Failed to decode 'rel' value (%#x vs expected %#x): %s\n",
14162306a36Sopenharmony_ci			 intel_pt_insn.rel, dat->expected_rel, dat->asm_rep);
14262306a36Sopenharmony_ci		return -1;
14362306a36Sopenharmony_ci	}
14462306a36Sopenharmony_ci
14562306a36Sopenharmony_ci	pr_debug("Decoded ok: %s\n", dat->asm_rep);
14662306a36Sopenharmony_ci
14762306a36Sopenharmony_ci	return 0;
14862306a36Sopenharmony_ci}
14962306a36Sopenharmony_ci
15062306a36Sopenharmony_cistatic int test_data_set(const struct test_data *dat_set, int x86_64)
15162306a36Sopenharmony_ci{
15262306a36Sopenharmony_ci	const struct test_data *dat;
15362306a36Sopenharmony_ci	int ret = 0;
15462306a36Sopenharmony_ci
15562306a36Sopenharmony_ci	for (dat = dat_set; dat->expected_length; dat++) {
15662306a36Sopenharmony_ci		if (test_data_item(dat, x86_64))
15762306a36Sopenharmony_ci			ret = -1;
15862306a36Sopenharmony_ci	}
15962306a36Sopenharmony_ci
16062306a36Sopenharmony_ci	return ret;
16162306a36Sopenharmony_ci}
16262306a36Sopenharmony_ci
16362306a36Sopenharmony_ci/**
16462306a36Sopenharmony_ci * test__insn_x86 - test x86 instruction decoder - new instructions.
16562306a36Sopenharmony_ci *
16662306a36Sopenharmony_ci * This function implements a test that decodes a selection of instructions and
16762306a36Sopenharmony_ci * checks the results.  The Intel PT function that further categorizes
16862306a36Sopenharmony_ci * instructions (i.e. intel_pt_get_insn()) is also checked.
16962306a36Sopenharmony_ci *
17062306a36Sopenharmony_ci * The instructions are originally in insn-x86-dat-src.c which has been
17162306a36Sopenharmony_ci * processed by scripts gen-insn-x86-dat.sh and gen-insn-x86-dat.awk to produce
17262306a36Sopenharmony_ci * insn-x86-dat-32.c and insn-x86-dat-64.c which are included into this program.
17362306a36Sopenharmony_ci * i.e. to add new instructions to the test, edit insn-x86-dat-src.c, run the
17462306a36Sopenharmony_ci * gen-insn-x86-dat.sh script, make perf, and then run the test.
17562306a36Sopenharmony_ci *
17662306a36Sopenharmony_ci * If the test passes %0 is returned, otherwise %-1 is returned.  Use the
17762306a36Sopenharmony_ci * verbose (-v) option to see all the instructions and whether or not they
17862306a36Sopenharmony_ci * decoded successfully.
17962306a36Sopenharmony_ci */
18062306a36Sopenharmony_ciint test__insn_x86(struct test_suite *test __maybe_unused, int subtest __maybe_unused)
18162306a36Sopenharmony_ci{
18262306a36Sopenharmony_ci	int ret = 0;
18362306a36Sopenharmony_ci
18462306a36Sopenharmony_ci	if (test_data_set(test_data_32, 0))
18562306a36Sopenharmony_ci		ret = -1;
18662306a36Sopenharmony_ci
18762306a36Sopenharmony_ci	if (test_data_set(test_data_64, 1))
18862306a36Sopenharmony_ci		ret = -1;
18962306a36Sopenharmony_ci
19062306a36Sopenharmony_ci	return ret;
19162306a36Sopenharmony_ci}
192