win32u: Move NtUserSetThreadDesktop implementation from user32.

Signed-off-by: Jacek Caban <jacek@codeweavers.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>

dlls/user32/user_main.c

@@ -311,7 +311,7 @@ static void winstation_init(void)
     {
         handle = CreateDesktopW( desktop ? desktop : get_default_desktop(),
                                  NULL, NULL, 0, DESKTOP_ALL_ACCESS, NULL );
-        if (handle) SetThreadDesktop( handle );
+        if (handle) NtUserSetThreadDesktop( handle );
     }
     HeapFree( GetProcessHeap(), 0, buffer );
 

dlls/user32/winstation.c

@@ -362,14 +362,8 @@ HDESK WINAPI OpenDesktopW( LPCWSTR name, DWORD flags, BOOL inherit, ACCESS_MASK
  */
 BOOL WINAPI SetThreadDesktop( HDESK handle )
 {
-    BOOL ret;
+    BOOL ret = NtUserSetThreadDesktop( handle );
 
-    SERVER_START_REQ( set_thread_desktop )
-    {
-        req->handle = wine_server_obj_handle( handle );
-        ret = !wine_server_call_err( req );
-    }
-    SERVER_END_REQ;
     if (ret)  /* reset the desktop windows */
     {
         struct user_thread_info *thread_info = get_user_thread_info();

dlls/win32u/syscall.c

@@ -99,6 +99,7 @@ static void * const syscalls[] =
     NtUserGetProcessWindowStation,
     NtUserGetThreadDesktop,
     NtUserSetProcessWindowStation,
+    NtUserSetThreadDesktop,
 };
 
 static BYTE arguments[ARRAY_SIZE(syscalls)];

dlls/win32u/win32u.spec

@@ -1232,7 +1232,7 @@
 @ stub NtUserSetSystemMenu
 @ stub NtUserSetSystemTimer
 @ stub NtUserSetTargetForResourceBrokering
-@ stub NtUserSetThreadDesktop
+@ stdcall -syscall NtUserSetThreadDesktop(long)
 @ stub NtUserSetThreadInputBlocked
 @ stub NtUserSetThreadLayoutHandles
 @ stub NtUserSetThreadState

dlls/win32u/winstation.c

@@ -106,3 +106,21 @@ HDESK WINAPI NtUserGetThreadDesktop( DWORD thread )
     SERVER_END_REQ;
     return ret;
 }
+
+/***********************************************************************
+ *           NtUserSetThreadDesktop   (win32u.@)
+ */
+BOOL WINAPI NtUserSetThreadDesktop( HDESK handle )
+{
+    BOOL ret;
+
+    SERVER_START_REQ( set_thread_desktop )
+    {
+        req->handle = wine_server_obj_handle( handle );
+        ret = !wine_server_call_err( req );
+    }
+    SERVER_END_REQ;
+
+    /* FIXME: reset uset thread info */
+    return ret;
+}

dlls/wow64win/syscall.h

@@ -85,6 +85,7 @@
     SYSCALL_ENTRY( NtUserCloseWindowStation ) \
     SYSCALL_ENTRY( NtUserGetProcessWindowStation ) \
     SYSCALL_ENTRY( NtUserGetThreadDesktop ) \
-    SYSCALL_ENTRY( NtUserSetProcessWindowStation )
+    SYSCALL_ENTRY( NtUserSetProcessWindowStation ) \
+    SYSCALL_ENTRY( NtUserSetThreadDesktop )
 
 #endif /* __WOW64WIN_SYSCALL_H */

dlls/wow64win/user.c

@@ -59,3 +59,10 @@ NTSTATUS WINAPI wow64_NtUserGetThreadDesktop( UINT *args )
 
     return HandleToUlong( NtUserGetThreadDesktop( thread ));
 }
+
+NTSTATUS WINAPI wow64_NtUserSetThreadDesktop( UINT *args )
+{
+    HDESK handle = get_handle( &args );
+
+    return NtUserSetThreadDesktop( handle );
+}

include/ntuser.h

@@ -27,5 +27,6 @@ BOOL    WINAPI NtUserCloseWindowStation( HWINSTA handle );
 HWINSTA WINAPI NtUserGetProcessWindowStation(void);
 HDESK   WINAPI NtUserGetThreadDesktop( DWORD thread );
 BOOL    WINAPI NtUserSetProcessWindowStation( HWINSTA handle );
+BOOL    WINAPI NtUserSetThreadDesktop( HDESK handle );
 
 #endif /* _NTUSER_ */