rdbo / libmem

Advanced Game Hacking Library for C, Modern C++, Rust and Python (Windows/Linux/FreeBSD) (Process/Memory Hacking) (Hooking/Detouring) (Cross Platform) (x86/x64/ARM/ARM64) (DLL/SO Injection) (Internal/External) (Assembler/Disassembler)
GNU Affero General Public License v3.0
764 stars 92 forks source link

vftable, vtable #129

Closed illegal-instruction-co closed 1 year ago

illegal-instruction-co commented 1 year ago

Would be nice to add resolve vftable or vtable helper class to hook some methods in compiling time.

I am thinking to use libmem to reduce dependencies on my next projects for unit tests by hooks. I never saw a methodogy like that for testing process, but i think it would be really usefull.

rdbo commented 1 year ago

I am not sure what do you mean by hooking methods in compile time, but there is an API for hooking vtables. There are no classes, because it is compatible with C, and there is no C++ specific version, only C/C++. To hook a VMT:

lm_vmt_t some_object_vmt;

LM_VmtNew(*(lm_address_t **)some_object_addr, &some_object_vmt);
LM_VmtHook(&some_object_vmt, 0, hk_some_function); /* hook function of index 0 from some_object's VMT */

/* freeing the VMT */
LM_VmtFree(&some_object_vmt);
illegal-instruction-co commented 1 year ago

I am not sure what do you mean by hooking methods in compile time, but there is an API for hooking vtables. There are no classes, because it is compatible with C, and there is no C++ specific version, only C/C++. To hook a VMT:

lm_vmt_t some_object_vmt;

LM_VmtNew(*(lm_address_t **)some_object_addr, &some_object_vmt);
LM_VmtHook(&some_object_vmt, 0, hk_some_function); /* hook function of index 0 from some_object's VMT */

/* freeing the VMT */
LM_VmtFree(&some_object_vmt);

I didnt meant hooking in compile time (:, its about the get address of virtual method in compile time. So basically, if the index changed because of the main code refactored, test case will be healthy as not needed to change.

rdbo commented 1 year ago

You can get the address of the function using LM_VmtGetOriginal:

lm_vmt_t some_object_vmt;
lm_address_t orig_addr;

/* ... */

orig_addr = LM_VmtGetOriginal(&some_object_vmt, 5); /* get original function address at index 5 */

Also, if the index changed because of the main code refactored - not sure what this means either, the index should never change, unless the class itself has changed (e.g they added a new virtual function before the one you're accessing) or at least that's my understanding of it.

illegal-instruction-co commented 1 year ago

You can get the address of the function using LM_VmtGetOriginal:

lm_vmt_t some_object_vmt;
lm_address_t orig_addr;

/* ... */

orig_addr = LM_VmtGetOriginal(&some_object_vmt, 5); /* get original function address at index 5 */

Also, if the index changed because of the main code refactored - not sure what this means either, the index should never change, unless the class itself has changed (e.g they added a new virtual function before the one you're accessing) or at least that's my understanding of it.

exactly you got correct.whenever they ( actually we since im not hacking some one else's memory :) ) added new virtual function, the test case shouldnt also need refactoring. so in the compile time, we can create a helper class to solve actual index of function ( or address )

rdbo commented 1 year ago

If you want to get an index from the function address + object, I don't think C++ allows you to get the address of a virtual function. But assuming you got the address, should be something like this (untested):

lm_size_t get_function_index(lm_vmt_t *vmt, lm_address_t func_addr)
{
        lm_size_t i;
        lm_address_t addr = LM_ADDRESS_BAD;
        for (;; ++i) {
                addr = LM_VmtGetOriginal(vmt, i);
                if (addr == func_addr)
                        break;
        }

        return i;
}
illegal-instruction-co commented 1 year ago

Yes, we can find the function index by bruteforcing checks by a loop if it is equals to address; but in the test cases you should write your code as easy to understand and you are right we are not able to get address of virtual method, but i bet compiler solving it at the compiling time and packs to code in that way. So, somehow we can get address or index at compiling time too. So it could be like this: int index = lm_smth(Foo.Vmethod);

rdbo commented 1 year ago

Yes, we can find the function index by bruteforcing checks by a loop if it is equals to address; but in the test cases you should write your code as easy to understand and you are right we are not able to get address of virtual method, but i bet compiler solving it at the compiling time and packs to code in that way. So, somehow we can get address or index at compiling time too. So it could be like this: int index = lm_smth(Foo.Vmethod);

Again, if you have the function address, you can use the get_function_index function I provided. Thing is, C++ itself does not allow you to get the address of a virtual method at compile time. So this functionality would not be possible. Also, the code in the tests are just to make sure libmem works, the examples for users to read have their own folder (although it is lacking some more examples right now). Best way is probably to just hardcode your indices and avoid changing them, or using signature scanning to find the function address, and then using the get_function_index method to find the index.

illegal-instruction-co commented 1 year ago

signature scanning would be best approach. thank you, i am closing this issue.