953a81afc0
Applying a function signature with afs to a function was using a combination of the existing args and wrong stack addresses to apply the args to function variables. This generally resulted in incorrect locations. Instead, we now use the info from the cc.
112 lines
3.1 KiB
C
112 lines
3.1 KiB
C
// SPDX-FileCopyrightText: 2020 Florian Märkl <info@florianmaerkl.de>
|
|
// SPDX-License-Identifier: LGPL-3.0-only
|
|
|
|
#include <rz_analysis.h>
|
|
|
|
#include "minunit.h"
|
|
#include "test_sdb.h"
|
|
|
|
static Sdb *ref_db() {
|
|
Sdb *db = sdb_new0();
|
|
sdb_set(db, "cc.sectarian.ret", "rax", 0);
|
|
sdb_set(db, "cc.sectarian.arg1", "rcx", 0);
|
|
sdb_set(db, "cc.sectarian.arg0", "rdx", 0);
|
|
sdb_set(db, "cc.sectarian.argn", "stack", 0);
|
|
sdb_set(db, "sectarian", "cc", 0);
|
|
return db;
|
|
}
|
|
|
|
static Sdb *ref_db_self_err() {
|
|
Sdb *db = sdb_new0();
|
|
sdb_set(db, "cc.sectarian.ret", "rax", 0);
|
|
sdb_set(db, "cc.sectarian.self", "rsi", 0);
|
|
sdb_set(db, "cc.sectarian.error", "rdi", 0);
|
|
sdb_set(db, "cc.sectarian.arg1", "rcx", 0);
|
|
sdb_set(db, "cc.sectarian.arg0", "rdx", 0);
|
|
sdb_set(db, "cc.sectarian.argn", "stack", 0);
|
|
sdb_set(db, "sectarian", "cc", 0);
|
|
return db;
|
|
}
|
|
|
|
static RzAnalysis *ref_analysis() {
|
|
RzAnalysis *analysis = rz_analysis_new();
|
|
rz_analysis_cc_set(analysis, "rax sectarian(rdx, rcx, stack)");
|
|
return analysis;
|
|
}
|
|
|
|
static RzAnalysis *ref_analysis_self_err() {
|
|
RzAnalysis *analysis = rz_analysis_new();
|
|
rz_analysis_cc_set(analysis, "rax sectarian(rdx, rcx, stack)");
|
|
rz_analysis_cc_set_self(analysis, "sectarian", "rsi");
|
|
rz_analysis_cc_set_error(analysis, "sectarian", "rdi");
|
|
return analysis;
|
|
}
|
|
|
|
bool test_rz_analysis_cc_set() {
|
|
RzAnalysis *analysis = ref_analysis();
|
|
|
|
Sdb *ref = ref_db();
|
|
assert_sdb_eq(analysis->sdb_cc, ref, "set cc");
|
|
sdb_free(ref);
|
|
|
|
rz_analysis_free(analysis);
|
|
mu_end;
|
|
}
|
|
|
|
bool test_rz_analysis_cc_set_self_err() {
|
|
RzAnalysis *analysis = ref_analysis_self_err();
|
|
|
|
Sdb *ref = ref_db_self_err();
|
|
assert_sdb_eq(analysis->sdb_cc, ref, "set cc");
|
|
sdb_free(ref);
|
|
|
|
rz_analysis_free(analysis);
|
|
mu_end;
|
|
}
|
|
|
|
bool test_rz_analysis_cc_get() {
|
|
RzAnalysis *analysis = ref_analysis();
|
|
char *v = rz_analysis_cc_get(analysis, "sectarian");
|
|
mu_assert_streq(v, "rax sectarian (rdx, rcx, stack);", "get cc");
|
|
free(v);
|
|
const char *vv = rz_analysis_cc_self(analysis, "sectarian");
|
|
mu_assert_null(vv, "get self");
|
|
vv = rz_analysis_cc_error(analysis, "sectarian");
|
|
mu_assert_null(vv, "get error");
|
|
rz_analysis_free(analysis);
|
|
mu_end;
|
|
}
|
|
|
|
bool test_rz_analysis_cc_get_self_err() {
|
|
RzAnalysis *analysis = ref_analysis_self_err();
|
|
char *v = rz_analysis_cc_get(analysis, "sectarian");
|
|
mu_assert_streq(v, "rax rsi.sectarian (rdx, rcx, stack) rdi;", "get cc");
|
|
free(v);
|
|
const char *vv = rz_analysis_cc_self(analysis, "sectarian");
|
|
mu_assert_streq(vv, "rsi", "get self");
|
|
vv = rz_analysis_cc_error(analysis, "sectarian");
|
|
mu_assert_streq(vv, "rdi", "get error");
|
|
rz_analysis_free(analysis);
|
|
mu_end;
|
|
}
|
|
|
|
bool test_rz_analysis_cc_del() {
|
|
RzAnalysis *analysis = ref_analysis();
|
|
rz_analysis_cc_del(analysis, "sectarian");
|
|
Sdb *ref = sdb_new0();
|
|
assert_sdb_eq(analysis->sdb_cc, ref, "deleted");
|
|
sdb_free(ref);
|
|
rz_analysis_free(analysis);
|
|
mu_end;
|
|
}
|
|
|
|
bool all_tests() {
|
|
mu_run_test(test_rz_analysis_cc_set);
|
|
mu_run_test(test_rz_analysis_cc_set_self_err);
|
|
mu_run_test(test_rz_analysis_cc_get);
|
|
mu_run_test(test_rz_analysis_cc_get_self_err);
|
|
mu_run_test(test_rz_analysis_cc_del);
|
|
return tests_passed != tests_run;
|
|
}
|
|
|
|
mu_main(all_tests)
|