dlls/user32/winstation.c

   1 /*
   2  * Window stations and desktops
   3  *
   4  * Copyright 2002 Alexandre Julliard
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include "ntstatus.h"
  22 #define WIN32_NO_STATUS
  23
  24 #include <stdarg.h>
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "winnls.h"
  28 #include "winerror.h"
  29 #include "wingdi.h"
  30 #include "winuser.h"
  31 #include "winternl.h"
  32 #include "ddk/wdm.h"
  33 #include "wine/server.h"
  34 #include "wine/debug.h"
  35 #include "user_private.h"
  36
  37 WINE_DEFAULT_DEBUG_CHANNEL(winstation);
  38
  39
  40 /* callback for enumeration functions */
  41 struct enum_proc_lparam
  42 {
  43     NAMEENUMPROCA func;
  44     LPARAM        lparam;
  45 };
  46
  47 static BOOL CALLBACK enum_names_WtoA( LPWSTR name, LPARAM lparam )
  48 {
  49     struct enum_proc_lparam *data = (struct enum_proc_lparam *)lparam;
  50     char buffer[MAX_PATH];
  51
  52     if (!WideCharToMultiByte( CP_ACP, 0, name, -1, buffer, sizeof(buffer), NULL, NULL ))
  53         return FALSE;
  54     return data->func( buffer, data->lparam );
  55 }
  56
  57 /* return a handle to the directory where window station objects are created */
  58 static HANDLE get_winstations_dir_handle(void)
  59 {
  60     static HANDLE handle = NULL;
  61     WCHAR buffer[64];
  62     UNICODE_STRING str;
  63     OBJECT_ATTRIBUTES attr;
  64
  65     if (!handle)
  66     {
  67         HANDLE dir;
  68
  69         swprintf( buffer, ARRAY_SIZE(buffer), L"\\Sessions\\%u\\Windows\\WindowStations", NtCurrentTeb()->Peb->SessionId );
  70         RtlInitUnicodeString( &str, buffer );
  71         InitializeObjectAttributes( &attr, &str, 0, 0, NULL );
  72         NtOpenDirectoryObject( &dir, DIRECTORY_CREATE_OBJECT | DIRECTORY_TRAVERSE, &attr );
  73         if (InterlockedCompareExchangePointer( &handle, dir, 0 ) != 0) /* someone beat us here */
  74             CloseHandle( dir );
  75     }
  76     return handle;
  77 }
  78
  79 static WCHAR default_name[29];
  80
  81 static BOOL WINAPI winstation_default_name_once( INIT_ONCE *once, void *param, void **context )
  82 {
  83     TOKEN_STATISTICS stats;
  84     BOOL ret;
  85
  86     ret = GetTokenInformation( GetCurrentProcessToken(), TokenStatistics, &stats, sizeof(stats), NULL );
  87     if (ret)
  88         swprintf( default_name, ARRAY_SIZE(default_name), L"Service-0x%x-%x$",
  89                   stats.AuthenticationId.HighPart, stats.AuthenticationId.LowPart );
  90
  91     return ret;
  92 }
  93
  94 static const WCHAR *get_winstation_default_name( void )
  95 {
  96     static INIT_ONCE once = INIT_ONCE_STATIC_INIT;
  97     BOOL ret;
  98
  99     ret = InitOnceExecuteOnce( &once, winstation_default_name_once, NULL, NULL );
 100     return ret ? default_name : NULL;
 101 }
 102
 103 /***********************************************************************
 104  *              CreateWindowStationA  (USER32.@)
 105  */
 106 HWINSTA WINAPI CreateWindowStationA( LPCSTR name, DWORD flags, ACCESS_MASK access,
 107                                      LPSECURITY_ATTRIBUTES sa )
 108 {
 109     WCHAR buffer[MAX_PATH];
 110
 111     if (!name) return CreateWindowStationW( NULL, flags, access, sa );
 112
 113     if (!MultiByteToWideChar( CP_ACP, 0, name, -1, buffer, MAX_PATH ))
 114     {
 115         SetLastError( ERROR_FILENAME_EXCED_RANGE );
 116         return 0;
 117     }
 118     return CreateWindowStationW( buffer, flags, access, sa );
 119 }
 120
 121
 122 /***********************************************************************
 123  *              CreateWindowStationW  (USER32.@)
 124  */
 125 HWINSTA WINAPI CreateWindowStationW( LPCWSTR name, DWORD flags, ACCESS_MASK access,
 126                                      LPSECURITY_ATTRIBUTES sa )
 127 {
 128     OBJECT_ATTRIBUTES attr;
 129     UNICODE_STRING str;
 130
 131     RtlInitUnicodeString( &str, name );
 132     if (!str.Length) RtlInitUnicodeString( &str, get_winstation_default_name() );
 133
 134     InitializeObjectAttributes( &attr, &str, OBJ_CASE_INSENSITIVE,
 135                                 get_winstations_dir_handle(), sa );
 136     if (!(flags & CWF_CREATE_ONLY)) attr.Attributes |= OBJ_OPENIF;
 137     if (sa && sa->bInheritHandle) attr.Attributes |= OBJ_INHERIT;
 138
 139     return NtUserCreateWindowStation( &attr, access, 0, 0, 0, 0, 0 );
 140 }
 141
 142
 143 /******************************************************************************
 144  *              OpenWindowStationA  (USER32.@)
 145  */
 146 HWINSTA WINAPI OpenWindowStationA( LPCSTR name, BOOL inherit, ACCESS_MASK access )
 147 {
 148     WCHAR buffer[MAX_PATH];
 149
 150     if (!name) return OpenWindowStationW( NULL, inherit, access );
 151
 152     if (!MultiByteToWideChar( CP_ACP, 0, name, -1, buffer, MAX_PATH ))
 153     {
 154         SetLastError( ERROR_FILENAME_EXCED_RANGE );
 155         return 0;
 156     }
 157     return OpenWindowStationW( buffer, inherit, access );
 158 }
 159
 160
 161 /******************************************************************************
 162  *              OpenWindowStationW  (USER32.@)
 163  */
 164 HWINSTA WINAPI OpenWindowStationW( LPCWSTR name, BOOL inherit, ACCESS_MASK access )
 165 {
 166     OBJECT_ATTRIBUTES attr;
 167     UNICODE_STRING str;
 168
 169     RtlInitUnicodeString( &str, name );
 170     if (!str.Length) RtlInitUnicodeString( &str, get_winstation_default_name() );
 171
 172     InitializeObjectAttributes( &attr, &str, OBJ_CASE_INSENSITIVE,
 173                                 get_winstations_dir_handle(), NULL );
 174     if (inherit) attr.Attributes |= OBJ_INHERIT;
 175
 176     return NtUserOpenWindowStation( &attr, access );
 177 }
 178
 179
 180 /******************************************************************************
 181  *              EnumWindowStationsA  (USER32.@)
 182  */
 183 BOOL WINAPI EnumWindowStationsA( WINSTAENUMPROCA func, LPARAM lparam )
 184 {
 185     struct enum_proc_lparam data;
 186     data.func   = func;
 187     data.lparam = lparam;
 188     return EnumWindowStationsW( enum_names_WtoA, (LPARAM)&data );
 189 }
 190
 191
 192 /******************************************************************************
 193  *              EnumWindowStationsW  (USER32.@)
 194  */
 195 BOOL WINAPI EnumWindowStationsW( WINSTAENUMPROCW func, LPARAM lparam )
 196 {
 197     unsigned int index = 0;
 198     WCHAR name[MAX_PATH];
 199     BOOL ret = TRUE;
 200     NTSTATUS status;
 201
 202     while (ret)
 203     {
 204         SERVER_START_REQ( enum_winstation )
 205         {
 206             req->index = index;
 207             wine_server_set_reply( req, name, sizeof(name) - sizeof(WCHAR) );
 208             status = wine_server_call( req );
 209             name[wine_server_reply_size(reply)/sizeof(WCHAR)] = 0;
 210             index = reply->next;
 211         }
 212         SERVER_END_REQ;
 213         if (status == STATUS_NO_MORE_ENTRIES)
 214             break;
 215         if (status)
 216         {
 217             SetLastError( RtlNtStatusToDosError( status ) );
 218             return FALSE;
 219         }
 220         ret = func( name, lparam );
 221     }
 222     return ret;
 223 }
 224
 225
 226 /***********************************************************************
 227  *              CreateDesktopA   (USER32.@)
 228  */
 229 HDESK WINAPI CreateDesktopA( LPCSTR name, LPCSTR device, LPDEVMODEA devmode,
 230                              DWORD flags, ACCESS_MASK access, LPSECURITY_ATTRIBUTES sa )
 231 {
 232     WCHAR buffer[MAX_PATH];
 233
 234     if (device || devmode)
 235     {
 236         SetLastError( ERROR_INVALID_PARAMETER );
 237         return 0;
 238     }
 239     if (!name) return CreateDesktopW( NULL, NULL, NULL, flags, access, sa );
 240
 241     if (!MultiByteToWideChar( CP_ACP, 0, name, -1, buffer, MAX_PATH ))
 242     {
 243         SetLastError( ERROR_FILENAME_EXCED_RANGE );
 244         return 0;
 245     }
 246     return CreateDesktopW( buffer, NULL, NULL, flags, access, sa );
 247 }
 248
 249
 250 /***********************************************************************
 251  *              CreateDesktopW   (USER32.@)
 252  */
 253 HDESK WINAPI CreateDesktopW( LPCWSTR name, LPCWSTR device, LPDEVMODEW devmode,
 254                              DWORD flags, ACCESS_MASK access, LPSECURITY_ATTRIBUTES sa )
 255 {
 256     OBJECT_ATTRIBUTES attr;
 257     UNICODE_STRING str;
 258
 259     if (device || devmode)
 260     {
 261         SetLastError( ERROR_INVALID_PARAMETER );
 262         return 0;
 263     }
 264
 265     RtlInitUnicodeString( &str, name );
 266     InitializeObjectAttributes( &attr, &str, OBJ_CASE_INSENSITIVE | OBJ_OPENIF,
 267                                 get_winstations_dir_handle(), NULL );
 268     if (sa && sa->bInheritHandle) attr.Attributes |= OBJ_INHERIT;
 269     return NtUserCreateDesktopEx( &attr, NULL, devmode, flags, access, 0 );
 270 }
 271
 272
 273 /******************************************************************************
 274  *              OpenDesktopA   (USER32.@)
 275  */
 276 HDESK WINAPI OpenDesktopA( LPCSTR name, DWORD flags, BOOL inherit, ACCESS_MASK access )
 277 {
 278     WCHAR buffer[MAX_PATH];
 279
 280     if (!name) return OpenDesktopW( NULL, flags, inherit, access );
 281
 282     if (!MultiByteToWideChar( CP_ACP, 0, name, -1, buffer, MAX_PATH ))
 283     {
 284         SetLastError( ERROR_FILENAME_EXCED_RANGE );
 285         return 0;
 286     }
 287     return OpenDesktopW( buffer, flags, inherit, access );
 288 }
 289
 290
 291 HDESK open_winstation_desktop( HWINSTA hwinsta, LPCWSTR name, DWORD flags, BOOL inherit, ACCESS_MASK access )
 292 {
 293     OBJECT_ATTRIBUTES attr;
 294     UNICODE_STRING str;
 295
 296     RtlInitUnicodeString( &str, name );
 297     InitializeObjectAttributes( &attr, &str, OBJ_CASE_INSENSITIVE, hwinsta, NULL );
 298     if (inherit) attr.Attributes |= OBJ_INHERIT;
 299     return NtUserOpenDesktop( &attr, flags, access );
 300 }
 301
 302
 303 /******************************************************************************
 304  *              OpenDesktopW   (USER32.@)
 305  */
 306 HDESK WINAPI OpenDesktopW( LPCWSTR name, DWORD flags, BOOL inherit, ACCESS_MASK access )
 307 {
 308     return open_winstation_desktop( NULL, name, flags, inherit, access );
 309 }
 310
 311
 312 /******************************************************************************
 313  *              EnumDesktopsA   (USER32.@)
 314  */
 315 BOOL WINAPI EnumDesktopsA( HWINSTA winsta, DESKTOPENUMPROCA func, LPARAM lparam )
 316 {
 317     struct enum_proc_lparam data;
 318     data.func   = func;
 319     data.lparam = lparam;
 320     return EnumDesktopsW( winsta, enum_names_WtoA, (LPARAM)&data );
 321 }
 322
 323
 324 /******************************************************************************
 325  *              EnumDesktopsW   (USER32.@)
 326  */
 327 BOOL WINAPI EnumDesktopsW( HWINSTA winsta, DESKTOPENUMPROCW func, LPARAM lparam )
 328 {
 329     unsigned int index = 0;
 330     WCHAR name[MAX_PATH];
 331     BOOL ret = TRUE;
 332     NTSTATUS status;
 333
 334     if (!winsta)
 335         winsta = NtUserGetProcessWindowStation();
 336
 337     while (ret)
 338     {
 339         SERVER_START_REQ( enum_desktop )
 340         {
 341             req->winstation = wine_server_obj_handle( winsta );
 342             req->index      = index;
 343             wine_server_set_reply( req, name, sizeof(name) - sizeof(WCHAR) );
 344             status = wine_server_call( req );
 345             name[wine_server_reply_size(reply)/sizeof(WCHAR)] = 0;
 346             index = reply->next;
 347         }
 348         SERVER_END_REQ;
 349         if (status == STATUS_NO_MORE_ENTRIES)
 350             break;
 351         if (status)
 352         {
 353             SetLastError( RtlNtStatusToDosError( status ) );
 354             return FALSE;
 355         }
 356         ret = func(name, lparam);
 357     }
 358     return ret;
 359 }
 360
 361
 362 /***********************************************************************
 363  *              GetUserObjectInformationA   (USER32.@)
 364  */
 365 BOOL WINAPI GetUserObjectInformationA( HANDLE handle, INT index, LPVOID info, DWORD len, LPDWORD needed )
 366 {
 367     /* check for information types returning strings */
 368     if (index == UOI_TYPE || index == UOI_NAME)
 369     {
 370         WCHAR buffer[MAX_PATH];
 371         DWORD lenA, lenW;
 372
 373         if (!NtUserGetObjectInformation( handle, index, buffer, sizeof(buffer), &lenW )) return FALSE;
 374         lenA = WideCharToMultiByte( CP_ACP, 0, buffer, -1, NULL, 0, NULL, NULL );
 375         if (needed) *needed = lenA;
 376         if (lenA > len)
 377         {
 378             /* If the buffer length supplied by the caller is insufficient, Windows returns a
 379                'needed' length based upon the Unicode byte length, so we should do similarly. */
 380             if (needed) *needed = lenW;
 381
 382             SetLastError( ERROR_INSUFFICIENT_BUFFER );
 383             return FALSE;
 384         }
 385         if (info) WideCharToMultiByte( CP_ACP, 0, buffer, -1, info, len, NULL, NULL );
 386         return TRUE;
 387     }
 388     return NtUserGetObjectInformation( handle, index, info, len, needed );
 389 }
 390
 391
 392 /******************************************************************************
 393  *              SetUserObjectInformationA   (USER32.@)
 394  */
 395 BOOL WINAPI SetUserObjectInformationA( HANDLE handle, INT index, LPVOID info, DWORD len )
 396 {
 397     return NtUserSetObjectInformation( handle, index, info, len );
 398 }
 399
 400
 401 /***********************************************************************
 402  *              GetUserObjectSecurity   (USER32.@)
 403  */
 404 BOOL WINAPI GetUserObjectSecurity( HANDLE handle, PSECURITY_INFORMATION info,
 405                                    PSECURITY_DESCRIPTOR sid, DWORD len, LPDWORD needed )
 406 {
 407     FIXME( "(%p %p %p len=%ld %p),stub!\n", handle, info, sid, len, needed );
 408     if (needed)
 409         *needed = sizeof(SECURITY_DESCRIPTOR);
 410     if (len < sizeof(SECURITY_DESCRIPTOR))
 411     {
 412         SetLastError( ERROR_INSUFFICIENT_BUFFER );
 413         return FALSE;
 414     }
 415     return InitializeSecurityDescriptor(sid, SECURITY_DESCRIPTOR_REVISION);
 416 }
 417
 418 /***********************************************************************
 419  *              SetUserObjectSecurity   (USER32.@)
 420  */
 421 BOOL WINAPI SetUserObjectSecurity( HANDLE handle, PSECURITY_INFORMATION info,
 422                                    PSECURITY_DESCRIPTOR sid )
 423 {
 424     FIXME( "(%p,%p,%p),stub!\n", handle, info, sid );
 425     return TRUE;
 426 }