?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
kernel32/tests: Fix 32-bit thread test failures on NT 5.x 64bit.
[wine/multimedia.git] / dlls / kernel32 / process.c
blob3a88aaa4a285778116abcb48c84af92b6f272a07
1 /*
2 * Win32 processes
3 *
4 * Copyright 1996, 1998 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 "config.h"
22 #include "wine/port.h"
23
24 #include <assert.h>
25 #include <ctype.h>
26 #include <errno.h>
27 #include <signal.h>
28 #include <stdarg.h>
29 #include <stdio.h>
30 #include <time.h>
31 #ifdef HAVE_SYS_TIME_H
32 # include <sys/time.h>
33 #endif
34 #ifdef HAVE_SYS_IOCTL_H
35 #include <sys/ioctl.h>
36 #endif
37 #ifdef HAVE_SYS_SOCKET_H
38 #include <sys/socket.h>
39 #endif
40 #ifdef HAVE_SYS_PRCTL_H
41 # include <sys/prctl.h>
42 #endif
43 #include <sys/types.h>
44 #ifdef HAVE_SYS_WAIT_H
45 # include <sys/wait.h>
46 #endif
47 #ifdef HAVE_UNISTD_H
48 # include <unistd.h>
49 #endif
50 #ifdef __APPLE__
51 #include <CoreFoundation/CoreFoundation.h>
52 #include <pthread.h>
53 #endif
54
55 #include "ntstatus.h"
56 #define WIN32_NO_STATUS
57 #include "winternl.h"
58 #include "kernel_private.h"
59 #include "psapi.h"
60 #include "wine/library.h"
61 #include "wine/server.h"
62 #include "wine/unicode.h"
63 #include "wine/debug.h"
64
65 WINE_DEFAULT_DEBUG_CHANNEL(process);
66 WINE_DECLARE_DEBUG_CHANNEL(file);
67 WINE_DECLARE_DEBUG_CHANNEL(relay);
68
69 #ifdef __APPLE__
70 extern char **__wine_get_main_environment(void);
71 #else
72 extern char **__wine_main_environ;
73 static char **__wine_get_main_environment(void) { return __wine_main_environ; }
74 #endif
75
76 typedef struct
77 {
78 LPSTR lpEnvAddress;
79 LPSTR lpCmdLine;
80 LPSTR lpCmdShow;
81 DWORD dwReserved;
82 } LOADPARMS32;
83
84 static DWORD shutdown_flags = 0;
85 static DWORD shutdown_priority = 0x280;
86 static BOOL is_wow64;
87 static const int is_win64 = (sizeof(void *) > sizeof(int));
88
89 HMODULE kernel32_handle = 0;
90
91 const WCHAR *DIR_Windows = NULL;
92 const WCHAR *DIR_System = NULL;
93 const WCHAR *DIR_SysWow64 = NULL;
94
95 /* Process flags */
96 #define PDB32_DEBUGGED 0x0001 /* Process is being debugged */
97 #define PDB32_WIN16_PROC 0x0008 /* Win16 process */
98 #define PDB32_DOS_PROC 0x0010 /* Dos process */
99 #define PDB32_CONSOLE_PROC 0x0020 /* Console process */
100 #define PDB32_FILE_APIS_OEM 0x0040 /* File APIs are OEM */
101 #define PDB32_WIN32S_PROC 0x8000 /* Win32s process */
102
103 static const WCHAR exeW[] = {'.','e','x','e',0};
104 static const WCHAR comW[] = {'.','c','o','m',0};
105 static const WCHAR batW[] = {'.','b','a','t',0};
106 static const WCHAR cmdW[] = {'.','c','m','d',0};
107 static const WCHAR pifW[] = {'.','p','i','f',0};
108 static const WCHAR winevdmW[] = {'w','i','n','e','v','d','m','.','e','x','e',0};
109
110 static void exec_process( LPCWSTR name );
111
112 extern void SHELL_LoadRegistry(void);
113
114
115 /***********************************************************************
116 * contains_path
117 */
118 static inline int contains_path( LPCWSTR name )
119 {
120 return ((*name && (name[1] == ':')) || strchrW(name, '/') || strchrW(name, '\\'));
121 }
122
123
124 /***********************************************************************
125 * is_special_env_var
126 *
127 * Check if an environment variable needs to be handled specially when
128 * passed through the Unix environment (i.e. prefixed with "WINE").
129 */
130 static inline int is_special_env_var( const char *var )
131 {
132 return (!strncmp( var, "PATH=", sizeof("PATH=")-1 ) ||
133 !strncmp( var, "PWD=", sizeof("PWD=")-1 ) ||
134 !strncmp( var, "HOME=", sizeof("HOME=")-1 ) ||
135 !strncmp( var, "TEMP=", sizeof("TEMP=")-1 ) ||
136 !strncmp( var, "TMP=", sizeof("TMP=")-1 ));
137 }
138
139
140 /***********************************************************************
141 * is_path_prefix
142 */
143 static inline unsigned int is_path_prefix( const WCHAR *prefix, const WCHAR *filename )
144 {
145 unsigned int len = strlenW( prefix );
146
147 if (strncmpiW( filename, prefix, len ) || filename[len] != '\\') return 0;
148 while (filename[len] == '\\') len++;
149 return len;
150 }
151
152
153 /***************************************************************************
154 * get_builtin_path
155 *
156 * Get the path of a builtin module when the native file does not exist.
157 */
158 static BOOL get_builtin_path( const WCHAR *libname, const WCHAR *ext, WCHAR *filename,
159 UINT size, struct binary_info *binary_info )
160 {
161 WCHAR *file_part;
162 UINT len;
163 void *redir_disabled = 0;
164 unsigned int flags = (sizeof(void*) > sizeof(int) ? BINARY_FLAG_64BIT : 0);
165
166 /* builtin names cannot be empty or contain spaces */
167 if (!libname[0] || strchrW( libname, ' ' ) || strchrW( libname, '\t' )) return FALSE;
168
169 if (is_wow64 && Wow64DisableWow64FsRedirection( &redir_disabled ))
170 Wow64RevertWow64FsRedirection( redir_disabled );
171
172 if (contains_path( libname ))
173 {
174 if (RtlGetFullPathName_U( libname, size * sizeof(WCHAR),
175 filename, &file_part ) > size * sizeof(WCHAR))
176 return FALSE; /* too long */
177
178 if ((len = is_path_prefix( DIR_System, filename )))
179 {
180 if (is_wow64 && redir_disabled) flags = BINARY_FLAG_64BIT;
181 }
182 else if (DIR_SysWow64 && (len = is_path_prefix( DIR_SysWow64, filename )))
183 {
184 flags = 0;
185 }
186 else return FALSE;
187
188 if (filename + len != file_part) return FALSE;
189 }
190 else
191 {
192 len = strlenW( DIR_System );
193 if (strlenW(libname) + len + 2 >= size) return FALSE; /* too long */
194 memcpy( filename, DIR_System, len * sizeof(WCHAR) );
195 file_part = filename + len;
196 if (file_part > filename && file_part[-1] != '\\') *file_part++ = '\\';
197 strcpyW( file_part, libname );
198 if (is_wow64 && redir_disabled) flags = BINARY_FLAG_64BIT;
199 }
200 if (ext && !strchrW( file_part, '.' ))
201 {
202 if (file_part + strlenW(file_part) + strlenW(ext) + 1 > filename + size)
203 return FALSE; /* too long */
204 strcatW( file_part, ext );
205 }
206 binary_info->type = BINARY_UNIX_LIB;
207 binary_info->flags = flags;
208 binary_info->res_start = NULL;
209 binary_info->res_end = NULL;
210 return TRUE;
211 }
212
213
214 /***********************************************************************
215 * open_exe_file
216 *
217 * Open a specific exe file, taking load order into account.
218 * Returns the file handle or 0 for a builtin exe.
219 */
220 static HANDLE open_exe_file( const WCHAR *name, struct binary_info *binary_info )
221 {
222 HANDLE handle;
223
224 TRACE("looking for %s\n", debugstr_w(name) );
225
226 if ((handle = CreateFileW( name, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_DELETE,
227 NULL, OPEN_EXISTING, 0, 0 )) == INVALID_HANDLE_VALUE)
228 {
229 WCHAR buffer[MAX_PATH];
230 /* file doesn't exist, check for builtin */
231 if (contains_path( name ) && get_builtin_path( name, NULL, buffer, sizeof(buffer), binary_info ))
232 handle = 0;
233 }
234 else MODULE_get_binary_info( handle, binary_info );
235
236 return handle;
237 }
238
239
240 /***********************************************************************
241 * find_exe_file
242 *
243 * Open an exe file, and return the full name and file handle.
244 * Returns FALSE if file could not be found.
245 */
246 static BOOL find_exe_file( const WCHAR *name, WCHAR *buffer, int buflen,
247 HANDLE *handle, struct binary_info *binary_info )
248 {
249 TRACE("looking for %s\n", debugstr_w(name) );
250
251 if (!SearchPathW( NULL, name, exeW, buflen, buffer, NULL ) &&
252 /* no builtin found, try native without extension in case it is a Unix app */
253 !SearchPathW( NULL, name, NULL, buflen, buffer, NULL )) return FALSE;
254
255 TRACE( "Trying native exe %s\n", debugstr_w(buffer) );
256 if ((*handle = CreateFileW( buffer, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_DELETE,
257 NULL, OPEN_EXISTING, 0, 0 )) != INVALID_HANDLE_VALUE)
258 {
259 MODULE_get_binary_info( *handle, binary_info );
260 return TRUE;
261 }
262 return FALSE;
263 }
264
265
266 /***********************************************************************
267 * build_initial_environment
268 *
269 * Build the Win32 environment from the Unix environment
270 */
271 static BOOL build_initial_environment(void)
272 {
273 SIZE_T size = 1;
274 char **e;
275 WCHAR *p, *endptr;
276 void *ptr;
277 char **env = __wine_get_main_environment();
278
279 /* Compute the total size of the Unix environment */
280 for (e = env; *e; e++)
281 {
282 if (is_special_env_var( *e )) continue;
283 size += MultiByteToWideChar( CP_UNIXCP, 0, *e, -1, NULL, 0 );
284 }
285 size *= sizeof(WCHAR);
286
287 /* Now allocate the environment */
288 ptr = NULL;
289 if (NtAllocateVirtualMemory(NtCurrentProcess(), &ptr, 0, &size,
290 MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE) != STATUS_SUCCESS)
291 return FALSE;
292
293 NtCurrentTeb()->Peb->ProcessParameters->Environment = p = ptr;
294 endptr = p + size / sizeof(WCHAR);
295
296 /* And fill it with the Unix environment */
297 for (e = env; *e; e++)
298 {
299 char *str = *e;
300
301 /* skip Unix special variables and use the Wine variants instead */
302 if (!strncmp( str, "WINE", 4 ))
303 {
304 if (is_special_env_var( str + 4 )) str += 4;
305 else if (!strncmp( str, "WINEPRELOADRESERVE=", 19 )) continue; /* skip it */
306 }
307 else if (is_special_env_var( str )) continue; /* skip it */
308
309 MultiByteToWideChar( CP_UNIXCP, 0, str, -1, p, endptr - p );
310 p += strlenW(p) + 1;
311 }
312 *p = 0;
313 return TRUE;
314 }
315
316
317 /***********************************************************************
318 * set_registry_variables
319 *
320 * Set environment variables by enumerating the values of a key;
321 * helper for set_registry_environment().
322 * Note that Windows happily truncates the value if it's too big.
323 */
324 static void set_registry_variables( HANDLE hkey, ULONG type )
325 {
326 static const WCHAR pathW[] = {'P','A','T','H'};
327 static const WCHAR sep[] = {';',0};
328 UNICODE_STRING env_name, env_value;
329 NTSTATUS status;
330 DWORD size;
331 int index;
332 char buffer[1024*sizeof(WCHAR) + sizeof(KEY_VALUE_FULL_INFORMATION)];
333 WCHAR tmpbuf[1024];
334 UNICODE_STRING tmp;
335 KEY_VALUE_FULL_INFORMATION *info = (KEY_VALUE_FULL_INFORMATION *)buffer;
336
337 tmp.Buffer = tmpbuf;
338 tmp.MaximumLength = sizeof(tmpbuf);
339
340 for (index = 0; ; index++)
341 {
342 status = NtEnumerateValueKey( hkey, index, KeyValueFullInformation,
343 buffer, sizeof(buffer), &size );
344 if (status != STATUS_SUCCESS && status != STATUS_BUFFER_OVERFLOW)
345 break;
346 if (info->Type != type)
347 continue;
348 env_name.Buffer = info->Name;
349 env_name.Length = env_name.MaximumLength = info->NameLength;
350 env_value.Buffer = (WCHAR *)(buffer + info->DataOffset);
351 env_value.Length = info->DataLength;
352 env_value.MaximumLength = sizeof(buffer) - info->DataOffset;
353 if (env_value.Length && !env_value.Buffer[env_value.Length/sizeof(WCHAR)-1])
354 env_value.Length -= sizeof(WCHAR); /* don't count terminating null if any */
355 if (!env_value.Length) continue;
356 if (info->Type == REG_EXPAND_SZ)
357 {
358 status = RtlExpandEnvironmentStrings_U( NULL, &env_value, &tmp, NULL );
359 if (status != STATUS_SUCCESS && status != STATUS_BUFFER_OVERFLOW) continue;
360 RtlCopyUnicodeString( &env_value, &tmp );
361 }
362 /* PATH is magic */
363 if (env_name.Length == sizeof(pathW) &&
364 !memicmpW( env_name.Buffer, pathW, sizeof(pathW)/sizeof(WCHAR) ) &&
365 !RtlQueryEnvironmentVariable_U( NULL, &env_name, &tmp ))
366 {
367 RtlAppendUnicodeToString( &tmp, sep );
368 if (RtlAppendUnicodeStringToString( &tmp, &env_value )) continue;
369 RtlCopyUnicodeString( &env_value, &tmp );
370 }
371 RtlSetEnvironmentVariable( NULL, &env_name, &env_value );
372 }
373 }
374
375
376 /***********************************************************************
377 * set_registry_environment
378 *
379 * Set the environment variables specified in the registry.
380 *
381 * Note: Windows handles REG_SZ and REG_EXPAND_SZ in one pass with the
382 * consequence that REG_EXPAND_SZ cannot be used reliably as it depends
383 * on the order in which the variables are processed. But on Windows it
384 * does not really matter since they only use %SystemDrive% and
385 * %SystemRoot% which are predefined. But Wine defines these in the
386 * registry, so we need two passes.
387 */
388 static BOOL set_registry_environment( BOOL volatile_only )
389 {
390 static const WCHAR env_keyW[] = {'M','a','c','h','i','n','e','\\',
391 'S','y','s','t','e','m','\\',
392 'C','u','r','r','e','n','t','C','o','n','t','r','o','l','S','e','t','\\',
393 'C','o','n','t','r','o','l','\\',
394 'S','e','s','s','i','o','n',' ','M','a','n','a','g','e','r','\\',
395 'E','n','v','i','r','o','n','m','e','n','t',0};
396 static const WCHAR envW[] = {'E','n','v','i','r','o','n','m','e','n','t',0};
397 static const WCHAR volatile_envW[] = {'V','o','l','a','t','i','l','e',' ','E','n','v','i','r','o','n','m','e','n','t',0};
398
399 OBJECT_ATTRIBUTES attr;
400 UNICODE_STRING nameW;
401 HANDLE hkey;
402 BOOL ret = FALSE;
403
404 attr.Length = sizeof(attr);
405 attr.RootDirectory = 0;
406 attr.ObjectName = &nameW;
407 attr.Attributes = 0;
408 attr.SecurityDescriptor = NULL;
409 attr.SecurityQualityOfService = NULL;
410
411 /* first the system environment variables */
412 RtlInitUnicodeString( &nameW, env_keyW );
413 if (!volatile_only && NtOpenKey( &hkey, KEY_READ, &attr ) == STATUS_SUCCESS)
414 {
415 set_registry_variables( hkey, REG_SZ );
416 set_registry_variables( hkey, REG_EXPAND_SZ );
417 NtClose( hkey );
418 ret = TRUE;
419 }
420
421 /* then the ones for the current user */
422 if (RtlOpenCurrentUser( KEY_READ, &attr.RootDirectory ) != STATUS_SUCCESS) return ret;
423 RtlInitUnicodeString( &nameW, envW );
424 if (!volatile_only && NtOpenKey( &hkey, KEY_READ, &attr ) == STATUS_SUCCESS)
425 {
426 set_registry_variables( hkey, REG_SZ );
427 set_registry_variables( hkey, REG_EXPAND_SZ );
428 NtClose( hkey );
429 }
430
431 RtlInitUnicodeString( &nameW, volatile_envW );
432 if (NtOpenKey( &hkey, KEY_READ, &attr ) == STATUS_SUCCESS)
433 {
434 set_registry_variables( hkey, REG_SZ );
435 set_registry_variables( hkey, REG_EXPAND_SZ );
436 NtClose( hkey );
437 }
438
439 NtClose( attr.RootDirectory );
440 return ret;
441 }
442
443
444 /***********************************************************************
445 * get_reg_value
446 */
447 static WCHAR *get_reg_value( HKEY hkey, const WCHAR *name )
448 {
449 char buffer[1024 * sizeof(WCHAR) + sizeof(KEY_VALUE_PARTIAL_INFORMATION)];
450 KEY_VALUE_PARTIAL_INFORMATION *info = (KEY_VALUE_PARTIAL_INFORMATION *)buffer;
451 DWORD len, size = sizeof(buffer);
452 WCHAR *ret = NULL;
453 UNICODE_STRING nameW;
454
455 RtlInitUnicodeString( &nameW, name );
456 if (NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation, buffer, size, &size ))
457 return NULL;
458
459 if (size <= FIELD_OFFSET( KEY_VALUE_PARTIAL_INFORMATION, Data )) return NULL;
460 len = (size - FIELD_OFFSET( KEY_VALUE_PARTIAL_INFORMATION, Data )) / sizeof(WCHAR);
461
462 if (info->Type == REG_EXPAND_SZ)
463 {
464 UNICODE_STRING value, expanded;
465
466 value.MaximumLength = len * sizeof(WCHAR);
467 value.Buffer = (WCHAR *)info->Data;
468 if (!value.Buffer[len - 1]) len--; /* don't count terminating null if any */
469 value.Length = len * sizeof(WCHAR);
470 expanded.Length = expanded.MaximumLength = 1024 * sizeof(WCHAR);
471 if (!(expanded.Buffer = HeapAlloc( GetProcessHeap(), 0, expanded.MaximumLength ))) return NULL;
472 if (!RtlExpandEnvironmentStrings_U( NULL, &value, &expanded, NULL )) ret = expanded.Buffer;
473 else RtlFreeUnicodeString( &expanded );
474 }
475 else if (info->Type == REG_SZ)
476 {
477 if ((ret = HeapAlloc( GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR) )))
478 {
479 memcpy( ret, info->Data, len * sizeof(WCHAR) );
480 ret[len] = 0;
481 }
482 }
483 return ret;
484 }
485
486
487 /***********************************************************************
488 * set_additional_environment
489 *
490 * Set some additional environment variables not specified in the registry.
491 */
492 static void set_additional_environment(void)
493 {
494 static const WCHAR profile_keyW[] = {'M','a','c','h','i','n','e','\\',
495 'S','o','f','t','w','a','r','e','\\',
496 'M','i','c','r','o','s','o','f','t','\\',
497 'W','i','n','d','o','w','s',' ','N','T','\\',
498 'C','u','r','r','e','n','t','V','e','r','s','i','o','n','\\',
499 'P','r','o','f','i','l','e','L','i','s','t',0};
500 static const WCHAR profiles_valueW[] = {'P','r','o','f','i','l','e','s','D','i','r','e','c','t','o','r','y',0};
501 static const WCHAR all_users_valueW[] = {'A','l','l','U','s','e','r','s','P','r','o','f','i','l','e','\0'};
502 static const WCHAR allusersW[] = {'A','L','L','U','S','E','R','S','P','R','O','F','I','L','E',0};
503 OBJECT_ATTRIBUTES attr;
504 UNICODE_STRING nameW;
505 WCHAR *profile_dir = NULL, *all_users_dir = NULL;
506 HANDLE hkey;
507 DWORD len;
508
509 /* set the ALLUSERSPROFILE variables */
510
511 attr.Length = sizeof(attr);
512 attr.RootDirectory = 0;
513 attr.ObjectName = &nameW;
514 attr.Attributes = 0;
515 attr.SecurityDescriptor = NULL;
516 attr.SecurityQualityOfService = NULL;
517 RtlInitUnicodeString( &nameW, profile_keyW );
518 if (!NtOpenKey( &hkey, KEY_READ, &attr ))
519 {
520 profile_dir = get_reg_value( hkey, profiles_valueW );
521 all_users_dir = get_reg_value( hkey, all_users_valueW );
522 NtClose( hkey );
523 }
524
525 if (profile_dir && all_users_dir)
526 {
527 WCHAR *value, *p;
528
529 len = strlenW(profile_dir) + strlenW(all_users_dir) + 2;
530 value = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
531 strcpyW( value, profile_dir );
532 p = value + strlenW(value);
533 if (p > value && p[-1] != '\\') *p++ = '\\';
534 strcpyW( p, all_users_dir );
535 SetEnvironmentVariableW( allusersW, value );
536 HeapFree( GetProcessHeap(), 0, value );
537 }
538
539 HeapFree( GetProcessHeap(), 0, all_users_dir );
540 HeapFree( GetProcessHeap(), 0, profile_dir );
541 }
542
543 /***********************************************************************
544 * set_wow64_environment
545 *
546 * Set the environment variables that change across 32/64/Wow64.
547 */
548 static void set_wow64_environment(void)
549 {
550 static const WCHAR archW[] = {'P','R','O','C','E','S','S','O','R','_','A','R','C','H','I','T','E','C','T','U','R','E',0};
551 static const WCHAR arch6432W[] = {'P','R','O','C','E','S','S','O','R','_','A','R','C','H','I','T','E','W','6','4','3','2',0};
552 static const WCHAR x86W[] = {'x','8','6',0};
553 static const WCHAR versionW[] = {'M','a','c','h','i','n','e','\\',
554 'S','o','f','t','w','a','r','e','\\',
555 'M','i','c','r','o','s','o','f','t','\\',
556 'W','i','n','d','o','w','s','\\',
557 'C','u','r','r','e','n','t','V','e','r','s','i','o','n',0};
558 static const WCHAR progdirW[] = {'P','r','o','g','r','a','m','F','i','l','e','s','D','i','r',0};
559 static const WCHAR progdir86W[] = {'P','r','o','g','r','a','m','F','i','l','e','s','D','i','r',' ','(','x','8','6',')',0};
560 static const WCHAR progfilesW[] = {'P','r','o','g','r','a','m','F','i','l','e','s',0};
561 static const WCHAR progw6432W[] = {'P','r','o','g','r','a','m','W','6','4','3','2',0};
562 static const WCHAR commondirW[] = {'C','o','m','m','o','n','F','i','l','e','s','D','i','r',0};
563 static const WCHAR commondir86W[] = {'C','o','m','m','o','n','F','i','l','e','s','D','i','r',' ','(','x','8','6',')',0};
564 static const WCHAR commonfilesW[] = {'C','o','m','m','o','n','P','r','o','g','r','a','m','F','i','l','e','s',0};
565 static const WCHAR commonw6432W[] = {'C','o','m','m','o','n','P','r','o','g','r','a','m','W','6','4','3','2',0};
566
567 OBJECT_ATTRIBUTES attr;
568 UNICODE_STRING nameW;
569 WCHAR arch[64];
570 WCHAR *value;
571 HANDLE hkey;
572
573 /* set the PROCESSOR_ARCHITECTURE variable */
574
575 if (GetEnvironmentVariableW( arch6432W, arch, sizeof(arch)/sizeof(WCHAR) ))
576 {
577 if (is_win64)
578 {
579 SetEnvironmentVariableW( archW, arch );
580 SetEnvironmentVariableW( arch6432W, NULL );
581 }
582 }
583 else if (GetEnvironmentVariableW( archW, arch, sizeof(arch)/sizeof(WCHAR) ))
584 {
585 if (is_wow64)
586 {
587 SetEnvironmentVariableW( arch6432W, arch );
588 SetEnvironmentVariableW( archW, x86W );
589 }
590 }
591
592 attr.Length = sizeof(attr);
593 attr.RootDirectory = 0;
594 attr.ObjectName = &nameW;
595 attr.Attributes = 0;
596 attr.SecurityDescriptor = NULL;
597 attr.SecurityQualityOfService = NULL;
598 RtlInitUnicodeString( &nameW, versionW );
599 if (NtOpenKey( &hkey, KEY_READ | KEY_WOW64_64KEY, &attr )) return;
600
601 /* set the ProgramFiles variables */
602
603 if ((value = get_reg_value( hkey, progdirW )))
604 {
605 if (is_win64 || is_wow64) SetEnvironmentVariableW( progw6432W, value );
606 if (is_win64 || !is_wow64) SetEnvironmentVariableW( progfilesW, value );
607 HeapFree( GetProcessHeap(), 0, value );
608 }
609 if (is_wow64 && (value = get_reg_value( hkey, progdir86W )))
610 {
611 SetEnvironmentVariableW( progfilesW, value );
612 HeapFree( GetProcessHeap(), 0, value );
613 }
614
615 /* set the CommonProgramFiles variables */
616
617 if ((value = get_reg_value( hkey, commondirW )))
618 {
619 if (is_win64 || is_wow64) SetEnvironmentVariableW( commonw6432W, value );
620 if (is_win64 || !is_wow64) SetEnvironmentVariableW( commonfilesW, value );
621 HeapFree( GetProcessHeap(), 0, value );
622 }
623 if (is_wow64 && (value = get_reg_value( hkey, commondir86W )))
624 {
625 SetEnvironmentVariableW( commonfilesW, value );
626 HeapFree( GetProcessHeap(), 0, value );
627 }
628
629 NtClose( hkey );
630 }
631
632 /***********************************************************************
633 * set_library_wargv
634 *
635 * Set the Wine library Unicode argv global variables.
636 */
637 static void set_library_wargv( char **argv )
638 {
639 int argc;
640 char *q;
641 WCHAR *p;
642 WCHAR **wargv;
643 DWORD total = 0;
644
645 for (argc = 0; argv[argc]; argc++)
646 total += MultiByteToWideChar( CP_UNIXCP, 0, argv[argc], -1, NULL, 0 );
647
648 wargv = RtlAllocateHeap( GetProcessHeap(), 0,
649 total * sizeof(WCHAR) + (argc + 1) * sizeof(*wargv) );
650 p = (WCHAR *)(wargv + argc + 1);
651 for (argc = 0; argv[argc]; argc++)
652 {
653 DWORD reslen = MultiByteToWideChar( CP_UNIXCP, 0, argv[argc], -1, p, total );
654 wargv[argc] = p;
655 p += reslen;
656 total -= reslen;
657 }
658 wargv[argc] = NULL;
659
660 /* convert argv back from Unicode since it has to be in the Ansi codepage not the Unix one */
661
662 for (argc = 0; wargv[argc]; argc++)
663 total += WideCharToMultiByte( CP_ACP, 0, wargv[argc], -1, NULL, 0, NULL, NULL );
664
665 argv = RtlAllocateHeap( GetProcessHeap(), 0, total + (argc + 1) * sizeof(*argv) );
666 q = (char *)(argv + argc + 1);
667 for (argc = 0; wargv[argc]; argc++)
668 {
669 DWORD reslen = WideCharToMultiByte( CP_ACP, 0, wargv[argc], -1, q, total, NULL, NULL );
670 argv[argc] = q;
671 q += reslen;
672 total -= reslen;
673 }
674 argv[argc] = NULL;
675
676 __wine_main_argc = argc;
677 __wine_main_argv = argv;
678 __wine_main_wargv = wargv;
679 }
680
681
682 /***********************************************************************
683 * update_library_argv0
684 *
685 * Update the argv[0] global variable with the binary we have found.
686 */
687 static void update_library_argv0( const WCHAR *argv0 )
688 {
689 DWORD len = strlenW( argv0 );
690
691 if (len > strlenW( __wine_main_wargv[0] ))
692 {
693 __wine_main_wargv[0] = RtlAllocateHeap( GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR) );
694 }
695 strcpyW( __wine_main_wargv[0], argv0 );
696
697 len = WideCharToMultiByte( CP_ACP, 0, argv0, -1, NULL, 0, NULL, NULL );
698 if (len > strlen( __wine_main_argv[0] ) + 1)
699 {
700 __wine_main_argv[0] = RtlAllocateHeap( GetProcessHeap(), 0, len );
701 }
702 WideCharToMultiByte( CP_ACP, 0, argv0, -1, __wine_main_argv[0], len, NULL, NULL );
703 }
704
705
706 /***********************************************************************
707 * build_command_line
708 *
709 * Build the command line of a process from the argv array.
710 *
711 * Note that it does NOT necessarily include the file name.
712 * Sometimes we don't even have any command line options at all.
713 *
714 * We must quote and escape characters so that the argv array can be rebuilt
715 * from the command line:
716 * - spaces and tabs must be quoted
717 * 'a b' -> '"a b"'
718 * - quotes must be escaped
719 * '"' -> '\"'
720 * - if '\'s are followed by a '"', they must be doubled and followed by '\"',
721 * resulting in an odd number of '\' followed by a '"'
722 * '\"' -> '\\\"'
723 * '\\"' -> '\\\\\"'
724 * - '\'s that are not followed by a '"' can be left as is
725 * 'a\b' == 'a\b'
726 * 'a\\b' == 'a\\b'
727 */
728 static BOOL build_command_line( WCHAR **argv )
729 {
730 int len;
731 WCHAR **arg;
732 LPWSTR p;
733 RTL_USER_PROCESS_PARAMETERS* rupp = NtCurrentTeb()->Peb->ProcessParameters;
734
735 if (rupp->CommandLine.Buffer) return TRUE; /* already got it from the server */
736
737 len = 0;
738 for (arg = argv; *arg; arg++)
739 {
740 int has_space,bcount;
741 WCHAR* a;
742
743 has_space=0;
744 bcount=0;
745 a=*arg;
746 if( !*a ) has_space=1;
747 while (*a!='\0') {
748 if (*a=='\\') {
749 bcount++;
750 } else {
751 if (*a==' ' || *a=='\t') {
752 has_space=1;
753 } else if (*a=='"') {
754 /* doubling of '\' preceding a '"',
755 * plus escaping of said '"'
756 */
757 len+=2*bcount+1;
758 }
759 bcount=0;
760 }
761 a++;
762 }
763 len+=(a-*arg)+1 /* for the separating space */;
764 if (has_space)
765 len+=2; /* for the quotes */
766 }
767
768 if (!(rupp->CommandLine.Buffer = RtlAllocateHeap( GetProcessHeap(), 0, len * sizeof(WCHAR))))
769 return FALSE;
770
771 p = rupp->CommandLine.Buffer;
772 rupp->CommandLine.Length = (len - 1) * sizeof(WCHAR);
773 rupp->CommandLine.MaximumLength = len * sizeof(WCHAR);
774 for (arg = argv; *arg; arg++)
775 {
776 int has_space,has_quote;
777 WCHAR* a;
778
779 /* Check for quotes and spaces in this argument */
780 has_space=has_quote=0;
781 a=*arg;
782 if( !*a ) has_space=1;
783 while (*a!='\0') {
784 if (*a==' ' || *a=='\t') {
785 has_space=1;
786 if (has_quote)
787 break;
788 } else if (*a=='"') {
789 has_quote=1;
790 if (has_space)
791 break;
792 }
793 a++;
794 }
795
796 /* Now transfer it to the command line */
797 if (has_space)
798 *p++='"';
799 if (has_quote) {
800 int bcount;
801
802 bcount=0;
803 a=*arg;
804 while (*a!='\0') {
805 if (*a=='\\') {
806 *p++=*a;
807 bcount++;
808 } else {
809 if (*a=='"') {
810 int i;
811
812 /* Double all the '\\' preceding this '"', plus one */
813 for (i=0;i<=bcount;i++)
814 *p++='\\';
815 *p++='"';
816 } else {
817 *p++=*a;
818 }
819 bcount=0;
820 }
821 a++;
822 }
823 } else {
824 WCHAR* x = *arg;
825 while ((*p=*x++)) p++;
826 }
827 if (has_space)
828 *p++='"';
829 *p++=' ';
830 }
831 if (p > rupp->CommandLine.Buffer)
832 p--; /* remove last space */
833 *p = '\0';
834
835 return TRUE;
836 }
837
838
839 /***********************************************************************
840 * init_current_directory
841 *
842 * Initialize the current directory from the Unix cwd or the parent info.
843 */
844 static void init_current_directory( CURDIR *cur_dir )
845 {
846 UNICODE_STRING dir_str;
847 const char *pwd;
848 char *cwd;
849 int size;
850
851 /* if we received a cur dir from the parent, try this first */
852
853 if (cur_dir->DosPath.Length)
854 {
855 if (RtlSetCurrentDirectory_U( &cur_dir->DosPath ) == STATUS_SUCCESS) goto done;
856 }
857
858 /* now try to get it from the Unix cwd */
859
860 for (size = 256; ; size *= 2)
861 {
862 if (!(cwd = HeapAlloc( GetProcessHeap(), 0, size ))) break;
863 if (getcwd( cwd, size )) break;
864 HeapFree( GetProcessHeap(), 0, cwd );
865 if (errno == ERANGE) continue;
866 cwd = NULL;
867 break;
868 }
869
870 /* try to use PWD if it is valid, so that we don't resolve symlinks */
871
872 pwd = getenv( "PWD" );
873 if (cwd)
874 {
875 struct stat st1, st2;
876
877 if (!pwd || stat( pwd, &st1 ) == -1 ||
878 (!stat( cwd, &st2 ) && (st1.st_dev != st2.st_dev || st1.st_ino != st2.st_ino)))
879 pwd = cwd;
880 }
881
882 if (pwd)
883 {
884 ANSI_STRING unix_name;
885 UNICODE_STRING nt_name;
886 RtlInitAnsiString( &unix_name, pwd );
887 if (!wine_unix_to_nt_file_name( &unix_name, &nt_name ))
888 {
889 UNICODE_STRING dos_path;
890 /* skip the \??\ prefix, nt_name is 0 terminated */
891 RtlInitUnicodeString( &dos_path, nt_name.Buffer + 4 );
892 RtlSetCurrentDirectory_U( &dos_path );
893 RtlFreeUnicodeString( &nt_name );
894 }
895 }
896
897 if (!cur_dir->DosPath.Length) /* still not initialized */
898 {
899 MESSAGE("Warning: could not find DOS drive for current working directory '%s', "
900 "starting in the Windows directory.\n", cwd ? cwd : "" );
901 RtlInitUnicodeString( &dir_str, DIR_Windows );
902 RtlSetCurrentDirectory_U( &dir_str );
903 }
904 HeapFree( GetProcessHeap(), 0, cwd );
905
906 done:
907 TRACE( "starting in %s %p\n", debugstr_w( cur_dir->DosPath.Buffer ), cur_dir->Handle );
908 }
909
910
911 /***********************************************************************
912 * init_windows_dirs
913 *
914 * Initialize the windows and system directories from the environment.
915 */
916 static void init_windows_dirs(void)
917 {
918 extern void CDECL __wine_init_windows_dir( const WCHAR *windir, const WCHAR *sysdir );
919
920 static const WCHAR windirW[] = {'w','i','n','d','i','r',0};
921 static const WCHAR winsysdirW[] = {'w','i','n','s','y','s','d','i','r',0};
922 static const WCHAR default_windirW[] = {'C',':','\\','w','i','n','d','o','w','s',0};
923 static const WCHAR default_sysdirW[] = {'\\','s','y','s','t','e','m','3','2',0};
924 static const WCHAR default_syswow64W[] = {'\\','s','y','s','w','o','w','6','4',0};
925
926 DWORD len;
927 WCHAR *buffer;
928
929 if ((len = GetEnvironmentVariableW( windirW, NULL, 0 )))
930 {
931 buffer = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
932 GetEnvironmentVariableW( windirW, buffer, len );
933 DIR_Windows = buffer;
934 }
935 else DIR_Windows = default_windirW;
936
937 if ((len = GetEnvironmentVariableW( winsysdirW, NULL, 0 )))
938 {
939 buffer = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
940 GetEnvironmentVariableW( winsysdirW, buffer, len );
941 DIR_System = buffer;
942 }
943 else
944 {
945 len = strlenW( DIR_Windows );
946 buffer = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) + sizeof(default_sysdirW) );
947 memcpy( buffer, DIR_Windows, len * sizeof(WCHAR) );
948 memcpy( buffer + len, default_sysdirW, sizeof(default_sysdirW) );
949 DIR_System = buffer;
950 }
951
952 if (!CreateDirectoryW( DIR_Windows, NULL ) && GetLastError() != ERROR_ALREADY_EXISTS)
953 ERR( "directory %s could not be created, error %u\n",
954 debugstr_w(DIR_Windows), GetLastError() );
955 if (!CreateDirectoryW( DIR_System, NULL ) && GetLastError() != ERROR_ALREADY_EXISTS)
956 ERR( "directory %s could not be created, error %u\n",
957 debugstr_w(DIR_System), GetLastError() );
958
959 if (is_win64 || is_wow64) /* SysWow64 is always defined on 64-bit */
960 {
961 len = strlenW( DIR_Windows );
962 buffer = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) + sizeof(default_syswow64W) );
963 memcpy( buffer, DIR_Windows, len * sizeof(WCHAR) );
964 memcpy( buffer + len, default_syswow64W, sizeof(default_syswow64W) );
965 DIR_SysWow64 = buffer;
966 if (!CreateDirectoryW( DIR_SysWow64, NULL ) && GetLastError() != ERROR_ALREADY_EXISTS)
967 ERR( "directory %s could not be created, error %u\n",
968 debugstr_w(DIR_SysWow64), GetLastError() );
969 }
970
971 TRACE_(file)( "WindowsDir = %s\n", debugstr_w(DIR_Windows) );
972 TRACE_(file)( "SystemDir = %s\n", debugstr_w(DIR_System) );
973
974 /* set the directories in ntdll too */
975 __wine_init_windows_dir( DIR_Windows, DIR_System );
976 }
977
978
979 /***********************************************************************
980 * start_wineboot
981 *
982 * Start the wineboot process if necessary. Return the handles to wait on.
983 */
984 static void start_wineboot( HANDLE handles[2] )
985 {
986 static const WCHAR wineboot_eventW[] = {'_','_','w','i','n','e','b','o','o','t','_','e','v','e','n','t',0};
987
988 handles[1] = 0;
989 if (!(handles[0] = CreateEventW( NULL, TRUE, FALSE, wineboot_eventW )))
990 {
991 ERR( "failed to create wineboot event, expect trouble\n" );
992 return;
993 }
994 if (GetLastError() != ERROR_ALREADY_EXISTS) /* we created it */
995 {
996 static const WCHAR wineboot[] = {'\\','w','i','n','e','b','o','o','t','.','e','x','e',0};
997 static const WCHAR args[] = {' ','-','-','i','n','i','t',0};
998 STARTUPINFOW si;
999 PROCESS_INFORMATION pi;
1000 void *redir;
1001 WCHAR app[MAX_PATH];
1002 WCHAR cmdline[MAX_PATH + (sizeof(wineboot) + sizeof(args)) / sizeof(WCHAR)];
1003
1004 memset( &si, 0, sizeof(si) );
1005 si.cb = sizeof(si);
1006 si.dwFlags = STARTF_USESTDHANDLES;
1007 si.hStdInput = 0;
1008 si.hStdOutput = 0;
1009 si.hStdError = GetStdHandle( STD_ERROR_HANDLE );
1010
1011 GetSystemDirectoryW( app, MAX_PATH - sizeof(wineboot)/sizeof(WCHAR) );
1012 lstrcatW( app, wineboot );
1013
1014 Wow64DisableWow64FsRedirection( &redir );
1015 strcpyW( cmdline, app );
1016 strcatW( cmdline, args );
1017 if (CreateProcessW( app, cmdline, NULL, NULL, FALSE, DETACHED_PROCESS, NULL, NULL, &si, &pi ))
1018 {
1019 TRACE( "started wineboot pid %04x tid %04x\n", pi.dwProcessId, pi.dwThreadId );
1020 CloseHandle( pi.hThread );
1021 handles[1] = pi.hProcess;
1022 }
1023 else
1024 {
1025 ERR( "failed to start wineboot, err %u\n", GetLastError() );
1026 CloseHandle( handles[0] );
1027 handles[0] = 0;
1028 }
1029 Wow64RevertWow64FsRedirection( redir );
1030 }
1031 }
1032
1033
1034 #ifdef __i386__
1035 extern DWORD call_process_entry( PEB *peb, LPTHREAD_START_ROUTINE entry );
1036 __ASM_GLOBAL_FUNC( call_process_entry,
1037 "pushl %ebp\n\t"
1038 __ASM_CFI(".cfi_adjust_cfa_offset 4\n\t")
1039 __ASM_CFI(".cfi_rel_offset %ebp,0\n\t")
1040 "movl %esp,%ebp\n\t"
1041 __ASM_CFI(".cfi_def_cfa_register %ebp\n\t")
1042 "subl $12,%esp\n\t" /* deliberately mis-align the stack by 8, Doom 3 needs this */
1043 "pushl 8(%ebp)\n\t"
1044 "call *12(%ebp)\n\t"
1045 "leave\n\t"
1046 __ASM_CFI(".cfi_def_cfa %esp,4\n\t")
1047 __ASM_CFI(".cfi_same_value %ebp\n\t")
1048 "ret" )
1049 #else
1050 static inline DWORD call_process_entry( PEB *peb, LPTHREAD_START_ROUTINE entry )
1051 {
1052 return entry( peb );
1053 }
1054 #endif
1055
1056 /***********************************************************************
1057 * start_process
1058 *
1059 * Startup routine of a new process. Runs on the new process stack.
1060 */
1061 static DWORD WINAPI start_process( PEB *peb )
1062 {
1063 IMAGE_NT_HEADERS *nt;
1064 LPTHREAD_START_ROUTINE entry;
1065
1066 nt = RtlImageNtHeader( peb->ImageBaseAddress );
1067 entry = (LPTHREAD_START_ROUTINE)((char *)peb->ImageBaseAddress +
1068 nt->OptionalHeader.AddressOfEntryPoint);
1069
1070 if (!nt->OptionalHeader.AddressOfEntryPoint)
1071 {
1072 ERR( "%s doesn't have an entry point, it cannot be executed\n",
1073 debugstr_w(peb->ProcessParameters->ImagePathName.Buffer) );
1074 ExitThread( 1 );
1075 }
1076
1077 if (TRACE_ON(relay))
1078 DPRINTF( "%04x:Starting process %s (entryproc=%p)\n", GetCurrentThreadId(),
1079 debugstr_w(peb->ProcessParameters->ImagePathName.Buffer), entry );
1080
1081 SetLastError( 0 ); /* clear error code */
1082 if (peb->BeingDebugged) DbgBreakPoint();
1083 return call_process_entry( peb, entry );
1084 }
1085
1086
1087 /***********************************************************************
1088 * set_process_name
1089 *
1090 * Change the process name in the ps output.
1091 */
1092 static void set_process_name( int argc, char *argv[] )
1093 {
1094 #ifdef HAVE_SETPROCTITLE
1095 setproctitle("-%s", argv[1]);
1096 #endif
1097
1098 #ifdef HAVE_PRCTL
1099 int i, offset;
1100 char *p, *prctl_name = argv[1];
1101 char *end = argv[argc-1] + strlen(argv[argc-1]) + 1;
1102
1103 #ifndef PR_SET_NAME
1104 # define PR_SET_NAME 15
1105 #endif
1106
1107 if ((p = strrchr( prctl_name, '\\' ))) prctl_name = p + 1;
1108 if ((p = strrchr( prctl_name, '/' ))) prctl_name = p + 1;
1109
1110 if (prctl( PR_SET_NAME, prctl_name ) != -1)
1111 {
1112 offset = argv[1] - argv[0];
1113 memmove( argv[1] - offset, argv[1], end - argv[1] );
1114 memset( end - offset, 0, offset );
1115 for (i = 1; i < argc; i++) argv[i-1] = argv[i] - offset;
1116 argv[i-1] = NULL;
1117 }
1118 else
1119 #endif /* HAVE_PRCTL */
1120 {
1121 /* remove argv[0] */
1122 memmove( argv, argv + 1, argc * sizeof(argv[0]) );
1123 }
1124 }
1125
1126
1127 /***********************************************************************
1128 * __wine_kernel_init
1129 *
1130 * Wine initialisation: load and start the main exe file.
1131 */
1132 void CDECL __wine_kernel_init(void)
1133 {
1134 static const WCHAR kernel32W[] = {'k','e','r','n','e','l','3','2',0};
1135 static const WCHAR dotW[] = {'.',0};
1136
1137 WCHAR *p, main_exe_name[MAX_PATH+1];
1138 PEB *peb = NtCurrentTeb()->Peb;
1139 RTL_USER_PROCESS_PARAMETERS *params = peb->ProcessParameters;
1140 HANDLE boot_events[2];
1141 BOOL got_environment = TRUE;
1142
1143 /* Initialize everything */
1144
1145 setbuf(stdout,NULL);
1146 setbuf(stderr,NULL);
1147 kernel32_handle = GetModuleHandleW(kernel32W);
1148 IsWow64Process( GetCurrentProcess(), &is_wow64 );
1149
1150 LOCALE_Init();
1151
1152 if (!params->Environment)
1153 {
1154 /* Copy the parent environment */
1155 if (!build_initial_environment()) exit(1);
1156
1157 /* convert old configuration to new format */
1158 convert_old_config();
1159
1160 got_environment = set_registry_environment( FALSE );
1161 set_additional_environment();
1162 }
1163
1164 init_windows_dirs();
1165 init_current_directory( &params->CurrentDirectory );
1166
1167 set_process_name( __wine_main_argc, __wine_main_argv );
1168 set_library_wargv( __wine_main_argv );
1169 boot_events[0] = boot_events[1] = 0;
1170
1171 if (peb->ProcessParameters->ImagePathName.Buffer)
1172 {
1173 strcpyW( main_exe_name, peb->ProcessParameters->ImagePathName.Buffer );
1174 }
1175 else
1176 {
1177 struct binary_info binary_info;
1178
1179 if (!SearchPathW( NULL, __wine_main_wargv[0], exeW, MAX_PATH, main_exe_name, NULL ) &&
1180 !get_builtin_path( __wine_main_wargv[0], exeW, main_exe_name, MAX_PATH, &binary_info ))
1181 {
1182 MESSAGE( "wine: cannot find '%s'\n", __wine_main_argv[0] );
1183 ExitProcess( GetLastError() );
1184 }
1185 update_library_argv0( main_exe_name );
1186 if (!build_command_line( __wine_main_wargv )) goto error;
1187 start_wineboot( boot_events );
1188 }
1189
1190 /* if there's no extension, append a dot to prevent LoadLibrary from appending .dll */
1191 p = strrchrW( main_exe_name, '.' );
1192 if (!p || strchrW( p, '/' ) || strchrW( p, '\\' )) strcatW( main_exe_name, dotW );
1193
1194 TRACE( "starting process name=%s argv[0]=%s\n",
1195 debugstr_w(main_exe_name), debugstr_w(__wine_main_wargv[0]) );
1196
1197 RtlInitUnicodeString( &NtCurrentTeb()->Peb->ProcessParameters->DllPath,
1198 MODULE_get_dll_load_path(main_exe_name) );
1199
1200 if (boot_events[0])
1201 {
1202 DWORD timeout = 2 * 60 * 1000, count = 1;
1203
1204 if (boot_events[1]) count++;
1205 if (!got_environment) timeout = 5 * 60 * 1000; /* initial prefix creation can take longer */
1206 if (WaitForMultipleObjects( count, boot_events, FALSE, timeout ) == WAIT_TIMEOUT)
1207 ERR( "boot event wait timed out\n" );
1208 CloseHandle( boot_events[0] );
1209 if (boot_events[1]) CloseHandle( boot_events[1] );
1210 /* reload environment now that wineboot has run */
1211 set_registry_environment( got_environment );
1212 set_additional_environment();
1213 }
1214 set_wow64_environment();
1215
1216 if (!(peb->ImageBaseAddress = LoadLibraryExW( main_exe_name, 0, DONT_RESOLVE_DLL_REFERENCES )))
1217 {
1218 DWORD_PTR args[1];
1219 WCHAR msgW[1024];
1220 char msg[1024];
1221 DWORD error = GetLastError();
1222
1223 /* if Win16/DOS format, or unavailable address, exec a new process with the proper setup */
1224 if (error == ERROR_BAD_EXE_FORMAT ||
1225 error == ERROR_INVALID_ADDRESS ||
1226 error == ERROR_NOT_ENOUGH_MEMORY)
1227 {
1228 if (!getenv("WINEPRELOADRESERVE")) exec_process( main_exe_name );
1229 /* if we get back here, it failed */
1230 }
1231 else if (error == ERROR_MOD_NOT_FOUND)
1232 {
1233 if ((p = strrchrW( main_exe_name, '\\' ))) p++;
1234 else p = main_exe_name;
1235 if (!strcmpiW( p, winevdmW ) && __wine_main_argc > 3)
1236 {
1237 /* args 1 and 2 are --app-name full_path */
1238 MESSAGE( "wine: could not run %s: 16-bit/DOS support missing\n",
1239 debugstr_w(__wine_main_wargv[3]) );
1240 ExitProcess( ERROR_BAD_EXE_FORMAT );
1241 }
1242 MESSAGE( "wine: cannot find %s\n", debugstr_w(main_exe_name) );
1243 ExitProcess( ERROR_FILE_NOT_FOUND );
1244 }
1245 args[0] = (DWORD_PTR)main_exe_name;
1246 FormatMessageW( FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_ARGUMENT_ARRAY,
1247 NULL, error, 0, msgW, sizeof(msgW)/sizeof(WCHAR), (__ms_va_list *)args );
1248 WideCharToMultiByte( CP_UNIXCP, 0, msgW, -1, msg, sizeof(msg), NULL, NULL );
1249 MESSAGE( "wine: %s", msg );
1250 ExitProcess( error );
1251 }
1252
1253 if (!params->CurrentDirectory.Handle) chdir("/"); /* avoid locking removable devices */
1254
1255 LdrInitializeThunk( start_process, 0, 0, 0 );
1256
1257 error:
1258 ExitProcess( GetLastError() );
1259 }
1260
1261
1262 /***********************************************************************
1263 * build_argv
1264 *
1265 * Build an argv array from a command-line.
1266 * 'reserved' is the number of args to reserve before the first one.
1267 */
1268 static char **build_argv( const WCHAR *cmdlineW, int reserved )
1269 {
1270 int argc;
1271 char** argv;
1272 char *arg,*s,*d,*cmdline;
1273 int in_quotes,bcount,len;
1274
1275 len = WideCharToMultiByte( CP_UNIXCP, 0, cmdlineW, -1, NULL, 0, NULL, NULL );
1276 if (!(cmdline = HeapAlloc( GetProcessHeap(), 0, len ))) return NULL;
1277 WideCharToMultiByte( CP_UNIXCP, 0, cmdlineW, -1, cmdline, len, NULL, NULL );
1278
1279 argc=reserved+1;
1280 bcount=0;
1281 in_quotes=0;
1282 s=cmdline;
1283 while (1) {
1284 if (*s=='\0' || ((*s==' ' || *s=='\t') && !in_quotes)) {
1285 /* space */
1286 argc++;
1287 /* skip the remaining spaces */
1288 while (*s==' ' || *s=='\t') {
1289 s++;
1290 }
1291 if (*s=='\0')
1292 break;
1293 bcount=0;
1294 continue;
1295 } else if (*s=='\\') {
1296 /* '\', count them */
1297 bcount++;
1298 } else if ((*s=='"') && ((bcount & 1)==0)) {
1299 /* unescaped '"' */
1300 in_quotes=!in_quotes;
1301 bcount=0;
1302 } else {
1303 /* a regular character */
1304 bcount=0;
1305 }
1306 s++;
1307 }
1308 if (!(argv = HeapAlloc( GetProcessHeap(), 0, argc*sizeof(*argv) + len )))
1309 {
1310 HeapFree( GetProcessHeap(), 0, cmdline );
1311 return NULL;
1312 }
1313
1314 arg = d = s = (char *)(argv + argc);
1315 memcpy( d, cmdline, len );
1316 bcount=0;
1317 in_quotes=0;
1318 argc=reserved;
1319 while (*s) {
1320 if ((*s==' ' || *s=='\t') && !in_quotes) {
1321 /* Close the argument and copy it */
1322 *d=0;
1323 argv[argc++]=arg;
1324
1325 /* skip the remaining spaces */
1326 do {
1327 s++;
1328 } while (*s==' ' || *s=='\t');
1329
1330 /* Start with a new argument */
1331 arg=d=s;
1332 bcount=0;
1333 } else if (*s=='\\') {
1334 /* '\\' */
1335 *d++=*s++;
1336 bcount++;
1337 } else if (*s=='"') {
1338 /* '"' */
1339 if ((bcount & 1)==0) {
1340 /* Preceded by an even number of '\', this is half that
1341 * number of '\', plus a '"' which we discard.
1342 */
1343 d-=bcount/2;
1344 s++;
1345 in_quotes=!in_quotes;
1346 } else {
1347 /* Preceded by an odd number of '\', this is half that
1348 * number of '\' followed by a '"'
1349 */
1350 d=d-bcount/2-1;
1351 *d++='"';
1352 s++;
1353 }
1354 bcount=0;
1355 } else {
1356 /* a regular character */
1357 *d++=*s++;
1358 bcount=0;
1359 }
1360 }
1361 if (*arg) {
1362 *d='\0';
1363 argv[argc++]=arg;
1364 }
1365 argv[argc]=NULL;
1366
1367 HeapFree( GetProcessHeap(), 0, cmdline );
1368 return argv;
1369 }
1370
1371
1372 /***********************************************************************
1373 * build_envp
1374 *
1375 * Build the environment of a new child process.
1376 */
1377 static char **build_envp( const WCHAR *envW )
1378 {
1379 static const char * const unix_vars[] = { "PATH", "TEMP", "TMP", "HOME" };
1380
1381 const WCHAR *end;
1382 char **envp;
1383 char *env, *p;
1384 int count = 1, length;
1385 unsigned int i;
1386
1387 for (end = envW; *end; count++) end += strlenW(end) + 1;
1388 end++;
1389 length = WideCharToMultiByte( CP_UNIXCP, 0, envW, end - envW, NULL, 0, NULL, NULL );
1390 if (!(env = HeapAlloc( GetProcessHeap(), 0, length ))) return NULL;
1391 WideCharToMultiByte( CP_UNIXCP, 0, envW, end - envW, env, length, NULL, NULL );
1392
1393 for (p = env; *p; p += strlen(p) + 1)
1394 if (is_special_env_var( p )) length += 4; /* prefix it with "WINE" */
1395
1396 for (i = 0; i < sizeof(unix_vars)/sizeof(unix_vars[0]); i++)
1397 {
1398 if (!(p = getenv(unix_vars[i]))) continue;
1399 length += strlen(unix_vars[i]) + strlen(p) + 2;
1400 count++;
1401 }
1402
1403 if ((envp = HeapAlloc( GetProcessHeap(), 0, count * sizeof(*envp) + length )))
1404 {
1405 char **envptr = envp;
1406 char *dst = (char *)(envp + count);
1407
1408 /* some variables must not be modified, so we get them directly from the unix env */
1409 for (i = 0; i < sizeof(unix_vars)/sizeof(unix_vars[0]); i++)
1410 {
1411 if (!(p = getenv(unix_vars[i]))) continue;
1412 *envptr++ = strcpy( dst, unix_vars[i] );
1413 strcat( dst, "=" );
1414 strcat( dst, p );
1415 dst += strlen(dst) + 1;
1416 }
1417
1418 /* now put the Windows environment strings */
1419 for (p = env; *p; p += strlen(p) + 1)
1420 {
1421 if (*p == '=') continue; /* skip drive curdirs, this crashes some unix apps */
1422 if (!strncmp( p, "WINEPRELOADRESERVE=", sizeof("WINEPRELOADRESERVE=")-1 )) continue;
1423 if (!strncmp( p, "WINELOADERNOEXEC=", sizeof("WINELOADERNOEXEC=")-1 )) continue;
1424 if (!strncmp( p, "WINESERVERSOCKET=", sizeof("WINESERVERSOCKET=")-1 )) continue;
1425 if (is_special_env_var( p )) /* prefix it with "WINE" */
1426 {
1427 *envptr++ = strcpy( dst, "WINE" );
1428 strcat( dst, p );
1429 }
1430 else
1431 {
1432 *envptr++ = strcpy( dst, p );
1433 }
1434 dst += strlen(dst) + 1;
1435 }
1436 *envptr = 0;
1437 }
1438 HeapFree( GetProcessHeap(), 0, env );
1439 return envp;
1440 }
1441
1442
1443 /***********************************************************************
1444 * fork_and_exec
1445 *
1446 * Fork and exec a new Unix binary, checking for errors.
1447 */
1448 static int fork_and_exec( const char *filename, const WCHAR *cmdline, const WCHAR *env,
1449 const char *newdir, DWORD flags, STARTUPINFOW *startup )
1450 {
1451 int fd[2], stdin_fd = -1, stdout_fd = -1, stderr_fd = -1;
1452 int pid, err;
1453 char **argv, **envp;
1454
1455 if (!env) env = GetEnvironmentStringsW();
1456
1457 #ifdef HAVE_PIPE2
1458 if (pipe2( fd, O_CLOEXEC ) == -1)
1459 #endif
1460 {
1461 if (pipe(fd) == -1)
1462 {
1463 SetLastError( ERROR_TOO_MANY_OPEN_FILES );
1464 return -1;
1465 }
1466 fcntl( fd[0], F_SETFD, FD_CLOEXEC );
1467 fcntl( fd[1], F_SETFD, FD_CLOEXEC );
1468 }
1469
1470 if (!(flags & (CREATE_NEW_PROCESS_GROUP | CREATE_NEW_CONSOLE | DETACHED_PROCESS)))
1471 {
1472 HANDLE hstdin, hstdout, hstderr;
1473
1474 if (startup->dwFlags & STARTF_USESTDHANDLES)
1475 {
1476 hstdin = startup->hStdInput;
1477 hstdout = startup->hStdOutput;
1478 hstderr = startup->hStdError;
1479 }
1480 else
1481 {
1482 hstdin = GetStdHandle(STD_INPUT_HANDLE);
1483 hstdout = GetStdHandle(STD_OUTPUT_HANDLE);
1484 hstderr = GetStdHandle(STD_ERROR_HANDLE);
1485 }
1486
1487 if (is_console_handle( hstdin ))
1488 hstdin = wine_server_ptr_handle( console_handle_unmap( hstdin ));
1489 if (is_console_handle( hstdout ))
1490 hstdout = wine_server_ptr_handle( console_handle_unmap( hstdout ));
1491 if (is_console_handle( hstderr ))
1492 hstderr = wine_server_ptr_handle( console_handle_unmap( hstderr ));
1493 wine_server_handle_to_fd( hstdin, FILE_READ_DATA, &stdin_fd, NULL );
1494 wine_server_handle_to_fd( hstdout, FILE_WRITE_DATA, &stdout_fd, NULL );
1495 wine_server_handle_to_fd( hstderr, FILE_WRITE_DATA, &stderr_fd, NULL );
1496 }
1497
1498 argv = build_argv( cmdline, 0 );
1499 envp = build_envp( env );
1500
1501 if (!(pid = fork())) /* child */
1502 {
1503 if (!(pid = fork())) /* grandchild */
1504 {
1505 close( fd[0] );
1506
1507 if (flags & (CREATE_NEW_PROCESS_GROUP | CREATE_NEW_CONSOLE | DETACHED_PROCESS))
1508 {
1509 int nullfd = open( "/dev/null", O_RDWR );
1510 setsid();
1511 /* close stdin and stdout */
1512 if (nullfd != -1)
1513 {
1514 dup2( nullfd, 0 );
1515 dup2( nullfd, 1 );
1516 close( nullfd );
1517 }
1518 }
1519 else
1520 {
1521 if (stdin_fd != -1)
1522 {
1523 dup2( stdin_fd, 0 );
1524 close( stdin_fd );
1525 }
1526 if (stdout_fd != -1)
1527 {
1528 dup2( stdout_fd, 1 );
1529 close( stdout_fd );
1530 }
1531 if (stderr_fd != -1)
1532 {
1533 dup2( stderr_fd, 2 );
1534 close( stderr_fd );
1535 }
1536 }
1537
1538 /* Reset signals that we previously set to SIG_IGN */
1539 signal( SIGPIPE, SIG_DFL );
1540
1541 if (newdir) chdir(newdir);
1542
1543 if (argv && envp) execve( filename, argv, envp );
1544 }
1545
1546 if (pid <= 0) /* grandchild if exec failed or child if fork failed */
1547 {
1548 err = errno;
1549 write( fd[1], &err, sizeof(err) );
1550 _exit(1);
1551 }
1552
1553 _exit(0); /* child if fork succeeded */
1554 }
1555 HeapFree( GetProcessHeap(), 0, argv );
1556 HeapFree( GetProcessHeap(), 0, envp );
1557 if (stdin_fd != -1) close( stdin_fd );
1558 if (stdout_fd != -1) close( stdout_fd );
1559 if (stderr_fd != -1) close( stderr_fd );
1560 close( fd[1] );
1561 if (pid != -1)
1562 {
1563 /* reap child */
1564 do {
1565 err = waitpid(pid, NULL, 0);
1566 } while (err < 0 && errno == EINTR);
1567
1568 if (read( fd[0], &err, sizeof(err) ) > 0) /* exec or second fork failed */
1569 {
1570 errno = err;
1571 pid = -1;
1572 }
1573 }
1574 if (pid == -1) FILE_SetDosError();
1575 close( fd[0] );
1576 return pid;
1577 }
1578
1579
1580 static inline DWORD append_string( void **ptr, const WCHAR *str )
1581 {
1582 DWORD len = strlenW( str );
1583 memcpy( *ptr, str, len * sizeof(WCHAR) );
1584 *ptr = (WCHAR *)*ptr + len;
1585 return len * sizeof(WCHAR);
1586 }
1587
1588 /***********************************************************************
1589 * create_startup_info
1590 */
1591 static startup_info_t *create_startup_info( LPCWSTR filename, LPCWSTR cmdline,
1592 LPCWSTR cur_dir, LPWSTR env, DWORD flags,
1593 const STARTUPINFOW *startup, DWORD *info_size )
1594 {
1595 const RTL_USER_PROCESS_PARAMETERS *cur_params;
1596 const WCHAR *title;
1597 startup_info_t *info;
1598 DWORD size;
1599 void *ptr;
1600 UNICODE_STRING newdir;
1601 WCHAR imagepath[MAX_PATH];
1602 HANDLE hstdin, hstdout, hstderr;
1603
1604 if(!GetLongPathNameW( filename, imagepath, MAX_PATH ))
1605 lstrcpynW( imagepath, filename, MAX_PATH );
1606 if(!GetFullPathNameW( imagepath, MAX_PATH, imagepath, NULL ))
1607 lstrcpynW( imagepath, filename, MAX_PATH );
1608
1609 cur_params = NtCurrentTeb()->Peb->ProcessParameters;
1610
1611 newdir.Buffer = NULL;
1612 if (cur_dir)
1613 {
1614 if (RtlDosPathNameToNtPathName_U( cur_dir, &newdir, NULL, NULL ))
1615 cur_dir = newdir.Buffer + 4; /* skip \??\ prefix */
1616 else
1617 cur_dir = NULL;
1618 }
1619 if (!cur_dir)
1620 {
1621 if (NtCurrentTeb()->Tib.SubSystemTib) /* FIXME: hack */
1622 cur_dir = ((WIN16_SUBSYSTEM_TIB *)NtCurrentTeb()->Tib.SubSystemTib)->curdir.DosPath.Buffer;
1623 else
1624 cur_dir = cur_params->CurrentDirectory.DosPath.Buffer;
1625 }
1626 title = startup->lpTitle ? startup->lpTitle : imagepath;
1627
1628 size = sizeof(*info);
1629 size += strlenW( cur_dir ) * sizeof(WCHAR);
1630 size += cur_params->DllPath.Length;
1631 size += strlenW( imagepath ) * sizeof(WCHAR);
1632 size += strlenW( cmdline ) * sizeof(WCHAR);
1633 size += strlenW( title ) * sizeof(WCHAR);
1634 if (startup->lpDesktop) size += strlenW( startup->lpDesktop ) * sizeof(WCHAR);
1635 /* FIXME: shellinfo */
1636 if (startup->lpReserved2 && startup->cbReserved2) size += startup->cbReserved2;
1637 size = (size + 1) & ~1;
1638 *info_size = size;
1639
1640 if (!(info = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ))) goto done;
1641
1642 info->console_flags = cur_params->ConsoleFlags;
1643 if (flags & CREATE_NEW_PROCESS_GROUP) info->console_flags = 1;
1644 if (flags & CREATE_NEW_CONSOLE) info->console = wine_server_obj_handle(KERNEL32_CONSOLE_ALLOC);
1645
1646 if (startup->dwFlags & STARTF_USESTDHANDLES)
1647 {
1648 hstdin = startup->hStdInput;
1649 hstdout = startup->hStdOutput;
1650 hstderr = startup->hStdError;
1651 }
1652 else
1653 {
1654 hstdin = GetStdHandle( STD_INPUT_HANDLE );
1655 hstdout = GetStdHandle( STD_OUTPUT_HANDLE );
1656 hstderr = GetStdHandle( STD_ERROR_HANDLE );
1657 }
1658 info->hstdin = wine_server_obj_handle( hstdin );
1659 info->hstdout = wine_server_obj_handle( hstdout );
1660 info->hstderr = wine_server_obj_handle( hstderr );
1661 if ((flags & (CREATE_NEW_CONSOLE | DETACHED_PROCESS)) != 0)
1662 {
1663 /* this is temporary (for console handles). We have no way to control that the handle is invalid in child process otherwise */
1664 if (is_console_handle(hstdin)) info->hstdin = wine_server_obj_handle( INVALID_HANDLE_VALUE );
1665 if (is_console_handle(hstdout)) info->hstdout = wine_server_obj_handle( INVALID_HANDLE_VALUE );
1666 if (is_console_handle(hstderr)) info->hstderr = wine_server_obj_handle( INVALID_HANDLE_VALUE );
1667 }
1668 else
1669 {
1670 if (is_console_handle(hstdin)) info->hstdin = console_handle_unmap(hstdin);
1671 if (is_console_handle(hstdout)) info->hstdout = console_handle_unmap(hstdout);
1672 if (is_console_handle(hstderr)) info->hstderr = console_handle_unmap(hstderr);
1673 }
1674
1675 info->x = startup->dwX;
1676 info->y = startup->dwY;
1677 info->xsize = startup->dwXSize;
1678 info->ysize = startup->dwYSize;
1679 info->xchars = startup->dwXCountChars;
1680 info->ychars = startup->dwYCountChars;
1681 info->attribute = startup->dwFillAttribute;
1682 info->flags = startup->dwFlags;
1683 info->show = startup->wShowWindow;
1684
1685 ptr = info + 1;
1686 info->curdir_len = append_string( &ptr, cur_dir );
1687 info->dllpath_len = cur_params->DllPath.Length;
1688 memcpy( ptr, cur_params->DllPath.Buffer, cur_params->DllPath.Length );
1689 ptr = (char *)ptr + cur_params->DllPath.Length;
1690 info->imagepath_len = append_string( &ptr, imagepath );
1691 info->cmdline_len = append_string( &ptr, cmdline );
1692 info->title_len = append_string( &ptr, title );
1693 if (startup->lpDesktop) info->desktop_len = append_string( &ptr, startup->lpDesktop );
1694 if (startup->lpReserved2 && startup->cbReserved2)
1695 {
1696 info->runtime_len = startup->cbReserved2;
1697 memcpy( ptr, startup->lpReserved2, startup->cbReserved2 );
1698 }
1699
1700 done:
1701 RtlFreeUnicodeString( &newdir );
1702 return info;
1703 }
1704
1705 /***********************************************************************
1706 * get_alternate_loader
1707 *
1708 * Get the name of the alternate (32 or 64 bit) Wine loader.
1709 */
1710 static const char *get_alternate_loader( char **ret_env )
1711 {
1712 char *env;
1713 const char *loader = NULL;
1714 const char *loader_env = getenv( "WINELOADER" );
1715
1716 *ret_env = NULL;
1717
1718 if (wine_get_build_dir()) loader = is_win64 ? "loader/wine" : "server/../loader/wine64";
1719
1720 if (loader_env)
1721 {
1722 int len = strlen( loader_env );
1723 if (!is_win64)
1724 {
1725 if (!(env = HeapAlloc( GetProcessHeap(), 0, sizeof("WINELOADER=") + len + 2 ))) return NULL;
1726 strcpy( env, "WINELOADER=" );
1727 strcat( env, loader_env );
1728 strcat( env, "64" );
1729 }
1730 else
1731 {
1732 if (!(env = HeapAlloc( GetProcessHeap(), 0, sizeof("WINELOADER=") + len ))) return NULL;
1733 strcpy( env, "WINELOADER=" );
1734 strcat( env, loader_env );
1735 len += sizeof("WINELOADER=") - 1;
1736 if (!strcmp( env + len - 2, "64" )) env[len - 2] = 0;
1737 }
1738 if (!loader)
1739 {
1740 if ((loader = strrchr( env, '/' ))) loader++;
1741 else loader = env;
1742 }
1743 *ret_env = env;
1744 }
1745 if (!loader) loader = is_win64 ? "wine" : "wine64";
1746 return loader;
1747 }
1748
1749 #ifdef __APPLE__
1750 /***********************************************************************
1751 * terminate_main_thread
1752 *
1753 * On some versions of Mac OS X, the execve system call fails with
1754 * ENOTSUP if the process has multiple threads. Wine is always multi-
1755 * threaded on Mac OS X because it specifically reserves the main thread
1756 * for use by the system frameworks (see apple_main_thread() in
1757 * libs/wine/loader.c). So, when we need to exec without first forking,
1758 * we need to terminate the main thread first. We do this by installing
1759 * a custom run loop source onto the main run loop and signaling it.
1760 * The source's "perform" callback is pthread_exit and it will be
1761 * executed on the main thread, terminating it.
1762 *
1763 * Returns TRUE if there's still hope the main thread has terminated or
1764 * will soon. Return FALSE if we've given up.
1765 */
1766 static BOOL terminate_main_thread(void)
1767 {
1768 static int delayms;
1769
1770 if (!delayms)
1771 {
1772 CFRunLoopSourceContext source_context = { 0 };
1773 CFRunLoopSourceRef source;
1774
1775 source_context.perform = pthread_exit;
1776 if (!(source = CFRunLoopSourceCreate( NULL, 0, &source_context )))
1777 return FALSE;
1778
1779 CFRunLoopAddSource( CFRunLoopGetMain(), source, kCFRunLoopCommonModes );
1780 CFRunLoopSourceSignal( source );
1781 CFRunLoopWakeUp( CFRunLoopGetMain() );
1782 CFRelease( source );
1783
1784 delayms = 20;
1785 }
1786
1787 if (delayms > 1000)
1788 return FALSE;
1789
1790 usleep(delayms * 1000);
1791 delayms *= 2;
1792
1793 return TRUE;
1794 }
1795 #endif
1796
1797 /***********************************************************************
1798 * exec_loader
1799 */
1800 static pid_t exec_loader( LPCWSTR cmd_line, unsigned int flags, int socketfd,
1801 int stdin_fd, int stdout_fd, const char *unixdir, char *winedebug,
1802 const struct binary_info *binary_info, int exec_only )
1803 {
1804 pid_t pid;
1805 char *wineloader = NULL;
1806 const char *loader = NULL;
1807 char **argv;
1808
1809 argv = build_argv( cmd_line, 1 );
1810
1811 if (!is_win64 ^ !(binary_info->flags & BINARY_FLAG_64BIT))
1812 loader = get_alternate_loader( &wineloader );
1813
1814 if (exec_only || !(pid = fork())) /* child */
1815 {
1816 if (exec_only || !(pid = fork())) /* grandchild */
1817 {
1818 char preloader_reserve[64], socket_env[64];
1819
1820 if (flags & (CREATE_NEW_PROCESS_GROUP | CREATE_NEW_CONSOLE | DETACHED_PROCESS))
1821 {
1822 int fd = open( "/dev/null", O_RDWR );
1823 setsid();
1824 /* close stdin and stdout */
1825 if (fd != -1)
1826 {
1827 dup2( fd, 0 );
1828 dup2( fd, 1 );
1829 close( fd );
1830 }
1831 }
1832 else
1833 {
1834 if (stdin_fd != -1) dup2( stdin_fd, 0 );
1835 if (stdout_fd != -1) dup2( stdout_fd, 1 );
1836 }
1837
1838 if (stdin_fd != -1) close( stdin_fd );
1839 if (stdout_fd != -1) close( stdout_fd );
1840
1841 /* Reset signals that we previously set to SIG_IGN */
1842 signal( SIGPIPE, SIG_DFL );
1843
1844 sprintf( socket_env, "WINESERVERSOCKET=%u", socketfd );
1845 sprintf( preloader_reserve, "WINEPRELOADRESERVE=%lx-%lx",
1846 (unsigned long)binary_info->res_start, (unsigned long)binary_info->res_end );
1847
1848 putenv( preloader_reserve );
1849 putenv( socket_env );
1850 if (winedebug) putenv( winedebug );
1851 if (wineloader) putenv( wineloader );
1852 if (unixdir) chdir(unixdir);
1853
1854 if (argv)
1855 {
1856 do
1857 {
1858 wine_exec_wine_binary( loader, argv, getenv("WINELOADER") );
1859 }
1860 #ifdef __APPLE__
1861 while (errno == ENOTSUP && exec_only && terminate_main_thread());
1862 #else
1863 while (0);
1864 #endif
1865 }
1866 _exit(1);
1867 }
1868
1869 _exit(pid == -1);
1870 }
1871
1872 if (pid != -1)
1873 {
1874 /* reap child */
1875 pid_t wret;
1876 do {
1877 wret = waitpid(pid, NULL, 0);
1878 } while (wret < 0 && errno == EINTR);
1879 }
1880
1881 HeapFree( GetProcessHeap(), 0, wineloader );
1882 HeapFree( GetProcessHeap(), 0, argv );
1883 return pid;
1884 }
1885
1886 /***********************************************************************
1887 * create_process
1888 *
1889 * Create a new process. If hFile is a valid handle we have an exe
1890 * file, otherwise it is a Winelib app.
1891 */
1892 static BOOL create_process( HANDLE hFile, LPCWSTR filename, LPWSTR cmd_line, LPWSTR env,
1893 LPCWSTR cur_dir, LPSECURITY_ATTRIBUTES psa, LPSECURITY_ATTRIBUTES tsa,
1894 BOOL inherit, DWORD flags, LPSTARTUPINFOW startup,
1895 LPPROCESS_INFORMATION info, LPCSTR unixdir,
1896 const struct binary_info *binary_info, int exec_only )
1897 {
1898 BOOL ret, success = FALSE;
1899 HANDLE process_info;
1900 WCHAR *env_end;
1901 char *winedebug = NULL;
1902 startup_info_t *startup_info;
1903 DWORD startup_info_size;
1904 int socketfd[2], stdin_fd = -1, stdout_fd = -1;
1905 pid_t pid;
1906 int err;
1907
1908 if (!is_win64 && !is_wow64 && (binary_info->flags & BINARY_FLAG_64BIT))
1909 {
1910 ERR( "starting 64-bit process %s not supported in 32-bit wineprefix\n", debugstr_w(filename) );
1911 SetLastError( ERROR_BAD_EXE_FORMAT );
1912 return FALSE;
1913 }
1914
1915 /* create the socket for the new process */
1916
1917 if (socketpair( PF_UNIX, SOCK_STREAM, 0, socketfd ) == -1)
1918 {
1919 SetLastError( ERROR_TOO_MANY_OPEN_FILES );
1920 return FALSE;
1921 }
1922 #ifdef SO_PASSCRED
1923 else
1924 {
1925 int enable = 1;
1926 setsockopt( socketfd[0], SOL_SOCKET, SO_PASSCRED, &enable, sizeof(enable) );
1927 }
1928 #endif
1929
1930 if (exec_only) /* things are much simpler in this case */
1931 {
1932 wine_server_send_fd( socketfd[1] );
1933 close( socketfd[1] );
1934 SERVER_START_REQ( new_process )
1935 {
1936 req->create_flags = flags;
1937 req->socket_fd = socketfd[1];
1938 req->exe_file = wine_server_obj_handle( hFile );
1939 ret = !wine_server_call_err( req );
1940 }
1941 SERVER_END_REQ;
1942
1943 if (ret) exec_loader( cmd_line, flags, socketfd[0], stdin_fd, stdout_fd, unixdir,
1944 winedebug, binary_info, TRUE );
1945
1946 close( socketfd[0] );
1947 return FALSE;
1948 }
1949
1950 RtlAcquirePebLock();
1951
1952 if (!(startup_info = create_startup_info( filename, cmd_line, cur_dir, env, flags, startup,
1953 &startup_info_size )))
1954 {
1955 RtlReleasePebLock();
1956 close( socketfd[0] );
1957 close( socketfd[1] );
1958 return FALSE;
1959 }
1960 if (!env) env = NtCurrentTeb()->Peb->ProcessParameters->Environment;
1961 env_end = env;
1962 while (*env_end)
1963 {
1964 static const WCHAR WINEDEBUG[] = {'W','I','N','E','D','E','B','U','G','=',0};
1965 if (!winedebug && !strncmpW( env_end, WINEDEBUG, sizeof(WINEDEBUG)/sizeof(WCHAR) - 1 ))
1966 {
1967 DWORD len = WideCharToMultiByte( CP_UNIXCP, 0, env_end, -1, NULL, 0, NULL, NULL );
1968 if ((winedebug = HeapAlloc( GetProcessHeap(), 0, len )))
1969 WideCharToMultiByte( CP_UNIXCP, 0, env_end, -1, winedebug, len, NULL, NULL );
1970 }
1971 env_end += strlenW(env_end) + 1;
1972 }
1973 env_end++;
1974
1975 wine_server_send_fd( socketfd[1] );
1976 close( socketfd[1] );
1977
1978 /* create the process on the server side */
1979
1980 SERVER_START_REQ( new_process )
1981 {
1982 req->inherit_all = inherit;
1983 req->create_flags = flags;
1984 req->socket_fd = socketfd[1];
1985 req->exe_file = wine_server_obj_handle( hFile );
1986 req->process_access = PROCESS_ALL_ACCESS;
1987 req->process_attr = (psa && (psa->nLength >= sizeof(*psa)) && psa->bInheritHandle) ? OBJ_INHERIT : 0;
1988 req->thread_access = THREAD_ALL_ACCESS;
1989 req->thread_attr = (tsa && (tsa->nLength >= sizeof(*tsa)) && tsa->bInheritHandle) ? OBJ_INHERIT : 0;
1990 req->info_size = startup_info_size;
1991
1992 wine_server_add_data( req, startup_info, startup_info_size );
1993 wine_server_add_data( req, env, (env_end - env) * sizeof(WCHAR) );
1994 if ((ret = !wine_server_call_err( req )))
1995 {
1996 info->dwProcessId = (DWORD)reply->pid;
1997 info->dwThreadId = (DWORD)reply->tid;
1998 info->hProcess = wine_server_ptr_handle( reply->phandle );
1999 info->hThread = wine_server_ptr_handle( reply->thandle );
2000 }
2001 process_info = wine_server_ptr_handle( reply->info );
2002 }
2003 SERVER_END_REQ;
2004
2005 RtlReleasePebLock();
2006 if (!ret)
2007 {
2008 close( socketfd[0] );
2009 HeapFree( GetProcessHeap(), 0, startup_info );
2010 HeapFree( GetProcessHeap(), 0, winedebug );
2011 return FALSE;
2012 }
2013
2014 if (!(flags & (CREATE_NEW_CONSOLE | DETACHED_PROCESS)))
2015 {
2016 if (startup_info->hstdin)
2017 wine_server_handle_to_fd( wine_server_ptr_handle(startup_info->hstdin),
2018 FILE_READ_DATA, &stdin_fd, NULL );
2019 if (startup_info->hstdout)
2020 wine_server_handle_to_fd( wine_server_ptr_handle(startup_info->hstdout),
2021 FILE_WRITE_DATA, &stdout_fd, NULL );
2022 }
2023 HeapFree( GetProcessHeap(), 0, startup_info );
2024
2025 /* create the child process */
2026
2027 pid = exec_loader( cmd_line, flags, socketfd[0], stdin_fd, stdout_fd, unixdir,
2028 winedebug, binary_info, FALSE );
2029
2030 if (stdin_fd != -1) close( stdin_fd );
2031 if (stdout_fd != -1) close( stdout_fd );
2032 close( socketfd[0] );
2033 HeapFree( GetProcessHeap(), 0, winedebug );
2034 if (pid == -1)
2035 {
2036 FILE_SetDosError();
2037 goto error;
2038 }
2039
2040 /* wait for the new process info to be ready */
2041
2042 WaitForSingleObject( process_info, INFINITE );
2043 SERVER_START_REQ( get_new_process_info )
2044 {
2045 req->info = wine_server_obj_handle( process_info );
2046 wine_server_call( req );
2047 success = reply->success;
2048 err = reply->exit_code;
2049 }
2050 SERVER_END_REQ;
2051
2052 if (!success)
2053 {
2054 SetLastError( err ? err : ERROR_INTERNAL_ERROR );
2055 goto error;
2056 }
2057 CloseHandle( process_info );
2058 return success;
2059
2060 error:
2061 CloseHandle( process_info );
2062 CloseHandle( info->hProcess );
2063 CloseHandle( info->hThread );
2064 info->hProcess = info->hThread = 0;
2065 info->dwProcessId = info->dwThreadId = 0;
2066 return FALSE;
2067 }
2068
2069
2070 /***********************************************************************
2071 * create_vdm_process
2072 *
2073 * Create a new VDM process for a 16-bit or DOS application.
2074 */
2075 static BOOL create_vdm_process( LPCWSTR filename, LPWSTR cmd_line, LPWSTR env, LPCWSTR cur_dir,
2076 LPSECURITY_ATTRIBUTES psa, LPSECURITY_ATTRIBUTES tsa,
2077 BOOL inherit, DWORD flags, LPSTARTUPINFOW startup,
2078 LPPROCESS_INFORMATION info, LPCSTR unixdir,
2079 const struct binary_info *binary_info, int exec_only )
2080 {
2081 static const WCHAR argsW[] = {'%','s',' ','-','-','a','p','p','-','n','a','m','e',' ','"','%','s','"',' ','%','s',0};
2082
2083 BOOL ret;
2084 LPWSTR new_cmd_line = HeapAlloc( GetProcessHeap(), 0,
2085 (strlenW(filename) + strlenW(cmd_line) + 30) * sizeof(WCHAR) );
2086
2087 if (!new_cmd_line)
2088 {
2089 SetLastError( ERROR_OUTOFMEMORY );
2090 return FALSE;
2091 }
2092 sprintfW( new_cmd_line, argsW, winevdmW, filename, cmd_line );
2093 ret = create_process( 0, winevdmW, new_cmd_line, env, cur_dir, psa, tsa, inherit,
2094 flags, startup, info, unixdir, binary_info, exec_only );
2095 HeapFree( GetProcessHeap(), 0, new_cmd_line );
2096 return ret;
2097 }
2098
2099
2100 /***********************************************************************
2101 * create_cmd_process
2102 *
2103 * Create a new cmd shell process for a .BAT file.
2104 */
2105 static BOOL create_cmd_process( LPCWSTR filename, LPWSTR cmd_line, LPVOID env, LPCWSTR cur_dir,
2106 LPSECURITY_ATTRIBUTES psa, LPSECURITY_ATTRIBUTES tsa,
2107 BOOL inherit, DWORD flags, LPSTARTUPINFOW startup,
2108 LPPROCESS_INFORMATION info )
2109
2110 {
2111 static const WCHAR comspecW[] = {'C','O','M','S','P','E','C',0};
2112 static const WCHAR slashcW[] = {' ','/','c',' ',0};
2113 WCHAR comspec[MAX_PATH];
2114 WCHAR *newcmdline;
2115 BOOL ret;
2116
2117 if (!GetEnvironmentVariableW( comspecW, comspec, sizeof(comspec)/sizeof(WCHAR) ))
2118 return FALSE;
2119 if (!(newcmdline = HeapAlloc( GetProcessHeap(), 0,
2120 (strlenW(comspec) + 4 + strlenW(cmd_line) + 1) * sizeof(WCHAR))))
2121 return FALSE;
2122
2123 strcpyW( newcmdline, comspec );
2124 strcatW( newcmdline, slashcW );
2125 strcatW( newcmdline, cmd_line );
2126 ret = CreateProcessW( comspec, newcmdline, psa, tsa, inherit,
2127 flags, env, cur_dir, startup, info );
2128 HeapFree( GetProcessHeap(), 0, newcmdline );
2129 return ret;
2130 }
2131
2132
2133 /*************************************************************************
2134 * get_file_name
2135 *
2136 * Helper for CreateProcess: retrieve the file name to load from the
2137 * app name and command line. Store the file name in buffer, and
2138 * return a possibly modified command line.
2139 * Also returns a handle to the opened file if it's a Windows binary.
2140 */
2141 static LPWSTR get_file_name( LPCWSTR appname, LPWSTR cmdline, LPWSTR buffer,
2142 int buflen, HANDLE *handle, struct binary_info *binary_info )
2143 {
2144 static const WCHAR quotesW[] = {'"','%','s','"',0};
2145
2146 WCHAR *name, *pos, *first_space, *ret = NULL;
2147 const WCHAR *p;
2148
2149 /* if we have an app name, everything is easy */
2150
2151 if (appname)
2152 {
2153 /* use the unmodified app name as file name */
2154 lstrcpynW( buffer, appname, buflen );
2155 *handle = open_exe_file( buffer, binary_info );
2156 if (!(ret = cmdline) || !cmdline[0])
2157 {
2158 /* no command-line, create one */
2159 if ((ret = HeapAlloc( GetProcessHeap(), 0, (strlenW(appname) + 3) * sizeof(WCHAR) )))
2160 sprintfW( ret, quotesW, appname );
2161 }
2162 return ret;
2163 }
2164
2165 /* first check for a quoted file name */
2166
2167 if ((cmdline[0] == '"') && ((p = strchrW( cmdline + 1, '"' ))))
2168 {
2169 int len = p - cmdline - 1;
2170 /* extract the quoted portion as file name */
2171 if (!(name = HeapAlloc( GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR) ))) return NULL;
2172 memcpy( name, cmdline + 1, len * sizeof(WCHAR) );
2173 name[len] = 0;
2174
2175 if (!find_exe_file( name, buffer, buflen, handle, binary_info )) goto done;
2176 ret = cmdline; /* no change necessary */
2177 goto done;
2178 }
2179
2180 /* now try the command-line word by word */
2181
2182 if (!(name = HeapAlloc( GetProcessHeap(), 0, (strlenW(cmdline) + 1) * sizeof(WCHAR) )))
2183 return NULL;
2184 pos = name;
2185 p = cmdline;
2186 first_space = NULL;
2187
2188 for (;;)
2189 {
2190 while (*p && *p != ' ' && *p != '\t') *pos++ = *p++;
2191 *pos = 0;
2192 if (find_exe_file( name, buffer, buflen, handle, binary_info ))
2193 {
2194 ret = cmdline;
2195 break;
2196 }
2197 if (!first_space) first_space = pos;
2198 if (!(*pos++ = *p++)) break;
2199 }
2200
2201 if (!ret)
2202 {
2203 SetLastError( ERROR_FILE_NOT_FOUND );
2204 }
2205 else if (first_space) /* build a new command-line with quotes */
2206 {
2207 if (!(ret = HeapAlloc( GetProcessHeap(), 0, (strlenW(cmdline) + 3) * sizeof(WCHAR) )))
2208 goto done;
2209 sprintfW( ret, quotesW, name );
2210 strcatW( ret, p );
2211 }
2212
2213 done:
2214 HeapFree( GetProcessHeap(), 0, name );
2215 return ret;
2216 }
2217
2218
2219 /* Steam hotpatches CreateProcessA and W, so to prevent it from crashing use an internal function */
2220 static BOOL create_process_impl( LPCWSTR app_name, LPWSTR cmd_line, LPSECURITY_ATTRIBUTES process_attr,
2221 LPSECURITY_ATTRIBUTES thread_attr, BOOL inherit, DWORD flags,
2222 LPVOID env, LPCWSTR cur_dir, LPSTARTUPINFOW startup_info,
2223 LPPROCESS_INFORMATION info )
2224 {
2225 BOOL retv = FALSE;
2226 HANDLE hFile = 0;
2227 char *unixdir = NULL;
2228 WCHAR name[MAX_PATH];
2229 WCHAR *tidy_cmdline, *p, *envW = env;
2230 struct binary_info binary_info;
2231
2232 /* Process the AppName and/or CmdLine to get module name and path */
2233
2234 TRACE("app %s cmdline %s\n", debugstr_w(app_name), debugstr_w(cmd_line) );
2235
2236 if (!(tidy_cmdline = get_file_name( app_name, cmd_line, name, sizeof(name)/sizeof(WCHAR),
2237 &hFile, &binary_info )))
2238 return FALSE;
2239 if (hFile == INVALID_HANDLE_VALUE) goto done;
2240
2241 /* Warn if unsupported features are used */
2242
2243 if (flags & (IDLE_PRIORITY_CLASS | HIGH_PRIORITY_CLASS | REALTIME_PRIORITY_CLASS |
2244 CREATE_NEW_PROCESS_GROUP | CREATE_SEPARATE_WOW_VDM | CREATE_SHARED_WOW_VDM |
2245 CREATE_DEFAULT_ERROR_MODE | CREATE_NO_WINDOW |
2246 PROFILE_USER | PROFILE_KERNEL | PROFILE_SERVER))
2247 WARN("(%s,...): ignoring some flags in %x\n", debugstr_w(name), flags);
2248
2249 if (cur_dir)
2250 {
2251 if (!(unixdir = wine_get_unix_file_name( cur_dir )))
2252 {
2253 SetLastError(ERROR_DIRECTORY);
2254 goto done;
2255 }
2256 }
2257 else
2258 {
2259 WCHAR buf[MAX_PATH];
2260 if (GetCurrentDirectoryW(MAX_PATH, buf)) unixdir = wine_get_unix_file_name( buf );
2261 }
2262
2263 if (env && !(flags & CREATE_UNICODE_ENVIRONMENT)) /* convert environment to unicode */
2264 {
2265 char *e = env;
2266 DWORD lenW;
2267
2268 while (*e) e += strlen(e) + 1;
2269 e++; /* final null */
2270 lenW = MultiByteToWideChar( CP_ACP, 0, env, e - (char*)env, NULL, 0 );
2271 envW = HeapAlloc( GetProcessHeap(), 0, lenW * sizeof(WCHAR) );
2272 MultiByteToWideChar( CP_ACP, 0, env, e - (char*)env, envW, lenW );
2273 flags |= CREATE_UNICODE_ENVIRONMENT;
2274 }
2275
2276 info->hThread = info->hProcess = 0;
2277 info->dwProcessId = info->dwThreadId = 0;
2278
2279 if (binary_info.flags & BINARY_FLAG_DLL)
2280 {
2281 TRACE( "not starting %s since it is a dll\n", debugstr_w(name) );
2282 SetLastError( ERROR_BAD_EXE_FORMAT );
2283 }
2284 else switch (binary_info.type)
2285 {
2286 case BINARY_PE:
2287 TRACE( "starting %s as Win%d binary (%p-%p)\n",
2288 debugstr_w(name), (binary_info.flags & BINARY_FLAG_64BIT) ? 64 : 32,
2289 binary_info.res_start, binary_info.res_end );
2290 retv = create_process( hFile, name, tidy_cmdline, envW, cur_dir, process_attr, thread_attr,
2291 inherit, flags, startup_info, info, unixdir, &binary_info, FALSE );
2292 break;
2293 case BINARY_OS216:
2294 case BINARY_WIN16:
2295 case BINARY_DOS:
2296 TRACE( "starting %s as Win16/DOS binary\n", debugstr_w(name) );
2297 retv = create_vdm_process( name, tidy_cmdline, envW, cur_dir, process_attr, thread_attr,
2298 inherit, flags, startup_info, info, unixdir, &binary_info, FALSE );
2299 break;
2300 case BINARY_UNIX_LIB:
2301 TRACE( "starting %s as %d-bit Winelib app\n",
2302 debugstr_w(name), (binary_info.flags & BINARY_FLAG_64BIT) ? 64 : 32 );
2303 retv = create_process( hFile, name, tidy_cmdline, envW, cur_dir, process_attr, thread_attr,
2304 inherit, flags, startup_info, info, unixdir, &binary_info, FALSE );
2305 break;
2306 case BINARY_UNKNOWN:
2307 /* check for .com or .bat extension */
2308 if ((p = strrchrW( name, '.' )))
2309 {
2310 if (!strcmpiW( p, comW ) || !strcmpiW( p, pifW ))
2311 {
2312 TRACE( "starting %s as DOS binary\n", debugstr_w(name) );
2313 retv = create_vdm_process( name, tidy_cmdline, envW, cur_dir, process_attr, thread_attr,
2314 inherit, flags, startup_info, info, unixdir,
2315 &binary_info, FALSE );
2316 break;
2317 }
2318 if (!strcmpiW( p, batW ) || !strcmpiW( p, cmdW ) )
2319 {
2320 TRACE( "starting %s as batch binary\n", debugstr_w(name) );
2321 retv = create_cmd_process( name, tidy_cmdline, envW, cur_dir, process_attr, thread_attr,
2322 inherit, flags, startup_info, info );
2323 break;
2324 }
2325 }
2326 /* fall through */
2327 case BINARY_UNIX_EXE:
2328 {
2329 /* unknown file, try as unix executable */
2330 char *unix_name;
2331
2332 TRACE( "starting %s as Unix binary\n", debugstr_w(name) );
2333
2334 if ((unix_name = wine_get_unix_file_name( name )))
2335 {
2336 retv = (fork_and_exec( unix_name, tidy_cmdline, envW, unixdir, flags, startup_info ) != -1);
2337 HeapFree( GetProcessHeap(), 0, unix_name );
2338 }
2339 }
2340 break;
2341 }
2342 if (hFile) CloseHandle( hFile );
2343
2344 done:
2345 if (tidy_cmdline != cmd_line) HeapFree( GetProcessHeap(), 0, tidy_cmdline );
2346 if (envW != env) HeapFree( GetProcessHeap(), 0, envW );
2347 HeapFree( GetProcessHeap(), 0, unixdir );
2348 if (retv)
2349 TRACE( "started process pid %04x tid %04x\n", info->dwProcessId, info->dwThreadId );
2350 return retv;
2351 }
2352
2353
2354 /**********************************************************************
2355 * CreateProcessA (KERNEL32.@)
2356 */
2357 BOOL WINAPI DECLSPEC_HOTPATCH CreateProcessA( LPCSTR app_name, LPSTR cmd_line, LPSECURITY_ATTRIBUTES process_attr,
2358 LPSECURITY_ATTRIBUTES thread_attr, BOOL inherit,
2359 DWORD flags, LPVOID env, LPCSTR cur_dir,
2360 LPSTARTUPINFOA startup_info, LPPROCESS_INFORMATION info )
2361 {
2362 BOOL ret = FALSE;
2363 WCHAR *app_nameW = NULL, *cmd_lineW = NULL, *cur_dirW = NULL;
2364 UNICODE_STRING desktopW, titleW;
2365 STARTUPINFOW infoW;
2366
2367 desktopW.Buffer = NULL;
2368 titleW.Buffer = NULL;
2369 if (app_name && !(app_nameW = FILE_name_AtoW( app_name, TRUE ))) goto done;
2370 if (cmd_line && !(cmd_lineW = FILE_name_AtoW( cmd_line, TRUE ))) goto done;
2371 if (cur_dir && !(cur_dirW = FILE_name_AtoW( cur_dir, TRUE ))) goto done;
2372
2373 if (startup_info->lpDesktop) RtlCreateUnicodeStringFromAsciiz( &desktopW, startup_info->lpDesktop );
2374 if (startup_info->lpTitle) RtlCreateUnicodeStringFromAsciiz( &titleW, startup_info->lpTitle );
2375
2376 memcpy( &infoW, startup_info, sizeof(infoW) );
2377 infoW.lpDesktop = desktopW.Buffer;
2378 infoW.lpTitle = titleW.Buffer;
2379
2380 if (startup_info->lpReserved)
2381 FIXME("StartupInfo.lpReserved is used, please report (%s)\n",
2382 debugstr_a(startup_info->lpReserved));
2383
2384 ret = create_process_impl( app_nameW, cmd_lineW, process_attr, thread_attr,
2385 inherit, flags, env, cur_dirW, &infoW, info );
2386 done:
2387 HeapFree( GetProcessHeap(), 0, app_nameW );
2388 HeapFree( GetProcessHeap(), 0, cmd_lineW );
2389 HeapFree( GetProcessHeap(), 0, cur_dirW );
2390 RtlFreeUnicodeString( &desktopW );
2391 RtlFreeUnicodeString( &titleW );
2392 return ret;
2393 }
2394
2395
2396 /**********************************************************************
2397 * CreateProcessW (KERNEL32.@)
2398 */
2399 BOOL WINAPI DECLSPEC_HOTPATCH CreateProcessW( LPCWSTR app_name, LPWSTR cmd_line, LPSECURITY_ATTRIBUTES process_attr,
2400 LPSECURITY_ATTRIBUTES thread_attr, BOOL inherit, DWORD flags,
2401 LPVOID env, LPCWSTR cur_dir, LPSTARTUPINFOW startup_info,
2402 LPPROCESS_INFORMATION info )
2403 {
2404 return create_process_impl( app_name, cmd_line, process_attr, thread_attr,
2405 inherit, flags, env, cur_dir, startup_info, info);
2406 }
2407
2408
2409 /**********************************************************************
2410 * exec_process
2411 */
2412 static void exec_process( LPCWSTR name )
2413 {
2414 HANDLE hFile;
2415 WCHAR *p;
2416 STARTUPINFOW startup_info;
2417 PROCESS_INFORMATION info;
2418 struct binary_info binary_info;
2419
2420 hFile = open_exe_file( name, &binary_info );
2421 if (!hFile || hFile == INVALID_HANDLE_VALUE) return;
2422
2423 memset( &startup_info, 0, sizeof(startup_info) );
2424 startup_info.cb = sizeof(startup_info);
2425
2426 /* Determine executable type */
2427
2428 if (binary_info.flags & BINARY_FLAG_DLL) return;
2429 switch (binary_info.type)
2430 {
2431 case BINARY_PE:
2432 TRACE( "starting %s as Win%d binary (%p-%p)\n",
2433 debugstr_w(name), (binary_info.flags & BINARY_FLAG_64BIT) ? 64 : 32,
2434 binary_info.res_start, binary_info.res_end );
2435 create_process( hFile, name, GetCommandLineW(), NULL, NULL, NULL, NULL,
2436 FALSE, 0, &startup_info, &info, NULL, &binary_info, TRUE );
2437 break;
2438 case BINARY_UNIX_LIB:
2439 TRACE( "%s is a Unix library, starting as Winelib app\n", debugstr_w(name) );
2440 create_process( hFile, name, GetCommandLineW(), NULL, NULL, NULL, NULL,
2441 FALSE, 0, &startup_info, &info, NULL, &binary_info, TRUE );
2442 break;
2443 case BINARY_UNKNOWN:
2444 /* check for .com or .pif extension */
2445 if (!(p = strrchrW( name, '.' ))) break;
2446 if (strcmpiW( p, comW ) && strcmpiW( p, pifW )) break;
2447 /* fall through */
2448 case BINARY_OS216:
2449 case BINARY_WIN16:
2450 case BINARY_DOS:
2451 TRACE( "starting %s as Win16/DOS binary\n", debugstr_w(name) );
2452 create_vdm_process( name, GetCommandLineW(), NULL, NULL, NULL, NULL,
2453 FALSE, 0, &startup_info, &info, NULL, &binary_info, TRUE );
2454 break;
2455 default:
2456 break;
2457 }
2458 CloseHandle( hFile );
2459 }
2460
2461
2462 /***********************************************************************
2463 * wait_input_idle
2464 *
2465 * Wrapper to call WaitForInputIdle USER function
2466 */
2467 typedef DWORD (WINAPI *WaitForInputIdle_ptr)( HANDLE hProcess, DWORD dwTimeOut );
2468
2469 static DWORD wait_input_idle( HANDLE process, DWORD timeout )
2470 {
2471 HMODULE mod = GetModuleHandleA( "user32.dll" );
2472 if (mod)
2473 {
2474 WaitForInputIdle_ptr ptr = (WaitForInputIdle_ptr)GetProcAddress( mod, "WaitForInputIdle" );
2475 if (ptr) return ptr( process, timeout );
2476 }
2477 return 0;
2478 }
2479
2480
2481 /***********************************************************************
2482 * WinExec (KERNEL32.@)
2483 */
2484 UINT WINAPI WinExec( LPCSTR lpCmdLine, UINT nCmdShow )
2485 {
2486 PROCESS_INFORMATION info;
2487 STARTUPINFOA startup;
2488 char *cmdline;
2489 UINT ret;
2490
2491 memset( &startup, 0, sizeof(startup) );
2492 startup.cb = sizeof(startup);
2493 startup.dwFlags = STARTF_USESHOWWINDOW;
2494 startup.wShowWindow = nCmdShow;
2495
2496 /* cmdline needs to be writable for CreateProcess */
2497 if (!(cmdline = HeapAlloc( GetProcessHeap(), 0, strlen(lpCmdLine)+1 ))) return 0;
2498 strcpy( cmdline, lpCmdLine );
2499
2500 if (CreateProcessA( NULL, cmdline, NULL, NULL, FALSE,
2501 0, NULL, NULL, &startup, &info ))
2502 {
2503 /* Give 30 seconds to the app to come up */
2504 if (wait_input_idle( info.hProcess, 30000 ) == WAIT_FAILED)
2505 WARN("WaitForInputIdle failed: Error %d\n", GetLastError() );
2506 ret = 33;
2507 /* Close off the handles */
2508 CloseHandle( info.hThread );
2509 CloseHandle( info.hProcess );
2510 }
2511 else if ((ret = GetLastError()) >= 32)
2512 {
2513 FIXME("Strange error set by CreateProcess: %d\n", ret );
2514 ret = 11;
2515 }
2516 HeapFree( GetProcessHeap(), 0, cmdline );
2517 return ret;
2518 }
2519
2520
2521 /**********************************************************************
2522 * LoadModule (KERNEL32.@)
2523 */
2524 DWORD WINAPI LoadModule( LPCSTR name, LPVOID paramBlock )
2525 {
2526 LOADPARMS32 *params = paramBlock;
2527 PROCESS_INFORMATION info;
2528 STARTUPINFOA startup;
2529 DWORD ret;
2530 LPSTR cmdline, p;
2531 char filename[MAX_PATH];
2532 BYTE len;
2533
2534 if (!name) return ERROR_FILE_NOT_FOUND;
2535
2536 if (!SearchPathA( NULL, name, ".exe", sizeof(filename), filename, NULL ) &&
2537 !SearchPathA( NULL, name, NULL, sizeof(filename), filename, NULL ))
2538 return GetLastError();
2539
2540 len = (BYTE)params->lpCmdLine[0];
2541 if (!(cmdline = HeapAlloc( GetProcessHeap(), 0, strlen(filename) + len + 2 )))
2542 return ERROR_NOT_ENOUGH_MEMORY;
2543
2544 strcpy( cmdline, filename );
2545 p = cmdline + strlen(cmdline);
2546 *p++ = ' ';
2547 memcpy( p, params->lpCmdLine + 1, len );
2548 p[len] = 0;
2549
2550 memset( &startup, 0, sizeof(startup) );
2551 startup.cb = sizeof(startup);
2552 if (params->lpCmdShow)
2553 {
2554 startup.dwFlags = STARTF_USESHOWWINDOW;
2555 startup.wShowWindow = ((WORD *)params->lpCmdShow)[1];
2556 }
2557
2558 if (CreateProcessA( filename, cmdline, NULL, NULL, FALSE, 0,
2559 params->lpEnvAddress, NULL, &startup, &info ))
2560 {
2561 /* Give 30 seconds to the app to come up */
2562 if (wait_input_idle( info.hProcess, 30000 ) == WAIT_FAILED)
2563 WARN("WaitForInputIdle failed: Error %d\n", GetLastError() );
2564 ret = 33;
2565 /* Close off the handles */
2566 CloseHandle( info.hThread );
2567 CloseHandle( info.hProcess );
2568 }
2569 else if ((ret = GetLastError()) >= 32)
2570 {
2571 FIXME("Strange error set by CreateProcess: %u\n", ret );
2572 ret = 11;
2573 }
2574
2575 HeapFree( GetProcessHeap(), 0, cmdline );
2576 return ret;
2577 }
2578
2579
2580 /******************************************************************************
2581 * TerminateProcess (KERNEL32.@)
2582 *
2583 * Terminates a process.
2584 *
2585 * PARAMS
2586 * handle [I] Process to terminate.
2587 * exit_code [I] Exit code.
2588 *
2589 * RETURNS
2590 * Success: TRUE.
2591 * Failure: FALSE, check GetLastError().
2592 */
2593 BOOL WINAPI TerminateProcess( HANDLE handle, DWORD exit_code )
2594 {
2595 NTSTATUS status = NtTerminateProcess( handle, exit_code );
2596 if (status) SetLastError( RtlNtStatusToDosError(status) );
2597 return !status;
2598 }
2599
2600 /***********************************************************************
2601 * ExitProcess (KERNEL32.@)
2602 *
2603 * Exits the current process.
2604 *
2605 * PARAMS
2606 * status [I] Status code to exit with.
2607 *
2608 * RETURNS
2609 * Nothing.
2610 */
2611 #ifdef __i386__
2612 __ASM_STDCALL_FUNC( ExitProcess, 4, /* Shrinker depend on this particular ExitProcess implementation */
2613 "pushl %ebp\n\t"
2614 ".byte 0x8B, 0xEC\n\t" /* movl %esp, %ebp */
2615 ".byte 0x6A, 0x00\n\t" /* pushl $0 */
2616 ".byte 0x68, 0x00, 0x00, 0x00, 0x00\n\t" /* pushl $0 - 4 bytes immediate */
2617 "pushl 8(%ebp)\n\t"
2618 "call " __ASM_NAME("process_ExitProcess") __ASM_STDCALL(4) "\n\t"
2619 "leave\n\t"
2620 "ret $4" )
2621
2622 void WINAPI process_ExitProcess( DWORD status )
2623 {
2624 LdrShutdownProcess();
2625 NtTerminateProcess(GetCurrentProcess(), status);
2626 exit(status);
2627 }
2628
2629 #else
2630
2631 void WINAPI ExitProcess( DWORD status )
2632 {
2633 LdrShutdownProcess();
2634 NtTerminateProcess(GetCurrentProcess(), status);
2635 exit(status);
2636 }
2637
2638 #endif
2639
2640 /***********************************************************************
2641 * GetExitCodeProcess [KERNEL32.@]
2642 *
2643 * Gets termination status of specified process.
2644 *
2645 * PARAMS
2646 * hProcess [in] Handle to the process.
2647 * lpExitCode [out] Address to receive termination status.
2648 *
2649 * RETURNS
2650 * Success: TRUE
2651 * Failure: FALSE
2652 */
2653 BOOL WINAPI GetExitCodeProcess( HANDLE hProcess, LPDWORD lpExitCode )
2654 {
2655 NTSTATUS status;
2656 PROCESS_BASIC_INFORMATION pbi;
2657
2658 status = NtQueryInformationProcess(hProcess, ProcessBasicInformation, &pbi,
2659 sizeof(pbi), NULL);
2660 if (status == STATUS_SUCCESS)
2661 {
2662 if (lpExitCode) *lpExitCode = pbi.ExitStatus;
2663 return TRUE;
2664 }
2665 SetLastError( RtlNtStatusToDosError(status) );
2666 return FALSE;
2667 }
2668
2669
2670 /***********************************************************************
2671 * SetErrorMode (KERNEL32.@)
2672 */
2673 UINT WINAPI SetErrorMode( UINT mode )
2674 {
2675 UINT old;
2676
2677 NtQueryInformationProcess( GetCurrentProcess(), ProcessDefaultHardErrorMode,
2678 &old, sizeof(old), NULL );
2679 NtSetInformationProcess( GetCurrentProcess(), ProcessDefaultHardErrorMode,
2680 &mode, sizeof(mode) );
2681 return old;
2682 }
2683
2684 /***********************************************************************
2685 * GetErrorMode (KERNEL32.@)
2686 */
2687 UINT WINAPI GetErrorMode( void )
2688 {
2689 UINT mode;
2690
2691 NtQueryInformationProcess( GetCurrentProcess(), ProcessDefaultHardErrorMode,
2692 &mode, sizeof(mode), NULL );
2693 return mode;
2694 }
2695
2696 /**********************************************************************
2697 * TlsAlloc [KERNEL32.@]
2698 *
2699 * Allocates a thread local storage index.
2700 *
2701 * RETURNS
2702 * Success: TLS index.
2703 * Failure: 0xFFFFFFFF
2704 */
2705 DWORD WINAPI TlsAlloc( void )
2706 {
2707 DWORD index;
2708 PEB * const peb = NtCurrentTeb()->Peb;
2709
2710 RtlAcquirePebLock();
2711 index = RtlFindClearBitsAndSet( peb->TlsBitmap, 1, 0 );
2712 if (index != ~0U) NtCurrentTeb()->TlsSlots[index] = 0; /* clear the value */
2713 else
2714 {
2715 index = RtlFindClearBitsAndSet( peb->TlsExpansionBitmap, 1, 0 );
2716 if (index != ~0U)
2717 {
2718 if (!NtCurrentTeb()->TlsExpansionSlots &&
2719 !(NtCurrentTeb()->TlsExpansionSlots = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY,
2720 8 * sizeof(peb->TlsExpansionBitmapBits) * sizeof(void*) )))
2721 {
2722 RtlClearBits( peb->TlsExpansionBitmap, index, 1 );
2723 index = ~0U;
2724 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
2725 }
2726 else
2727 {
2728 NtCurrentTeb()->TlsExpansionSlots[index] = 0; /* clear the value */
2729 index += TLS_MINIMUM_AVAILABLE;
2730 }
2731 }
2732 else SetLastError( ERROR_NO_MORE_ITEMS );
2733 }
2734 RtlReleasePebLock();
2735 return index;
2736 }
2737
2738
2739 /**********************************************************************
2740 * TlsFree [KERNEL32.@]
2741 *
2742 * Releases a thread local storage index, making it available for reuse.
2743 *
2744 * PARAMS
2745 * index [in] TLS index to free.
2746 *
2747 * RETURNS
2748 * Success: TRUE
2749 * Failure: FALSE
2750 */
2751 BOOL WINAPI TlsFree( DWORD index )
2752 {
2753 BOOL ret;
2754
2755 RtlAcquirePebLock();
2756 if (index >= TLS_MINIMUM_AVAILABLE)
2757 {
2758 ret = RtlAreBitsSet( NtCurrentTeb()->Peb->TlsExpansionBitmap, index - TLS_MINIMUM_AVAILABLE, 1 );
2759 if (ret) RtlClearBits( NtCurrentTeb()->Peb->TlsExpansionBitmap, index - TLS_MINIMUM_AVAILABLE, 1 );
2760 }
2761 else
2762 {
2763 ret = RtlAreBitsSet( NtCurrentTeb()->Peb->TlsBitmap, index, 1 );
2764 if (ret) RtlClearBits( NtCurrentTeb()->Peb->TlsBitmap, index, 1 );
2765 }
2766 if (ret) NtSetInformationThread( GetCurrentThread(), ThreadZeroTlsCell, &index, sizeof(index) );
2767 else SetLastError( ERROR_INVALID_PARAMETER );
2768 RtlReleasePebLock();
2769 return ret;
2770 }
2771
2772
2773 /**********************************************************************
2774 * TlsGetValue [KERNEL32.@]
2775 *
2776 * Gets value in a thread's TLS slot.
2777 *
2778 * PARAMS
2779 * index [in] TLS index to retrieve value for.
2780 *
2781 * RETURNS
2782 * Success: Value stored in calling thread's TLS slot for index.
2783 * Failure: 0 and GetLastError() returns NO_ERROR.
2784 */
2785 LPVOID WINAPI TlsGetValue( DWORD index )
2786 {
2787 LPVOID ret;
2788
2789 if (index < TLS_MINIMUM_AVAILABLE)
2790 {
2791 ret = NtCurrentTeb()->TlsSlots[index];
2792 }
2793 else
2794 {
2795 index -= TLS_MINIMUM_AVAILABLE;
2796 if (index >= 8 * sizeof(NtCurrentTeb()->Peb->TlsExpansionBitmapBits))
2797 {
2798 SetLastError( ERROR_INVALID_PARAMETER );
2799 return NULL;
2800 }
2801 if (!NtCurrentTeb()->TlsExpansionSlots) ret = NULL;
2802 else ret = NtCurrentTeb()->TlsExpansionSlots[index];
2803 }
2804 SetLastError( ERROR_SUCCESS );
2805 return ret;
2806 }
2807
2808
2809 /**********************************************************************
2810 * TlsSetValue [KERNEL32.@]
2811 *
2812 * Stores a value in the thread's TLS slot.
2813 *
2814 * PARAMS
2815 * index [in] TLS index to set value for.
2816 * value [in] Value to be stored.
2817 *
2818 * RETURNS
2819 * Success: TRUE
2820 * Failure: FALSE
2821 */
2822 BOOL WINAPI TlsSetValue( DWORD index, LPVOID value )
2823 {
2824 if (index < TLS_MINIMUM_AVAILABLE)
2825 {
2826 NtCurrentTeb()->TlsSlots[index] = value;
2827 }
2828 else
2829 {
2830 index -= TLS_MINIMUM_AVAILABLE;
2831 if (index >= 8 * sizeof(NtCurrentTeb()->Peb->TlsExpansionBitmapBits))
2832 {
2833 SetLastError( ERROR_INVALID_PARAMETER );
2834 return FALSE;
2835 }
2836 if (!NtCurrentTeb()->TlsExpansionSlots &&
2837 !(NtCurrentTeb()->TlsExpansionSlots = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY,
2838 8 * sizeof(NtCurrentTeb()->Peb->TlsExpansionBitmapBits) * sizeof(void*) )))
2839 {
2840 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
2841 return FALSE;
2842 }
2843 NtCurrentTeb()->TlsExpansionSlots[index] = value;
2844 }
2845 return TRUE;
2846 }
2847
2848
2849 /***********************************************************************
2850 * GetProcessFlags (KERNEL32.@)
2851 */
2852 DWORD WINAPI GetProcessFlags( DWORD processid )
2853 {
2854 IMAGE_NT_HEADERS *nt;
2855 DWORD flags = 0;
2856
2857 if (processid && processid != GetCurrentProcessId()) return 0;
2858
2859 if ((nt = RtlImageNtHeader( NtCurrentTeb()->Peb->ImageBaseAddress )))
2860 {
2861 if (nt->OptionalHeader.Subsystem == IMAGE_SUBSYSTEM_WINDOWS_CUI)
2862 flags |= PDB32_CONSOLE_PROC;
2863 }
2864 if (!AreFileApisANSI()) flags |= PDB32_FILE_APIS_OEM;
2865 if (IsDebuggerPresent()) flags |= PDB32_DEBUGGED;
2866 return flags;
2867 }
2868
2869
2870 /*********************************************************************
2871 * OpenProcess (KERNEL32.@)
2872 *
2873 * Opens a handle to a process.
2874 *
2875 * PARAMS
2876 * access [I] Desired access rights assigned to the returned handle.
2877 * inherit [I] Determines whether or not child processes will inherit the handle.
2878 * id [I] Process identifier of the process to get a handle to.
2879 *
2880 * RETURNS
2881 * Success: Valid handle to the specified process.
2882 * Failure: NULL, check GetLastError().
2883 */
2884 HANDLE WINAPI OpenProcess( DWORD access, BOOL inherit, DWORD id )
2885 {
2886 NTSTATUS status;
2887 HANDLE handle;
2888 OBJECT_ATTRIBUTES attr;
2889 CLIENT_ID cid;
2890
2891 cid.UniqueProcess = ULongToHandle(id);
2892 cid.UniqueThread = 0; /* FIXME ? */
2893
2894 attr.Length = sizeof(OBJECT_ATTRIBUTES);
2895 attr.RootDirectory = NULL;
2896 attr.Attributes = inherit ? OBJ_INHERIT : 0;
2897 attr.SecurityDescriptor = NULL;
2898 attr.SecurityQualityOfService = NULL;
2899 attr.ObjectName = NULL;
2900
2901 if (GetVersion() & 0x80000000) access = PROCESS_ALL_ACCESS;
2902
2903 status = NtOpenProcess(&handle, access, &attr, &cid);
2904 if (status != STATUS_SUCCESS)
2905 {
2906 SetLastError( RtlNtStatusToDosError(status) );
2907 return NULL;
2908 }
2909 return handle;
2910 }
2911
2912
2913 /*********************************************************************
2914 * GetProcessId (KERNEL32.@)
2915 *
2916 * Gets the a unique identifier of a process.
2917 *
2918 * PARAMS
2919 * hProcess [I] Handle to the process.
2920 *
2921 * RETURNS
2922 * Success: TRUE.
2923 * Failure: FALSE, check GetLastError().
2924 *
2925 * NOTES
2926 *
2927 * The identifier is unique only on the machine and only until the process
2928 * exits (including system shutdown).
2929 */
2930 DWORD WINAPI GetProcessId( HANDLE hProcess )
2931 {
2932 NTSTATUS status;
2933 PROCESS_BASIC_INFORMATION pbi;
2934
2935 status = NtQueryInformationProcess(hProcess, ProcessBasicInformation, &pbi,
2936 sizeof(pbi), NULL);
2937 if (status == STATUS_SUCCESS) return pbi.UniqueProcessId;
2938 SetLastError( RtlNtStatusToDosError(status) );
2939 return 0;
2940 }
2941
2942
2943 /*********************************************************************
2944 * CloseHandle (KERNEL32.@)
2945 *
2946 * Closes a handle.
2947 *
2948 * PARAMS
2949 * handle [I] Handle to close.
2950 *
2951 * RETURNS
2952 * Success: TRUE.
2953 * Failure: FALSE, check GetLastError().
2954 */
2955 BOOL WINAPI CloseHandle( HANDLE handle )
2956 {
2957 NTSTATUS status;
2958
2959 /* stdio handles need special treatment */
2960 if (handle == (HANDLE)STD_INPUT_HANDLE)
2961 handle = InterlockedExchangePointer( &NtCurrentTeb()->Peb->ProcessParameters->hStdInput, 0 );
2962 else if (handle == (HANDLE)STD_OUTPUT_HANDLE)
2963 handle = InterlockedExchangePointer( &NtCurrentTeb()->Peb->ProcessParameters->hStdOutput, 0 );
2964 else if (handle == (HANDLE)STD_ERROR_HANDLE)
2965 handle = InterlockedExchangePointer( &NtCurrentTeb()->Peb->ProcessParameters->hStdError, 0 );
2966
2967 if (is_console_handle(handle))
2968 return CloseConsoleHandle(handle);
2969
2970 status = NtClose( handle );
2971 if (status) SetLastError( RtlNtStatusToDosError(status) );
2972 return !status;
2973 }
2974
2975
2976 /*********************************************************************
2977 * GetHandleInformation (KERNEL32.@)
2978 */
2979 BOOL WINAPI GetHandleInformation( HANDLE handle, LPDWORD flags )
2980 {
2981 OBJECT_DATA_INFORMATION info;
2982 NTSTATUS status = NtQueryObject( handle, ObjectDataInformation, &info, sizeof(info), NULL );
2983
2984 if (status) SetLastError( RtlNtStatusToDosError(status) );
2985 else if (flags)
2986 {
2987 *flags = 0;
2988 if (info.InheritHandle) *flags |= HANDLE_FLAG_INHERIT;
2989 if (info.ProtectFromClose) *flags |= HANDLE_FLAG_PROTECT_FROM_CLOSE;
2990 }
2991 return !status;
2992 }
2993
2994
2995 /*********************************************************************
2996 * SetHandleInformation (KERNEL32.@)
2997 */
2998 BOOL WINAPI SetHandleInformation( HANDLE handle, DWORD mask, DWORD flags )
2999 {
3000 OBJECT_DATA_INFORMATION info;
3001 NTSTATUS status;
3002
3003 /* if not setting both fields, retrieve current value first */
3004 if ((mask & (HANDLE_FLAG_INHERIT | HANDLE_FLAG_PROTECT_FROM_CLOSE)) !=
3005 (HANDLE_FLAG_INHERIT | HANDLE_FLAG_PROTECT_FROM_CLOSE))
3006 {
3007 if ((status = NtQueryObject( handle, ObjectDataInformation, &info, sizeof(info), NULL )))
3008 {
3009 SetLastError( RtlNtStatusToDosError(status) );
3010 return FALSE;
3011 }
3012 }
3013 if (mask & HANDLE_FLAG_INHERIT)
3014 info.InheritHandle = (flags & HANDLE_FLAG_INHERIT) != 0;
3015 if (mask & HANDLE_FLAG_PROTECT_FROM_CLOSE)
3016 info.ProtectFromClose = (flags & HANDLE_FLAG_PROTECT_FROM_CLOSE) != 0;
3017
3018 status = NtSetInformationObject( handle, ObjectDataInformation, &info, sizeof(info) );
3019 if (status) SetLastError( RtlNtStatusToDosError(status) );
3020 return !status;
3021 }
3022
3023
3024 /*********************************************************************
3025 * DuplicateHandle (KERNEL32.@)
3026 */
3027 BOOL WINAPI DuplicateHandle( HANDLE source_process, HANDLE source,
3028 HANDLE dest_process, HANDLE *dest,
3029 DWORD access, BOOL inherit, DWORD options )
3030 {
3031 NTSTATUS status;
3032
3033 if (is_console_handle(source))
3034 {
3035 /* FIXME: this test is not sufficient, we need to test process ids, not handles */
3036 if (source_process != dest_process ||
3037 source_process != GetCurrentProcess())
3038 {
3039 SetLastError(ERROR_INVALID_PARAMETER);
3040 return FALSE;
3041 }
3042 *dest = DuplicateConsoleHandle( source, access, inherit, options );
3043 return (*dest != INVALID_HANDLE_VALUE);
3044 }
3045 status = NtDuplicateObject( source_process, source, dest_process, dest,
3046 access, inherit ? OBJ_INHERIT : 0, options );
3047 if (status) SetLastError( RtlNtStatusToDosError(status) );
3048 return !status;
3049 }
3050
3051
3052 /***********************************************************************
3053 * ConvertToGlobalHandle (KERNEL32.@)
3054 */
3055 HANDLE WINAPI ConvertToGlobalHandle(HANDLE hSrc)
3056 {
3057 HANDLE ret = INVALID_HANDLE_VALUE;
3058 DuplicateHandle( GetCurrentProcess(), hSrc, GetCurrentProcess(), &ret, 0, FALSE,
3059 DUP_HANDLE_MAKE_GLOBAL | DUP_HANDLE_SAME_ACCESS | DUP_HANDLE_CLOSE_SOURCE );
3060 return ret;
3061 }
3062
3063
3064 /***********************************************************************
3065 * SetHandleContext (KERNEL32.@)
3066 */
3067 BOOL WINAPI SetHandleContext(HANDLE hnd,DWORD context)
3068 {
3069 FIXME("(%p,%d), stub. In case this got called by WSOCK32/WS2_32: "
3070 "the external WINSOCK DLLs won't work with WINE, don't use them.\n",hnd,context);
3071 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3072 return FALSE;
3073 }
3074
3075
3076 /***********************************************************************
3077 * GetHandleContext (KERNEL32.@)
3078 */
3079 DWORD WINAPI GetHandleContext(HANDLE hnd)
3080 {
3081 FIXME("(%p), stub. In case this got called by WSOCK32/WS2_32: "
3082 "the external WINSOCK DLLs won't work with WINE, don't use them.\n",hnd);
3083 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3084 return 0;
3085 }
3086
3087
3088 /***********************************************************************
3089 * CreateSocketHandle (KERNEL32.@)
3090 */
3091 HANDLE WINAPI CreateSocketHandle(void)
3092 {
3093 FIXME("(), stub. In case this got called by WSOCK32/WS2_32: "
3094 "the external WINSOCK DLLs won't work with WINE, don't use them.\n");
3095 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3096 return INVALID_HANDLE_VALUE;
3097 }
3098
3099
3100 /***********************************************************************
3101 * SetPriorityClass (KERNEL32.@)
3102 */
3103 BOOL WINAPI SetPriorityClass( HANDLE hprocess, DWORD priorityclass )
3104 {
3105 NTSTATUS status;
3106 PROCESS_PRIORITY_CLASS ppc;
3107
3108 ppc.Foreground = FALSE;
3109 switch (priorityclass)
3110 {
3111 case IDLE_PRIORITY_CLASS:
3112 ppc.PriorityClass = PROCESS_PRIOCLASS_IDLE; break;
3113 case BELOW_NORMAL_PRIORITY_CLASS:
3114 ppc.PriorityClass = PROCESS_PRIOCLASS_BELOW_NORMAL; break;
3115 case NORMAL_PRIORITY_CLASS:
3116 ppc.PriorityClass = PROCESS_PRIOCLASS_NORMAL; break;
3117 case ABOVE_NORMAL_PRIORITY_CLASS:
3118 ppc.PriorityClass = PROCESS_PRIOCLASS_ABOVE_NORMAL; break;
3119 case HIGH_PRIORITY_CLASS:
3120 ppc.PriorityClass = PROCESS_PRIOCLASS_HIGH; break;
3121 case REALTIME_PRIORITY_CLASS:
3122 ppc.PriorityClass = PROCESS_PRIOCLASS_REALTIME; break;
3123 default:
3124 SetLastError(ERROR_INVALID_PARAMETER);
3125 return FALSE;
3126 }
3127
3128 status = NtSetInformationProcess(hprocess, ProcessPriorityClass,
3129 &ppc, sizeof(ppc));
3130
3131 if (status != STATUS_SUCCESS)
3132 {
3133 SetLastError( RtlNtStatusToDosError(status) );
3134 return FALSE;
3135 }
3136 return TRUE;
3137 }
3138
3139
3140 /***********************************************************************
3141 * GetPriorityClass (KERNEL32.@)
3142 */
3143 DWORD WINAPI GetPriorityClass(HANDLE hProcess)
3144 {
3145 NTSTATUS status;
3146 PROCESS_BASIC_INFORMATION pbi;
3147
3148 status = NtQueryInformationProcess(hProcess, ProcessBasicInformation, &pbi,
3149 sizeof(pbi), NULL);
3150 if (status != STATUS_SUCCESS)
3151 {
3152 SetLastError( RtlNtStatusToDosError(status) );
3153 return 0;
3154 }
3155 switch (pbi.BasePriority)
3156 {
3157 case PROCESS_PRIOCLASS_IDLE: return IDLE_PRIORITY_CLASS;
3158 case PROCESS_PRIOCLASS_BELOW_NORMAL: return BELOW_NORMAL_PRIORITY_CLASS;
3159 case PROCESS_PRIOCLASS_NORMAL: return NORMAL_PRIORITY_CLASS;
3160 case PROCESS_PRIOCLASS_ABOVE_NORMAL: return ABOVE_NORMAL_PRIORITY_CLASS;
3161 case PROCESS_PRIOCLASS_HIGH: return HIGH_PRIORITY_CLASS;
3162 case PROCESS_PRIOCLASS_REALTIME: return REALTIME_PRIORITY_CLASS;
3163 }
3164 SetLastError( ERROR_INVALID_PARAMETER );
3165 return 0;
3166 }
3167
3168
3169 /***********************************************************************
3170 * SetProcessAffinityMask (KERNEL32.@)
3171 */
3172 BOOL WINAPI SetProcessAffinityMask( HANDLE hProcess, DWORD_PTR affmask )
3173 {
3174 NTSTATUS status;
3175
3176 status = NtSetInformationProcess(hProcess, ProcessAffinityMask,
3177 &affmask, sizeof(DWORD_PTR));
3178 if (status)
3179 {
3180 SetLastError( RtlNtStatusToDosError(status) );
3181 return FALSE;
3182 }
3183 return TRUE;
3184 }
3185
3186
3187 /**********************************************************************
3188 * GetProcessAffinityMask (KERNEL32.@)
3189 */
3190 BOOL WINAPI GetProcessAffinityMask( HANDLE hProcess, PDWORD_PTR process_mask, PDWORD_PTR system_mask )
3191 {
3192 NTSTATUS status = STATUS_SUCCESS;
3193
3194 if (system_mask) *system_mask = (1 << NtCurrentTeb()->Peb->NumberOfProcessors) - 1;
3195 if (process_mask)
3196 {
3197 if ((status = NtQueryInformationProcess( hProcess, ProcessAffinityMask,
3198 process_mask, sizeof(*process_mask), NULL )))
3199 SetLastError( RtlNtStatusToDosError(status) );
3200 }
3201 return !status;
3202 }
3203
3204
3205 /***********************************************************************
3206 * GetProcessVersion (KERNEL32.@)
3207 */
3208 DWORD WINAPI GetProcessVersion( DWORD pid )
3209 {
3210 HANDLE process;
3211 NTSTATUS status;
3212 PROCESS_BASIC_INFORMATION pbi;
3213 SIZE_T count;
3214 PEB peb;
3215 IMAGE_DOS_HEADER dos;
3216 IMAGE_NT_HEADERS nt;
3217 DWORD ver = 0;
3218
3219 if (!pid || pid == GetCurrentProcessId())
3220 {
3221 IMAGE_NT_HEADERS *pnt;
3222
3223 if ((pnt = RtlImageNtHeader( NtCurrentTeb()->Peb->ImageBaseAddress )))
3224 return ((pnt->OptionalHeader.MajorSubsystemVersion << 16) |
3225 pnt->OptionalHeader.MinorSubsystemVersion);
3226 return 0;
3227 }
3228
3229 process = OpenProcess(PROCESS_VM_READ | PROCESS_QUERY_INFORMATION, FALSE, pid);
3230 if (!process) return 0;
3231
3232 status = NtQueryInformationProcess(process, ProcessBasicInformation, &pbi, sizeof(pbi), NULL);
3233 if (status) goto err;
3234
3235 status = NtReadVirtualMemory(process, pbi.PebBaseAddress, &peb, sizeof(peb), &count);
3236 if (status || count != sizeof(peb)) goto err;
3237
3238 memset(&dos, 0, sizeof(dos));
3239 status = NtReadVirtualMemory(process, peb.ImageBaseAddress, &dos, sizeof(dos), &count);
3240 if (status || count != sizeof(dos)) goto err;
3241 if (dos.e_magic != IMAGE_DOS_SIGNATURE) goto err;
3242
3243 memset(&nt, 0, sizeof(nt));
3244 status = NtReadVirtualMemory(process, (char *)peb.ImageBaseAddress + dos.e_lfanew, &nt, sizeof(nt), &count);
3245 if (status || count != sizeof(nt)) goto err;
3246 if (nt.Signature != IMAGE_NT_SIGNATURE) goto err;
3247
3248 ver = MAKELONG(nt.OptionalHeader.MinorSubsystemVersion, nt.OptionalHeader.MajorSubsystemVersion);
3249
3250 err:
3251 CloseHandle(process);
3252
3253 if (status != STATUS_SUCCESS)
3254 SetLastError(RtlNtStatusToDosError(status));
3255
3256 return ver;
3257 }
3258
3259
3260 /***********************************************************************
3261 * SetProcessWorkingSetSize [KERNEL32.@]
3262 * Sets the min/max working set sizes for a specified process.
3263 *
3264 * PARAMS
3265 * hProcess [I] Handle to the process of interest
3266 * minset [I] Specifies minimum working set size
3267 * maxset [I] Specifies maximum working set size
3268 *
3269 * RETURNS
3270 * Success: TRUE
3271 * Failure: FALSE
3272 */
3273 BOOL WINAPI SetProcessWorkingSetSize(HANDLE hProcess, SIZE_T minset,
3274 SIZE_T maxset)
3275 {
3276 WARN("(%p,%ld,%ld): stub - harmless\n",hProcess,minset,maxset);
3277 if(( minset == (SIZE_T)-1) && (maxset == (SIZE_T)-1)) {
3278 /* Trim the working set to zero */
3279 /* Swap the process out of physical RAM */
3280 }
3281 return TRUE;
3282 }
3283
3284 /***********************************************************************
3285 * K32EmptyWorkingSet (KERNEL32.@)
3286 */
3287 BOOL WINAPI K32EmptyWorkingSet(HANDLE hProcess)
3288 {
3289 return SetProcessWorkingSetSize(hProcess, (SIZE_T)-1, (SIZE_T)-1);
3290 }
3291
3292 /***********************************************************************
3293 * GetProcessWorkingSetSize (KERNEL32.@)
3294 */
3295 BOOL WINAPI GetProcessWorkingSetSize(HANDLE hProcess, PSIZE_T minset,
3296 PSIZE_T maxset)
3297 {
3298 FIXME("(%p,%p,%p): stub\n",hProcess,minset,maxset);
3299 /* 32 MB working set size */
3300 if (minset) *minset = 32*1024*1024;
3301 if (maxset) *maxset = 32*1024*1024;
3302 return TRUE;
3303 }
3304
3305
3306 /***********************************************************************
3307 * SetProcessShutdownParameters (KERNEL32.@)
3308 */
3309 BOOL WINAPI SetProcessShutdownParameters(DWORD level, DWORD flags)
3310 {
3311 FIXME("(%08x, %08x): partial stub.\n", level, flags);
3312 shutdown_flags = flags;
3313 shutdown_priority = level;
3314 return TRUE;
3315 }
3316
3317
3318 /***********************************************************************
3319 * GetProcessShutdownParameters (KERNEL32.@)
3320 *
3321 */
3322 BOOL WINAPI GetProcessShutdownParameters( LPDWORD lpdwLevel, LPDWORD lpdwFlags )
3323 {
3324 *lpdwLevel = shutdown_priority;
3325 *lpdwFlags = shutdown_flags;
3326 return TRUE;
3327 }
3328
3329
3330 /***********************************************************************
3331 * GetProcessPriorityBoost (KERNEL32.@)
3332 */
3333 BOOL WINAPI GetProcessPriorityBoost(HANDLE hprocess,PBOOL pDisablePriorityBoost)
3334 {
3335 FIXME("(%p,%p): semi-stub\n", hprocess, pDisablePriorityBoost);
3336
3337 /* Report that no boost is present.. */
3338 *pDisablePriorityBoost = FALSE;
3339
3340 return TRUE;
3341 }
3342
3343 /***********************************************************************
3344 * SetProcessPriorityBoost (KERNEL32.@)
3345 */
3346 BOOL WINAPI SetProcessPriorityBoost(HANDLE hprocess,BOOL disableboost)
3347 {
3348 FIXME("(%p,%d): stub\n",hprocess,disableboost);
3349 /* Say we can do it. I doubt the program will notice that we don't. */
3350 return TRUE;
3351 }
3352
3353
3354 /***********************************************************************
3355 * ReadProcessMemory (KERNEL32.@)
3356 */
3357 BOOL WINAPI ReadProcessMemory( HANDLE process, LPCVOID addr, LPVOID buffer, SIZE_T size,
3358 SIZE_T *bytes_read )
3359 {
3360 NTSTATUS status = NtReadVirtualMemory( process, addr, buffer, size, bytes_read );
3361 if (status) SetLastError( RtlNtStatusToDosError(status) );
3362 return !status;
3363 }
3364
3365
3366 /***********************************************************************
3367 * WriteProcessMemory (KERNEL32.@)
3368 */
3369 BOOL WINAPI WriteProcessMemory( HANDLE process, LPVOID addr, LPCVOID buffer, SIZE_T size,
3370 SIZE_T *bytes_written )
3371 {
3372 NTSTATUS status = NtWriteVirtualMemory( process, addr, buffer, size, bytes_written );
3373 if (status) SetLastError( RtlNtStatusToDosError(status) );
3374 return !status;
3375 }
3376
3377
3378 /****************************************************************************
3379 * FlushInstructionCache (KERNEL32.@)
3380 */
3381 BOOL WINAPI FlushInstructionCache(HANDLE hProcess, LPCVOID lpBaseAddress, SIZE_T dwSize)
3382 {
3383 NTSTATUS status;
3384 status = NtFlushInstructionCache( hProcess, lpBaseAddress, dwSize );
3385 if (status) SetLastError( RtlNtStatusToDosError(status) );
3386 return !status;
3387 }
3388
3389
3390 /******************************************************************
3391 * GetProcessIoCounters (KERNEL32.@)
3392 */
3393 BOOL WINAPI GetProcessIoCounters(HANDLE hProcess, PIO_COUNTERS ioc)
3394 {
3395 NTSTATUS status;
3396
3397 status = NtQueryInformationProcess(hProcess, ProcessIoCounters,
3398 ioc, sizeof(*ioc), NULL);
3399 if (status) SetLastError( RtlNtStatusToDosError(status) );
3400 return !status;
3401 }
3402
3403 /******************************************************************
3404 * GetProcessHandleCount (KERNEL32.@)
3405 */
3406 BOOL WINAPI GetProcessHandleCount(HANDLE hProcess, DWORD *cnt)
3407 {
3408 NTSTATUS status;
3409
3410 status = NtQueryInformationProcess(hProcess, ProcessHandleCount,
3411 cnt, sizeof(*cnt), NULL);
3412 if (status) SetLastError( RtlNtStatusToDosError(status) );
3413 return !status;
3414 }
3415
3416 /******************************************************************
3417 * QueryFullProcessImageNameA (KERNEL32.@)
3418 */
3419 BOOL WINAPI QueryFullProcessImageNameA(HANDLE hProcess, DWORD dwFlags, LPSTR lpExeName, PDWORD pdwSize)
3420 {
3421 BOOL retval;
3422 DWORD pdwSizeW = *pdwSize;
3423 LPWSTR lpExeNameW = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, *pdwSize * sizeof(WCHAR));
3424
3425 retval = QueryFullProcessImageNameW(hProcess, dwFlags, lpExeNameW, &pdwSizeW);
3426
3427 if(retval)
3428 retval = (0 != WideCharToMultiByte(CP_ACP, 0, lpExeNameW, -1,
3429 lpExeName, *pdwSize, NULL, NULL));
3430 if(retval)
3431 *pdwSize = strlen(lpExeName);
3432
3433 HeapFree(GetProcessHeap(), 0, lpExeNameW);
3434 return retval;
3435 }
3436
3437 /******************************************************************
3438 * QueryFullProcessImageNameW (KERNEL32.@)
3439 */
3440 BOOL WINAPI QueryFullProcessImageNameW(HANDLE hProcess, DWORD dwFlags, LPWSTR lpExeName, PDWORD pdwSize)
3441 {
3442 BYTE buffer[sizeof(UNICODE_STRING) + MAX_PATH*sizeof(WCHAR)]; /* this buffer should be enough */
3443 UNICODE_STRING *dynamic_buffer = NULL;
3444 UNICODE_STRING *result = NULL;
3445 NTSTATUS status;
3446 DWORD needed;
3447
3448 /* FIXME: On Windows, ProcessImageFileName return an NT path. In Wine it
3449 * is a DOS path and we depend on this. */
3450 status = NtQueryInformationProcess(hProcess, ProcessImageFileName, buffer,
3451 sizeof(buffer) - sizeof(WCHAR), &needed);
3452 if (status == STATUS_INFO_LENGTH_MISMATCH)
3453 {
3454 dynamic_buffer = HeapAlloc(GetProcessHeap(), 0, needed + sizeof(WCHAR));
3455 status = NtQueryInformationProcess(hProcess, ProcessImageFileName, (LPBYTE)dynamic_buffer, needed, &needed);
3456 result = dynamic_buffer;
3457 }
3458 else
3459 result = (PUNICODE_STRING)buffer;
3460
3461 if (status) goto cleanup;
3462
3463 if (dwFlags & PROCESS_NAME_NATIVE)
3464 {
3465 WCHAR drive[3];
3466 WCHAR device[1024];
3467 DWORD ntlen, devlen;
3468
3469 if (result->Buffer[1] != ':' || result->Buffer[0] < 'A' || result->Buffer[0] > 'Z')
3470 {
3471 /* We cannot convert it to an NT device path so fail */
3472 status = STATUS_NO_SUCH_DEVICE;
3473 goto cleanup;
3474 }
3475
3476 /* Find this drive's NT device path */
3477 drive[0] = result->Buffer[0];
3478 drive[1] = ':';
3479 drive[2] = 0;
3480 if (!QueryDosDeviceW(drive, device, sizeof(device)/sizeof(*device)))
3481 {
3482 status = STATUS_NO_SUCH_DEVICE;
3483 goto cleanup;
3484 }
3485
3486 devlen = lstrlenW(device);
3487 ntlen = devlen + (result->Length/sizeof(WCHAR) - 2);
3488 if (ntlen + 1 > *pdwSize)
3489 {
3490 SetLastError(ERROR_INSUFFICIENT_BUFFER);
3491 return 0;
3492 }
3493 *pdwSize = ntlen;
3494
3495 memcpy(lpExeName, device, devlen * sizeof(*device));
3496 memcpy(lpExeName + devlen, result->Buffer + 2, result->Length - 2 * sizeof(WCHAR));
3497 lpExeName[*pdwSize] = 0;
3498 TRACE("NT path: %s\n", debugstr_w(lpExeName));
3499 }
3500 else
3501 {
3502 if (result->Length/sizeof(WCHAR) + 1 > *pdwSize)
3503 {
3504 status = STATUS_BUFFER_TOO_SMALL;
3505 goto cleanup;
3506 }
3507
3508 *pdwSize = result->Length/sizeof(WCHAR);
3509 memcpy( lpExeName, result->Buffer, result->Length );
3510 lpExeName[*pdwSize] = 0;
3511 }
3512
3513 cleanup:
3514 HeapFree(GetProcessHeap(), 0, dynamic_buffer);
3515 if (status) SetLastError( RtlNtStatusToDosError(status) );
3516 return !status;
3517 }
3518
3519 /***********************************************************************
3520 * K32GetProcessImageFileNameA (KERNEL32.@)
3521 */
3522 DWORD WINAPI K32GetProcessImageFileNameA( HANDLE process, LPSTR file, DWORD size )
3523 {
3524 return QueryFullProcessImageNameA(process, PROCESS_NAME_NATIVE, file, &size) ? size : 0;
3525 }
3526
3527 /***********************************************************************
3528 * K32GetProcessImageFileNameW (KERNEL32.@)
3529 */
3530 DWORD WINAPI K32GetProcessImageFileNameW( HANDLE process, LPWSTR file, DWORD size )
3531 {
3532 return QueryFullProcessImageNameW(process, PROCESS_NAME_NATIVE, file, &size) ? size : 0;
3533 }
3534
3535 /***********************************************************************
3536 * K32EnumProcesses (KERNEL32.@)
3537 */
3538 BOOL WINAPI K32EnumProcesses(DWORD *lpdwProcessIDs, DWORD cb, DWORD *lpcbUsed)
3539 {
3540 SYSTEM_PROCESS_INFORMATION *spi;
3541 ULONG size = 0x4000;
3542 void *buf = NULL;
3543 NTSTATUS status;
3544
3545 do {
3546 size *= 2;
3547 HeapFree(GetProcessHeap(), 0, buf);
3548 buf = HeapAlloc(GetProcessHeap(), 0, size);
3549 if (!buf)
3550 return FALSE;
3551
3552 status = NtQuerySystemInformation(SystemProcessInformation, buf, size, NULL);
3553 } while(status == STATUS_INFO_LENGTH_MISMATCH);
3554
3555 if (status != STATUS_SUCCESS)
3556 {
3557 HeapFree(GetProcessHeap(), 0, buf);
3558 SetLastError(RtlNtStatusToDosError(status));
3559 return FALSE;
3560 }
3561
3562 spi = buf;
3563
3564 for (*lpcbUsed = 0; cb >= sizeof(DWORD); cb -= sizeof(DWORD))
3565 {
3566 *lpdwProcessIDs++ = HandleToUlong(spi->UniqueProcessId);
3567 *lpcbUsed += sizeof(DWORD);
3568
3569 if (spi->NextEntryOffset == 0)
3570 break;
3571
3572 spi = (SYSTEM_PROCESS_INFORMATION *)(((PCHAR)spi) + spi->NextEntryOffset);
3573 }
3574
3575 HeapFree(GetProcessHeap(), 0, buf);
3576 return TRUE;
3577 }
3578
3579 /***********************************************************************
3580 * K32QueryWorkingSet (KERNEL32.@)
3581 */
3582 BOOL WINAPI K32QueryWorkingSet( HANDLE process, LPVOID buffer, DWORD size )
3583 {
3584 NTSTATUS status;
3585
3586 TRACE( "(%p, %p, %d)\n", process, buffer, size );
3587
3588 status = NtQueryVirtualMemory( process, NULL, MemoryWorkingSetList, buffer, size, NULL );
3589
3590 if (status)
3591 {
3592 SetLastError( RtlNtStatusToDosError( status ) );
3593 return FALSE;
3594 }
3595 return TRUE;
3596 }
3597
3598 /***********************************************************************
3599 * K32QueryWorkingSetEx (KERNEL32.@)
3600 */
3601 BOOL WINAPI K32QueryWorkingSetEx( HANDLE process, LPVOID buffer, DWORD size )
3602 {
3603 NTSTATUS status;
3604
3605 TRACE( "(%p, %p, %d)\n", process, buffer, size );
3606
3607 status = NtQueryVirtualMemory( process, NULL, MemoryWorkingSetList, buffer, size, NULL );
3608
3609 if (status)
3610 {
3611 SetLastError( RtlNtStatusToDosError( status ) );
3612 return FALSE;
3613 }
3614 return TRUE;
3615 }
3616
3617 /***********************************************************************
3618 * K32GetProcessMemoryInfo (KERNEL32.@)
3619 *
3620 * Retrieve memory usage information for a given process
3621 *
3622 */
3623 BOOL WINAPI K32GetProcessMemoryInfo(HANDLE process,
3624 PPROCESS_MEMORY_COUNTERS pmc, DWORD cb)
3625 {
3626 NTSTATUS status;
3627 VM_COUNTERS vmc;
3628
3629 if (cb < sizeof(PROCESS_MEMORY_COUNTERS))
3630 {
3631 SetLastError(ERROR_INSUFFICIENT_BUFFER);
3632 return FALSE;
3633 }
3634
3635 status = NtQueryInformationProcess(process, ProcessVmCounters,
3636 &vmc, sizeof(vmc), NULL);
3637
3638 if (status)
3639 {
3640 SetLastError(RtlNtStatusToDosError(status));
3641 return FALSE;
3642 }
3643
3644 pmc->cb = sizeof(PROCESS_MEMORY_COUNTERS);
3645 pmc->PageFaultCount = vmc.PageFaultCount;
3646 pmc->PeakWorkingSetSize = vmc.PeakWorkingSetSize;
3647 pmc->WorkingSetSize = vmc.WorkingSetSize;
3648 pmc->QuotaPeakPagedPoolUsage = vmc.QuotaPeakPagedPoolUsage;
3649 pmc->QuotaPagedPoolUsage = vmc.QuotaPagedPoolUsage;
3650 pmc->QuotaPeakNonPagedPoolUsage = vmc.QuotaPeakNonPagedPoolUsage;
3651 pmc->QuotaNonPagedPoolUsage = vmc.QuotaNonPagedPoolUsage;
3652 pmc->PagefileUsage = vmc.PagefileUsage;
3653 pmc->PeakPagefileUsage = vmc.PeakPagefileUsage;
3654
3655 return TRUE;
3656 }
3657
3658 /***********************************************************************
3659 * ProcessIdToSessionId (KERNEL32.@)
3660 * This function is available on Terminal Server 4SP4 and Windows 2000
3661 */
3662 BOOL WINAPI ProcessIdToSessionId( DWORD procid, DWORD *sessionid_ptr )
3663 {
3664 /* According to MSDN, if the calling process is not in a terminal
3665 * services environment, then the sessionid returned is zero.
3666 */
3667 *sessionid_ptr = 0;
3668 return TRUE;
3669 }
3670
3671
3672 /***********************************************************************
3673 * RegisterServiceProcess (KERNEL32.@)
3674 *
3675 * A service process calls this function to ensure that it continues to run
3676 * even after a user logged off.
3677 */
3678 DWORD WINAPI RegisterServiceProcess(DWORD dwProcessId, DWORD dwType)
3679 {
3680 /* I don't think that Wine needs to do anything in this function */
3681 return 1; /* success */
3682 }
3683
3684
3685 /**********************************************************************
3686 * IsWow64Process (KERNEL32.@)
3687 */
3688 BOOL WINAPI IsWow64Process(HANDLE hProcess, PBOOL Wow64Process)
3689 {
3690 ULONG pbi;
3691 NTSTATUS status;
3692
3693 status = NtQueryInformationProcess( hProcess, ProcessWow64Information, &pbi, sizeof(pbi), NULL );
3694
3695 if (status != STATUS_SUCCESS)
3696 {
3697 SetLastError( RtlNtStatusToDosError( status ) );
3698 return FALSE;
3699 }
3700 *Wow64Process = (pbi != 0);
3701 return TRUE;
3702 }
3703
3704
3705 /***********************************************************************
3706 * GetCurrentProcess (KERNEL32.@)
3707 *
3708 * Get a handle to the current process.
3709 *
3710 * PARAMS
3711 * None.
3712 *
3713 * RETURNS
3714 * A handle representing the current process.
3715 */
3716 #undef GetCurrentProcess
3717 HANDLE WINAPI GetCurrentProcess(void)
3718 {
3719 return (HANDLE)~(ULONG_PTR)0;
3720 }
3721
3722 /***********************************************************************
3723 * GetLogicalProcessorInformation (KERNEL32.@)
3724 */
3725 BOOL WINAPI GetLogicalProcessorInformation(PSYSTEM_LOGICAL_PROCESSOR_INFORMATION buffer, PDWORD pBufLen)
3726 {
3727 NTSTATUS status;
3728
3729 TRACE("(%p,%p)\n", buffer, pBufLen);
3730
3731 if(!pBufLen)
3732 {
3733 SetLastError(ERROR_INVALID_PARAMETER);
3734 return FALSE;
3735 }
3736
3737 status = NtQuerySystemInformation( SystemLogicalProcessorInformation, buffer, *pBufLen, pBufLen);
3738
3739 if (status == STATUS_INFO_LENGTH_MISMATCH)
3740 {
3741 SetLastError( ERROR_INSUFFICIENT_BUFFER );
3742 return FALSE;
3743 }
3744 if (status != STATUS_SUCCESS)
3745 {
3746 SetLastError( RtlNtStatusToDosError( status ) );
3747 return FALSE;
3748 }
3749 return TRUE;
3750 }
3751
3752 /***********************************************************************
3753 * GetLogicalProcessorInformationEx (KERNEL32.@)
3754 */
3755 BOOL WINAPI GetLogicalProcessorInformationEx(LOGICAL_PROCESSOR_RELATIONSHIP relationship, PSYSTEM_LOGICAL_PROCESSOR_INFORMATION_EX buffer, PDWORD pBufLen)
3756 {
3757 FIXME("(%u,%p,%p): stub\n", relationship, buffer, pBufLen);
3758 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3759 return FALSE;
3760 }
3761
3762 /***********************************************************************
3763 * CmdBatNotification (KERNEL32.@)
3764 *
3765 * Notifies the system that a batch file has started or finished.
3766 *
3767 * PARAMS
3768 * bBatchRunning [I] TRUE if a batch file has started or
3769 * FALSE if a batch file has finished executing.
3770 *
3771 * RETURNS
3772 * Unknown.
3773 */
3774 BOOL WINAPI CmdBatNotification( BOOL bBatchRunning )
3775 {
3776 FIXME("%d\n", bBatchRunning);
3777 return FALSE;
3778 }
3779
3780
3781 /***********************************************************************
3782 * RegisterApplicationRestart (KERNEL32.@)
3783 */
3784 HRESULT WINAPI RegisterApplicationRestart(PCWSTR pwzCommandLine, DWORD dwFlags)
3785 {
3786 FIXME("(%s,%d)\n", debugstr_w(pwzCommandLine), dwFlags);
3787
3788 return S_OK;
3789 }
3790
3791 /**********************************************************************
3792 * WTSGetActiveConsoleSessionId (KERNEL32.@)
3793 */
3794 DWORD WINAPI WTSGetActiveConsoleSessionId(void)
3795 {
3796 FIXME("stub\n");
3797 return 0;
3798 }
3799
3800 /**********************************************************************
3801 * GetSystemDEPPolicy (KERNEL32.@)
3802 */
3803 DEP_SYSTEM_POLICY_TYPE WINAPI GetSystemDEPPolicy(void)
3804 {
3805 FIXME("stub\n");
3806 return OptIn;
3807 }
3808
3809 /**********************************************************************
3810 * SetProcessDEPPolicy (KERNEL32.@)
3811 */
3812 BOOL WINAPI SetProcessDEPPolicy(DWORD newDEP)
3813 {
3814 FIXME("(%d): stub\n", newDEP);
3815 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3816 return FALSE;
3817 }
3818
3819 /**********************************************************************
3820 * ApplicationRecoveryFinished (KERNEL32.@)
3821 */
3822 VOID WINAPI ApplicationRecoveryFinished(BOOL success)
3823 {
3824 FIXME(": stub\n");
3825 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3826 }
3827
3828 /**********************************************************************
3829 * ApplicationRecoveryInProgress (KERNEL32.@)
3830 */
3831 HRESULT WINAPI ApplicationRecoveryInProgress(PBOOL canceled)
3832 {
3833 FIXME(":%p stub\n", canceled);
3834 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3835 return E_FAIL;
3836 }
3837
3838 /**********************************************************************
3839 * RegisterApplicationRecoveryCallback (KERNEL32.@)
3840 */
3841 HRESULT WINAPI RegisterApplicationRecoveryCallback(APPLICATION_RECOVERY_CALLBACK callback, PVOID param, DWORD pingint, DWORD flags)
3842 {
3843 FIXME("%p, %p, %d, %d: stub\n", callback, param, pingint, flags);
3844 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3845 return E_FAIL;
3846 }
3847
3848 /**********************************************************************
3849 * GetNumaHighestNodeNumber (KERNEL32.@)
3850 */
3851 BOOL WINAPI GetNumaHighestNodeNumber(PULONG highestnode)
3852 {
3853 FIXME("(%p): stub\n", highestnode);
3854 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3855 return FALSE;
3856 }
3857
3858 /**********************************************************************
3859 * GetNumaNodeProcessorMask (KERNEL32.@)
3860 */
3861 BOOL WINAPI GetNumaNodeProcessorMask(UCHAR node, PULONGLONG mask)
3862 {
3863 FIXME("(%c %p): stub\n", node, mask);
3864 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3865 return FALSE;
3866 }
3867
3868 /**********************************************************************
3869 * GetNumaAvailableMemoryNode (KERNEL32.@)
3870 */
3871 BOOL WINAPI GetNumaAvailableMemoryNode(UCHAR node, PULONGLONG available_bytes)
3872 {
3873 FIXME("(%c %p): stub\n", node, available_bytes);
3874 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3875 return FALSE;
3876 }
3877
3878 /**********************************************************************
3879 * GetProcessDEPPolicy (KERNEL32.@)
3880 */
3881 BOOL WINAPI GetProcessDEPPolicy(HANDLE process, LPDWORD flags, PBOOL permanent)
3882 {
3883 FIXME("(%p %p %p): stub\n", process, flags, permanent);
3884 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
3885 return FALSE;
3886 }
3887
3888 /**********************************************************************
3889 * FlushProcessWriteBuffers (KERNEL32.@)
3890 */
3891 VOID WINAPI FlushProcessWriteBuffers(void)
3892 {
3893 static int once = 0;
3894
3895 if (!once++)
3896 FIXME(": stub\n");
3897 }
Maarten's multimedia branch