mirror of
https://github.com/ocornut/imgui.git
synced 2026-11-06 14:35:51 +01:00
310 lines
12 KiB
C++
310 lines
12 KiB
C++
#include <pthread.h>
|
||
#include <unistd.h>
|
||
#include <mutex>
|
||
#include <android/log.h>
|
||
#include <EGL/egl.h>
|
||
#include <GLES3/gl3.h>
|
||
#include <dlfcn.h>
|
||
#include <android/input.h>
|
||
#include "dobby.h"
|
||
#include "imgui.h"
|
||
#include "imgui_impl_opengl3.h"
|
||
|
||
#define TAG "PerfectMenu"
|
||
#define LOGI(...) __android_log_print(ANDROID_LOG_INFO, TAG, __VA_ARGS__)
|
||
#define LOGE(...) __android_log_print(ANDROID_LOG_ERROR, TAG, __VA_ARGS__)
|
||
|
||
// ========================= 全局数据结构 =========================
|
||
struct TouchState {
|
||
float x = 0.0f, y = 0.0f; // 经过校准后的屏幕坐标（用于 ImGui）
|
||
bool down = false; // 是否有手指按下
|
||
float renderW = 0.0f, renderH = 0.0f; // 实际渲染区域尺寸（eglQuerySurface）
|
||
|
||
// 自动校准参数
|
||
float windowOffsetX = 0.0f; // 窗口左上角相对于屏幕原点的 X 偏移
|
||
float windowOffsetY = 0.0f; // 窗口左上角相对于屏幕原点的 Y 偏移
|
||
bool calibrated = false; // 是否已完成校准
|
||
|
||
// 手动微调（用于极少数设备自动校准仍有误差的情况）
|
||
float manualOffsetX = 0.0f;
|
||
float manualOffsetY = 0.0f;
|
||
};
|
||
|
||
static TouchState g_Touch;
|
||
static std::mutex g_TouchMutex; // 保护 g_Touch 的互斥锁
|
||
static bool g_ImGuiInitialized = false;
|
||
static int g_FingerCount = 0; // 当前按下的手指数量
|
||
|
||
// ========================= 函数指针定义 =========================
|
||
// eglSwapBuffers
|
||
typedef EGLBoolean (*p_eglSwapBuffers)(EGLDisplay dpy, EGLSurface surface);
|
||
static p_eglSwapBuffers old_eglSwapBuffers = nullptr;
|
||
|
||
// InputConsumer::consume
|
||
typedef int (*p_InputConsumer_consume)(void* instance, void* factory, bool consumeBatches,
|
||
int64_t frameTime, uint32_t* outSeq, void** outEvent);
|
||
static p_InputConsumer_consume old_consume = nullptr;
|
||
|
||
// AMotionEvent 扩展函数（用于获取原始屏幕坐标）
|
||
typedef float (*p_AMotionEvent_getRawX)(const AInputEvent* motion_event, size_t pointer_index);
|
||
typedef float (*p_AMotionEvent_getRawY)(const AInputEvent* motion_event, size_t pointer_index);
|
||
static p_AMotionEvent_getRawX g_getRawX = nullptr;
|
||
static p_AMotionEvent_getRawY g_getRawY = nullptr;
|
||
|
||
// ========================= 辅助函数 =========================
|
||
/**
|
||
* 从 AInputEvent 中提取原始屏幕坐标，并自动校准窗口偏移
|
||
* 使用互斥锁保护触摸数据
|
||
*/
|
||
void handle_android_event(AInputEvent* event) {
|
||
if (AInputEvent_getType(event) != AINPUT_EVENT_TYPE_MOTION) return;
|
||
|
||
int32_t action = AMotionEvent_getAction(event);
|
||
int32_t pointer_idx = (action & AMOTION_EVENT_ACTION_POINTER_INDEX_MASK) >> AMOTION_EVENT_ACTION_POINTER_INDEX_SHIFT;
|
||
int32_t action_mask = action & AMOTION_EVENT_ACTION_MASK;
|
||
|
||
// 获取窗口相对坐标（系统已扣除状态栏/导航栏，但可能未考虑窗口偏移）
|
||
float winX = AMotionEvent_getX(event, pointer_idx);
|
||
float winY = AMotionEvent_getY(event, pointer_idx);
|
||
|
||
float finalX = winX, finalY = winY;
|
||
|
||
// 如果有 Raw 坐标函数，尝试自动校准
|
||
if (g_getRawX && g_getRawY) {
|
||
float rawX = g_getRawX(event, pointer_idx);
|
||
float rawY = g_getRawY(event, pointer_idx);
|
||
|
||
std::lock_guard<std::mutex> lock(g_TouchMutex);
|
||
|
||
// 首次按下时自动校准窗口偏移（取差值）
|
||
if (!g_Touch.calibrated && action_mask == AMOTION_EVENT_ACTION_DOWN) {
|
||
g_Touch.windowOffsetX = rawX - winX;
|
||
g_Touch.windowOffsetY = rawY - winY;
|
||
g_Touch.calibrated = true;
|
||
LOGI("Auto-calibrated: offsetX=%.1f, offsetY=%.1f", g_Touch.windowOffsetX, g_Touch.windowOffsetY);
|
||
}
|
||
|
||
// 使用原始坐标减去窗口偏移，得到真正渲染区域内的坐标
|
||
finalX = rawX - g_Touch.windowOffsetX;
|
||
finalY = rawY - g_Touch.windowOffsetY;
|
||
|
||
// 应用用户手动微调
|
||
finalX += g_Touch.manualOffsetX;
|
||
finalY += g_Touch.manualOffsetY;
|
||
}
|
||
|
||
// 更新手指计数和按下状态
|
||
{
|
||
std::lock_guard<std::mutex> lock(g_TouchMutex);
|
||
|
||
if (action_mask == AMOTION_EVENT_ACTION_DOWN || action_mask == AMOTION_EVENT_ACTION_POINTER_DOWN) {
|
||
g_FingerCount++;
|
||
g_Touch.down = true;
|
||
} else if (action_mask == AMOTION_EVENT_ACTION_UP || action_mask == AMOTION_EVENT_ACTION_POINTER_UP ||
|
||
action_mask == AMOTION_EVENT_ACTION_CANCEL) {
|
||
g_FingerCount--;
|
||
if (g_FingerCount <= 0) {
|
||
g_Touch.down = false;
|
||
g_FingerCount = 0;
|
||
}
|
||
}
|
||
|
||
// 始终用当前触摸事件提供的最主要手指坐标（简化处理，若需跟踪特定手指可扩展）
|
||
g_Touch.x = finalX;
|
||
g_Touch.y = finalY;
|
||
}
|
||
}
|
||
|
||
// ========================= Hook 函数 =========================
|
||
/**
|
||
* Hook InputConsumer::consume
|
||
* 此函数在输入线程中被调用，需要尽快返回，不能做耗时操作
|
||
*/
|
||
int hook_consume(void* a, void* b, bool c, int64_t d, uint32_t* e, void** f) {
|
||
int res = old_consume(a, b, c, d, e, f);
|
||
if (res == 0 && f && *f) {
|
||
handle_android_event((AInputEvent*)*f);
|
||
}
|
||
return res;
|
||
}
|
||
|
||
/**
|
||
* Hook eglSwapBuffers – 渲染主线程，绘制 ImGui 菜单
|
||
*/
|
||
EGLBoolean hook_eglSwapBuffers(EGLDisplay dpy, EGLSurface surface) {
|
||
// 获取渲染区域尺寸（物理像素）
|
||
EGLint w = 0, h = 0;
|
||
eglQuerySurface(dpy, surface, EGL_WIDTH, &w);
|
||
eglQuerySurface(dpy, surface, EGL_HEIGHT, &h);
|
||
if (w <= 0 || h <= 0) {
|
||
return old_eglSwapBuffers(dpy, surface);
|
||
}
|
||
|
||
// 更新渲染尺寸（线程安全读取）
|
||
{
|
||
std::lock_guard<std::mutex> lock(g_TouchMutex);
|
||
g_Touch.renderW = (float)w;
|
||
g_Touch.renderH = (float)h;
|
||
}
|
||
|
||
// 初始化 ImGui（只执行一次，OpenGL 上下文已激活）
|
||
if (!g_ImGuiInitialized) {
|
||
ImGui::CreateContext();
|
||
ImGui_ImplOpenGL3_Init("#version 300 es");
|
||
|
||
// 根据屏幕高度动态调整 UI 大小
|
||
float baseScale = (float)h / 1080.0f; // 以 1080p 为基准
|
||
if (baseScale < 1.0f) baseScale = 1.0f;
|
||
ImGui::GetStyle().ScaleAllSizes(2.5f * baseScale);
|
||
ImGui::GetIO().FontGlobalScale = 2.5f * baseScale;
|
||
|
||
g_ImGuiInitialized = true;
|
||
LOGI("ImGui initialized, render size: %dx%d", w, h);
|
||
}
|
||
|
||
// 获取当前触摸数据（线程安全拷贝）
|
||
float touchX, touchY, renderW, renderH;
|
||
bool touchDown;
|
||
{
|
||
std::lock_guard<std::mutex> lock(g_TouchMutex);
|
||
touchX = g_Touch.x;
|
||
touchY = g_Touch.y;
|
||
touchDown = g_Touch.down;
|
||
renderW = g_Touch.renderW;
|
||
renderH = g_Touch.renderH;
|
||
}
|
||
|
||
ImGuiIO& io = ImGui::GetIO();
|
||
io.DisplaySize = ImVec2(renderW, renderH);
|
||
io.MousePos = ImVec2(touchX, touchY);
|
||
io.MouseDown[0] = touchDown;
|
||
|
||
// 开始新帧
|
||
ImGui_ImplOpenGL3_NewFrame();
|
||
ImGui::NewFrame();
|
||
|
||
// 调试：在手指位置画一个圆点（绿色表示按下，红色表示未按下但显示位置）
|
||
if (touchDown) {
|
||
ImGui::GetForegroundDrawList()->AddCircleFilled(io.MousePos, 20.0f, IM_COL32(0, 255, 0, 180));
|
||
} else {
|
||
ImGui::GetForegroundDrawList()->AddCircle(io.MousePos, 20.0f, IM_COL32(255, 0, 0, 180), 0, 2.0f);
|
||
}
|
||
|
||
// ==================== 绘制主菜单 ====================
|
||
ImGui::SetNextWindowPos(ImVec2(50, 50), ImGuiCond_FirstUseEver);
|
||
ImGui::SetNextWindowSize(ImVec2(renderW * 0.7f, renderH * 0.6f), ImGuiCond_FirstUseEver);
|
||
|
||
if (ImGui::Begin("Perfect Menu", nullptr, ImGuiWindowFlags_NoCollapse)) {
|
||
// 状态显示
|
||
ImGui::Text("Render Area: %.0f x %.0f", renderW, renderH);
|
||
ImGui::Text("Touch Pos: (%.1f, %.1f) %s", touchX, touchY, touchDown ? "[DOWN]" : "[UP]");
|
||
|
||
// 校准信息
|
||
{
|
||
std::lock_guard<std::mutex> lock(g_TouchMutex);
|
||
ImGui::Text("Auto Offset: (%.1f, %.1f) Calibrated: %s",
|
||
g_Touch.windowOffsetX, g_Touch.windowOffsetY,
|
||
g_Touch.calibrated ? "Yes" : "No");
|
||
}
|
||
|
||
ImGui::Separator();
|
||
|
||
// 手动微调偏移（解决极少数设备自动校准不完美的情况）
|
||
float manualX = 0, manualY = 0;
|
||
{
|
||
std::lock_guard<std::mutex> lock(g_TouchMutex);
|
||
manualX = g_Touch.manualOffsetX;
|
||
manualY = g_Touch.manualOffsetY;
|
||
}
|
||
if (ImGui::DragFloat("Manual Offset X", &manualX, 0.5f, -200.0f, 200.0f)) {
|
||
std::lock_guard<std::mutex> lock(g_TouchMutex);
|
||
g_Touch.manualOffsetX = manualX;
|
||
}
|
||
if (ImGui::DragFloat("Manual Offset Y", &manualY, 0.5f, -200.0f, 200.0f)) {
|
||
std::lock_guard<std::mutex> lock(g_TouchMutex);
|
||
g_Touch.manualOffsetY = manualY;
|
||
}
|
||
|
||
ImGui::Separator();
|
||
|
||
if (ImGui::Button("Reset Menu Position", ImVec2(-1, 60))) {
|
||
ImGui::SetWindowPos(ImVec2(100, 100));
|
||
}
|
||
|
||
if (ImGui::Button("Recalibrate (Touch screen now)", ImVec2(-1, 60))) {
|
||
std::lock_guard<std::mutex> lock(g_TouchMutex);
|
||
g_Touch.calibrated = false; // 下次触摸时会重新校准
|
||
g_FingerCount = 0;
|
||
g_Touch.down = false;
|
||
LOGI("Recalibration requested, next touch will reset offset.");
|
||
}
|
||
|
||
ImGui::Separator();
|
||
|
||
// 你的功能示例
|
||
static bool featureEsp = true;
|
||
ImGui::Checkbox("ESP Players", &featureEsp);
|
||
|
||
static float aimFov = 30.0f;
|
||
ImGui::SliderFloat("Aim FOV", &aimFov, 0.0f, 180.0f);
|
||
|
||
ImGui::End();
|
||
}
|
||
|
||
// 渲染 ImGui
|
||
ImGui::Render();
|
||
GLint last_depth;
|
||
glGetIntegerv(GL_DEPTH_TEST, &last_depth);
|
||
glDisable(GL_DEPTH_TEST);
|
||
ImGui_ImplOpenGL3_RenderDrawData(ImGui::GetDrawData());
|
||
if (last_depth) glEnable(GL_DEPTH_TEST);
|
||
|
||
return old_eglSwapBuffers(dpy, surface);
|
||
}
|
||
|
||
// ========================= 初始化线程 =========================
|
||
void* init_thread(void*) {
|
||
sleep(5); // 等待目标进程完全启动
|
||
|
||
// 1. 获取 libandroid.so 中的 Raw 坐标函数
|
||
void* libandroid = dlopen("libandroid.so", RTLD_NOW);
|
||
if (libandroid) {
|
||
g_getRawX = (p_AMotionEvent_getRawX)dlsym(libandroid, "AMotionEvent_getRawX");
|
||
g_getRawY = (p_AMotionEvent_getRawY)dlsym(libandroid, "AMotionEvent_getRawY");
|
||
if (!g_getRawX || !g_getRawY) {
|
||
LOGE("Failed to get AMotionEvent_getRawX/Y, will use window coordinates only (may be inaccurate)");
|
||
} else {
|
||
LOGI("Raw coordinate functions acquired");
|
||
}
|
||
} else {
|
||
LOGE("libandroid.so not found, raw coordinates unavailable");
|
||
}
|
||
|
||
// 2. Hook eglSwapBuffers（渲染）
|
||
void* sym_egl = dlsym(RTLD_DEFAULT, "eglSwapBuffers");
|
||
if (sym_egl) {
|
||
DobbyHook(sym_egl, (void*)hook_eglSwapBuffers, (void**)&old_eglSwapBuffers);
|
||
LOGI("Hooked eglSwapBuffers");
|
||
} else {
|
||
LOGE("eglSwapBuffers not found");
|
||
}
|
||
|
||
// 3. Hook InputConsumer::consume（输入）
|
||
// 符号名来自 libinput.so，可能需要根据 Android 版本调整
|
||
void* consume_addr = DobbySymbolResolver("libinput.so", "_ZN7android13InputConsumer7consumeEPNS_26InputEventFactoryInterfaceEblPjPPNS_10InputEventE");
|
||
if (consume_addr) {
|
||
DobbyHook(consume_addr, (void*)hook_consume, (void**)&old_consume);
|
||
LOGI("Hooked InputConsumer::consume");
|
||
} else {
|
||
LOGE("InputConsumer::consume not found, touch input may not work");
|
||
}
|
||
|
||
return nullptr;
|
||
}
|
||
|
||
// ========================= 入口点 =========================
|
||
__attribute__((constructor)) void init_plugin() {
|
||
pthread_t t;
|
||
pthread_create(&t, nullptr, init_thread, nullptr);
|
||
}
|