fix ionice set not working on windows x64 due to LENGTH_MISMATCH (#1368)

Fix Process().ionice(0/1/2) length mismatch on Windows

1 files changed, 6 insertions, 5 deletions

diff --git a/psutil/_psutil_windows.c b/psutil/_psutil_windows.c
index edb5996c..29311992 100644
--- a/psutil/_psutil_windows.c
+++ b/psutil/_psutil_windows.c
@@ -2098,7 +2098,7 @@ static PyObject *
 psutil_proc_io_priority_get(PyObject *self, PyObject *args) {
     long pid;
     HANDLE hProcess;
-    PULONG IoPriority;
+    DWORD IoPriority;
 
     _NtQueryInformationProcess NtQueryInformationProcess =
         (_NtQueryInformationProcess)GetProcAddress(
@@ -2114,7 +2114,7 @@ psutil_proc_io_priority_get(PyObject *self, PyObject *args) {
         hProcess,
         ProcessIoPriority,
         &IoPriority,
-        sizeof(ULONG),
+        sizeof(DWORD),
         NULL
     );
     CloseHandle(hProcess);
@@ -2128,8 +2128,9 @@ psutil_proc_io_priority_get(PyObject *self, PyObject *args) {
 static PyObject *
 psutil_proc_io_priority_set(PyObject *self, PyObject *args) {
     long pid;
-    int prio;
+    DWORD prio;
     HANDLE hProcess;
+    DWORD dwDesiredAccess = PROCESS_QUERY_INFORMATION | PROCESS_SET_INFORMATION;
 
     _NtSetInformationProcess NtSetInformationProcess =
         (_NtSetInformationProcess)GetProcAddress(
@@ -2143,7 +2144,7 @@ psutil_proc_io_priority_set(PyObject *self, PyObject *args) {
 
     if (! PyArg_ParseTuple(args, "li", &pid, &prio))
         return NULL;
-    hProcess = psutil_handle_from_pid_waccess(pid, PROCESS_ALL_ACCESS);
+    hProcess = psutil_handle_from_pid_waccess(pid, dwDesiredAccess);
     if (hProcess == NULL)
         return NULL;
 
@@ -2151,7 +2152,7 @@ psutil_proc_io_priority_set(PyObject *self, PyObject *args) {
         hProcess,
         ProcessIoPriority,
         (PVOID)&prio,
-        sizeof((PVOID)prio)
+        sizeof(DWORD)
     );
 
     CloseHandle(hProcess);