portugalfuturista/rizin
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add Integration Test for RzBinVirtualFiles without RzCore (#1186)

Browse source
This commit is contained in:
Florian Märkl 2021-06-03 06:06:34 +02:00 committed by GitHub
parent 388aef5140
commit d0c66d2724
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
4 changed files with 80 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
librz/bin/bobj.c
View file

@ -580,6 +580,21 @@ RZ_API RzBinSymbol *rz_bin_object_get_symbol_of_import(RzBinObject *o, RzBinImpo
return ht_pp_find(o->import_name_symbols, imp->name, NULL);
}
RZ_API RzBinVirtualFile *rz_bin_object_get_virtual_file(RzBinObject *o, const char *name) {
rz_return_val_if_fail(o && name, NULL);
if (!o->vfiles) {
return NULL;
}
RzListIter *it;
RzBinVirtualFile *vf;
rz_list_foreach (o->vfiles, it, vf) {
if (!strcmp(vf->name, name)) {
return vf;
}
}
return NULL;
}
RZ_IPI RzBinObject *rz_bin_object_get_cur(RzBin *bin) {
rz_return_val_if_fail(bin && bin->cur, NULL);
return bin->cur->o;

1
librz/include/rz_bin.h
View file

@ -944,6 +944,7 @@ RZ_API ut64 rz_bin_object_get_vaddr(RzBinObject *o, ut64 paddr, ut64 vaddr);
RZ_API RzBinAddr *rz_bin_object_get_special_symbol(RzBinObject *o, RzBinSpecialSymbol sym);
RZ_API RzBinRelocStorage *rz_bin_object_patch_relocs(RzBinFile *bf, RzBinObject *o);
RZ_API RzBinSymbol *rz_bin_object_get_symbol_of_import(RzBinObject *o, RzBinImport *imp);
RZ_API RzBinVirtualFile *rz_bin_object_get_virtual_file(RzBinObject *o, const char *name);
RZ_API void rz_bin_mem_free(void *data);
// demangle functions

1
test/integration/meson.build
View file

@ -3,6 +3,7 @@ if get_option('enable_tests')
tests = [
'open_analyse_save_load_project',
'bin_vfiles'
]
unit_test_env = environment()

63
test/integration/test_bin_vfiles.c Normal file
View file

@ -0,0 +1,63 @@
// SPDX-FileCopyrightText: 2021 Florian Märkl <info@florianmaerkl.de>
// SPDX-License-Identifier: LGPL-3.0-only
#include <rz_bin.h>
#include "../unit/minunit.h"
/// Test using RzBin without RzCore and extract some data from a vfiile
bool test_bin_vfiles() {
// 1. Open the file as RzBuffer
RzBuffer *buf = rz_buf_new_file("bins/elf/ls", O_RDONLY, 0644);
mu_assert_notnull(buf, "open file");
// 2. Load the buffer into RzBin
RzBin *bin = rz_bin_new();
RzBinOptions opts = { 0 };
opts.filename = "<internal>";
opts.obj_opts.patch_relocs = true;
RzBinFile *bf = rz_bin_open_buf(bin, buf, &opts);
mu_assert_notnull(bf, "load bin file");
// 2. Extract the reloc-patched buffer
mu_assert_notnull(bf->o, "object");
mu_assert_notnull(bf->o->vfiles, "vfiles");
RzBinVirtualFile *patched = rz_bin_object_get_virtual_file(bf->o, "patched");
mu_assert_notnull(patched, "patched vfile");
// 3. Find an interesting reloc by an import name
mu_assert_notnull(bf->o->relocs, "relocs");
RzBinReloc *reloc = NULL;
for (size_t i = 0; i < bf->o->relocs->relocs_count; i++) {
RzBinReloc *r = bf->o->relocs->relocs[i];
if (!r->import) {
continue;
}
if (!strcmp(r->import->name, "isatty")) {
reloc = r;
break;
}
}
mu_assert_notnull(reloc, "found reloc by name");
mu_assert_eq(reloc->paddr, 0x00020ce8, "reloc paddr");
mu_assert_eq(reloc->vaddr, 0x00021ce8, "reloc vaddr");
mu_assert_neq(reloc->target_vaddr, 0, "target not 0");
mu_assert_neq(reloc->target_vaddr, UT64_MAX, "target not UT64_MAX");
// 4. Check the contents of the original buf and the vfile against the data in the reloc
ut64 val = rz_buf_read_le64_at(buf, reloc->paddr);
mu_assert_eq(val, 0, "original buf has nothing patched");
val = rz_buf_read_le64_at(patched->buf, reloc->paddr);
mu_assert_eq(val, reloc->target_vaddr, "read patched target from vfile");
// 5. Exit
rz_bin_free(bin);
mu_end;
}
int all_tests() {
mu_run_test(test_bin_vfiles);
return tests_passed != tests_run;
}
mu_main(all_tests)