#include "memflow.h"
#include <stdio.h>
#include <string.h>
struct FindProcessContext {
OsInstance *os;
const char *name;
ProcessInstance *target_process;
bool found;
};
bool find_process(struct FindProcessContext *find_context, Address addr) {
if (find_context->found) {
return false;
}
if (mf_osinstance_process_by_address(
find_context->os,
addr,
find_context->target_process
)) {
return true;
}
const struct ProcessInfo *info = mf_processinstance_info(find_context->target_process);
if (!strcmp(info->name, find_context->name)) {
find_context->found = true;
return false;
}
mf_processinstance_drop(*find_context->target_process);
return true;
}
int main(int argc, char *argv[]) {
int ret = 0;
log_init(3);
Inventory *inventory = inventory_scan();
printf("inventory initialized: %p\n", inventory);
const char *conn_name = argc > 1 ? argv[1] : "qemu";
const char *conn_arg = argc > 2 ? argv[2] : "";
const char *os_name = argc > 3 ? argv[3]: "win32";
const char *os_arg = argc > 4? argv[4]: "";
const char *target_proc = argc > 5? argv[5]: "notepad.exe";
ConnectorInstance connector, *conn = conn_name[0] ? &connector : NULL;
if (conn) {
if (inventory_create_connector(inventory, conn_name, conn_arg, conn)) {
log_error("unable to initialize connector");
inventory_free(inventory);
return 1;
}
printf("connector initialized: %p\n", connector.container.instance.instance);
}
OsInstance os;
if (inventory_create_os(inventory, os_name, os_arg, conn, &os)) {
log_error("unable to initialize os plugin");
inventory_free(inventory);
return 1;
}
printf("os plugin initialized: %p\n", os.container.instance.instance);
ProcessInstance target_process;
struct FindProcessContext find_context = {
&os,
target_proc,
&target_process,
false,
};
mf_osinstance_process_address_list_callback(&os, CALLBACK(Address, &find_context, find_process));
if (find_context.found) {
const struct ProcessInfo *info = mf_processinstance_info(&target_process);
printf("%s process found: 0x%lx] %d %s %s\n", target_proc, info->address,
info->pid, info->name, info->path);
mf_processinstance_drop(target_process);
} else {
printf("Unable to find %s\n", target_proc);
}
if (!(ret = mf_osinstance_process_by_name(&os, STR(target_proc), &target_process))) {
const struct ProcessInfo *info = mf_processinstance_info(&target_process);
printf("%s process found: 0x%lx] %d %s %s\n", target_proc, info->address,
info->pid, info->name, info->path);
mf_processinstance_drop(target_process);
} else {
printf("Unable to find %s\n", target_proc);
log_debug_errorcode(ret);
}
mf_osinstance_drop(os);
log_info("os plugin/connector freed");
inventory_free(inventory);
log_info("inventory freed");
return 0;
}