Java JNA - получить позицию рабочего стола - PullRequest
Новая
       25

Java JNA - получить позицию рабочего стола

4 голосов
/ 03 апреля 2019

Цель

Получение координат рабочего стола элемента / значка рабочего стола.

Попытка

Я получил дескриптор окна SysListView32, который содержит значки на рабочем столе, используя: 

HWND hWnd_Progman = User32.INSTANCE.FindWindow("Progman", "Program Manager");
HWND hWnd_SHELLDLL_DefView = User32.INSTANCE.FindWindowEx(hWnd_Progman, null, "SHELLDLL_DefView", null);
HWND hWnd_SysListView32 = User32.INSTANCE.FindWindowEx(hWnd_SHELLDLL_DefView, null, "SysListView32", "FolderView");

Я получил количество элементов рабочего стола: 

LRESULT result = User32.INSTANCE.SendMessage(hWnd_SysListView32, LVM_GETITEMCOUNT, new WPARAM(), new LPARAM());
long desktopIconCount = result.longValue();

У меня установлено положение элемента рабочего стола (проверка того, что SysListView32 является правильным списком для элементов рабочего стола).Переданные координаты x и y соответствуют смещению от левого верхнего края моего левого монитора до левого верхнего угла элемента рабочего стола.Код: 

int itemIndex = 0; // Allows 0 to desktopIconCount - 1.
int x = ...;
int y = ...;
LRESULT res = User32.INSTANCE.SendMessage(hWnd_SysListView32, LVM_SETITEMPOSITION, new WPARAM(itemIndex), new LPARAM((x & 0xFFFF) | (y << 16)));

Теперь, чтобы получить позицию элемента рабочего стола, необходимо отправить LVM_GETITEMPOSITION в SysListView32 и включить указатель на адрес, по которому он может записать позицию.Однако этот указатель должен быть действительным адресом в памяти процесса, принадлежащего SysListView32.Поэтому я попытался сделать следующее:

  • Получить процесс, принадлежащий SysListView32.
  • Выделить память в этом процессе.
  • Записать объект POINT вэта память (используется для позиции элемента).
  • Отправка LVM_GETITEMPOSITION в SysListView32 с указателем на эту выделенную память.
  • Считывание этого POINT объекта из памяти.На этом этапе процесс должен был записать в него позицию элемента рабочего стола.

Я пробовал это со следующим кодом: 

// Get the SysListView32 process handle.
IntByReference processIdRef = new IntByReference();
User32.INSTANCE.GetWindowThreadProcessId(hWnd_SysListView32, processIdRef);
HANDLE procHandle = Kernel32.INSTANCE.OpenProcess(
        Kernel32.PROCESS_VM_OPERATION | Kernel32.PROCESS_VM_WRITE | Kernel32.PROCESS_VM_READ,
        false, processIdRef.getValue());

// Allocate memory in the SysView32 process.
int pointSize = Native.getNativeSize(POINT.class)); // 8 bytes.
LPVOID pMem = MyKernel32.INSTANCE.VirtualAllocEx(procHandle, new LPVOID(), new SIZE_T(pointSize),
        MyKernel32.MEM_COMMIT, MyKernel32.PAGE_READWRITE);

// Put some POINT-sized object in the process its memory.
boolean success = Kernel32.INSTANCE.WriteProcessMemory(
        procHandle, pMem.getPointer(), pMem.getPointer(), pointSize, null);
if(!success) {
    System.out.println("Write error = " + Kernel32.INSTANCE.GetLastError());
    System.exit(1);
}

// Send the LVM_GETITEMPOSITION message to the SysListView32.
int itemIndex = 0; // Allows 0 to desktopIconCount - 1.
LRESULT res = MyUser32.INSTANCE.SendMessage(
        hWnd_SysListView32, LVM_GETITEMPOSITION, new WPARAM(itemIndex), pMem.getPointer());
System.out.println("Message result = " + res.longValue());

// Read the earlier POINT-sized written memory.
POINT point = new POINT();
success = Kernel32.INSTANCE.ReadProcessMemory(
        procHandle, pMem.getPointer(), point.getPointer(), pointSize, null);
if(!success) {
    System.out.println("Read error = " + Kernel32.INSTANCE.GetLastError());
    System.exit(1);
}
System.out.println("Point found: x=" + pos.x + ", y=" + pos.y);

Здесь, MyUser32 созданоследующим образом: 

interface MyUser32 extends User32 {
    static MyUser32 INSTANCE =
        (MyUser32) Native.load("user32", MyUser32.class, W32APIOptions.DEFAULT_OPTIONS);
    LRESULT SendMessage(HWND hWnd, int msg, WPARAM wParam, Pointer pointer);
}

и MyKernel32 создаются следующим образом: 

interface MyKernel32 extends Kernel32 {
    static final MyKernel32 INSTANCE =
            (MyKernel32) Native.load("kernel32", MyKernel32.class, W32APIOptions.DEFAULT_OPTIONS);
    static int MEM_COMMIT = 0x1000;
    static int PAGE_READWRITE = 0x04;
    LPVOID VirtualAllocEx(HANDLE hProcess, LPVOID lpAddress, SIZE_T dwSize, int flAllocationType, int flProtect);
}

Для завершения используются следующие дополнительные статические значения: 

static final int LVM_FIRST = 0x1000;
static final int LVM_GETITEMCOUNT = LVM_FIRST + 4;
static final int LVM_SETITEMPOSITION = LVM_FIRST + 15;
static final int LVM_GETITEMPOSITION = LVM_FIRST + 16;

Проблема

Вызов WriteProcessMemory часто завершается ошибкой с кодом ошибки 299 ERROR_PARTIAL_COPY, и даже если он не выходит из строя, возвращаемое значение POINT всегда равно (0,0).Я ожидаю, что проблема будет либо в объявлениях метода SendMessage / VirtualAllocEx в MyUser32 или MyKernel32, либо в том, что я неправильно понимаю, какой объект / указатель должен быть передан VirtualAllocEx или WriteProcessMemory.

Я провел много исследований и выяснил, как это должно работать в C / C ++, но я не смог найти ни одного примера рабочего кода, использующего JNA для моего случая.

Спасибо за проявленный интереси пытаюсь помочь, если ты прошел через мое сообщение.

1 Ответ

2 голосов
/ 14 апреля 2019

Проблема в том, что com.sun.jna.Native.getNativeSize (Class) не подходит для использования в этом случае. Проблема видна при использовании 32-битной виртуальной машины Java (не видна 64-битная виртуальная машина).

Для структур вышеупомянутая функция предполагает, что они передаются по ссылке (указатель на структуру) и, таким образом, функция возвращает значение Native.POINTER_SIZE . На 64-битной виртуальной машине это, к счастью, соответствует размеру структуры POINT . На 32-битной ВМ Native.POINTER_SIZE имеет размер 4 байта, поэтому может содержать только части структуры результата.

Наиболее важные части: чтобы определить размер структуры в JNA, используйте функцию Structure # size. В этом случае также полезно использовать последний параметр ReadProcessMemory . Функция возвращает количество прочитанных байтов и показывает разницу (4 против 8).

Дополнительные комментарии: не забудьте освободить выделенную память, а также закрыть дескриптор процесса для полученного.

Вот пример полного запуска (только отсутствующие импорта, протестированные с JNA 5.2): 

public class Test {

    private interface Kernel32 extends com.sun.jna.platform.win32.Kernel32 {

        Kernel32 INSTANCE = Native.load("kernel32", Kernel32.class, W32APIOptions.DEFAULT_OPTIONS);

        public Pointer VirtualAllocEx(HANDLE hProcess, Pointer lpAddress, SIZE_T dwSize, int flAllocationType, int flProtect);

        public boolean VirtualFreeEx(HANDLE hProcess, Pointer lpAddress, SIZE_T dwSize, int dwFreeType);

        int MEM_COMMIT = 0x00001000;
        int MEM_RESERVE = 0x00002000;
        int MEM_RESET = 0x00080000;
        int MEM_RESET_UNDO = 0x1000000;
        int MEM_LARGE_PAGES = 0x20000000;
        int MEM_PHYSICAL = 0x00400000;
        int MEM_TOP_DOWN = 0x00100000;
        int MEM_COALESCE_PLACEHOLDERS = 0x00000001;
        int MEM_PRESERVE_PLACEHOLDER = 0x00000002;
        int MEM_DECOMMIT = 0x4000;
        int MEM_RELEASE = 0x8000;
    }

    private static final int LVM_FIRST = 0x1000;
    private static final int LVM_GETITEMCOUNT = LVM_FIRST + 4;
    private static final int LVM_GETITEMPOSITION = LVM_FIRST + 16;

    public static void main(String[] args) throws IOException, InterruptedException {
        // Find the HWND for the "desktop" list view
        HWND hWnd_Progman = User32.INSTANCE.FindWindow("Progman", "Program Manager");
        HWND hWnd_SHELLDLL_DefView = User32.INSTANCE.FindWindowEx(hWnd_Progman, null, "SHELLDLL_DefView", null);
        HWND hWnd_SysListView32 = User32.INSTANCE.FindWindowEx(hWnd_SHELLDLL_DefView, null, "SysListView32", "FolderView");
        // Fetch the icon count
        int itemCount = User32.INSTANCE.SendMessage(hWnd_SysListView32, LVM_GETITEMCOUNT, new WPARAM(), new LPARAM()).intValue();
        System.out.println("Desktop Icons: " + itemCount);

        // Get the SysListView32 process handle.
        IntByReference processIdRef = new IntByReference();
        User32.INSTANCE.GetWindowThreadProcessId(hWnd_SysListView32, processIdRef);
        HANDLE procHandle = Kernel32.INSTANCE.OpenProcess(
                Kernel32.PROCESS_VM_OPERATION | Kernel32.PROCESS_VM_WRITE | Kernel32.PROCESS_VM_READ,
                false, processIdRef.getValue());

        // Allocate memory in the SysView32 process.
        int pointSize = new POINT().size(); // 8 bytes.
        Pointer pMem = Kernel32.INSTANCE.VirtualAllocEx(procHandle, null, new SIZE_T(pointSize),
                Kernel32.MEM_COMMIT, Kernel32.PAGE_READWRITE);

        for (int i = 0; i < itemCount; i++) {

            // Send the LVM_GETITEMPOSITION message to the SysListView32.
            LRESULT res = User32.INSTANCE.SendMessage(
                    hWnd_SysListView32, LVM_GETITEMPOSITION, new WPARAM(i), new LPARAM(Pointer.nativeValue(pMem)));

            if(res.intValue() != 1) {
                throw new IllegalStateException("Message sending failed");
            }

            // Read the earlier POINT-sized written memory.
            POINT point = new POINT();
            IntByReference read = new IntByReference();
            boolean success = Kernel32.INSTANCE.ReadProcessMemory(
                    procHandle, pMem, point.getPointer(), pointSize, read);

            if (!success) {
                System.out.println("Read error = " + Kernel32.INSTANCE.GetLastError());
                System.exit(1);
            }
            point.read();
            System.out.println("Point found: x=" + point.x + ", y=" + point.y);
        }

        // Release allocated memory
        Kernel32.INSTANCE.VirtualFreeEx(procHandle, pMem, new SIZE_T(0), Kernel32.MEM_RELEASE);

        // Close Process Handle
        Kernel32.INSTANCE.CloseHandle(procHandle);
    }
}
...