kernel32/tests: Add tests for FillConsoleOutputCharacterA.
[wine] / dlls / ntdll / process.c
1 /*
2  * NT basis DLL
3  *
4  * This file contains the Nt* API functions of NTDLL.DLL.
5  * In the original ntdll.dll they all seem to just call int 0x2e (down to the NTOSKRNL)
6  *
7  * Copyright 1996-1998 Marcus Meissner
8  *
9  * This library is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * This library is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with this library; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
22  */
23
24 #include <stdarg.h>
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <time.h>
29
30 #include "ntstatus.h"
31 #define WIN32_NO_STATUS
32 #include "wine/debug.h"
33 #include "windef.h"
34 #include "winternl.h"
35 #include "ntdll_misc.h"
36 #include "wine/server.h"
37
38 WINE_DEFAULT_DEBUG_CHANNEL(ntdll);
39
40 static ULONG execute_flags = MEM_EXECUTE_OPTION_DISABLE;
41
42 /*
43  *      Process object
44  */
45
46 /******************************************************************************
47  *  NtTerminateProcess                  [NTDLL.@]
48  *
49  *  Native applications must kill themselves when done
50  */
51 NTSTATUS WINAPI NtTerminateProcess( HANDLE handle, LONG exit_code )
52 {
53     NTSTATUS ret;
54     BOOL self;
55     SERVER_START_REQ( terminate_process )
56     {
57         req->handle    = wine_server_obj_handle( handle );
58         req->exit_code = exit_code;
59         ret = wine_server_call( req );
60         self = !ret && reply->self;
61     }
62     SERVER_END_REQ;
63     if (self) exit( exit_code );
64     return ret;
65 }
66
67 /******************************************************************************
68  *  RtlGetCurrentPeb  [NTDLL.@]
69  *
70  */
71 PEB * WINAPI RtlGetCurrentPeb(void)
72 {
73     return NtCurrentTeb()->Peb;
74 }
75
76 /***********************************************************************
77  *           __wine_make_process_system   (NTDLL.@)
78  *
79  * Mark the current process as a system process.
80  * Returns the event that is signaled when all non-system processes have exited.
81  */
82 HANDLE CDECL __wine_make_process_system(void)
83 {
84     HANDLE ret = 0;
85     SERVER_START_REQ( make_process_system )
86     {
87         if (!wine_server_call( req )) ret = wine_server_ptr_handle( reply->event );
88     }
89     SERVER_END_REQ;
90     return ret;
91 }
92
93 static UINT process_error_mode;
94
95 #define UNIMPLEMENTED_INFO_CLASS(c) \
96     case c: \
97         FIXME("(process=%p) Unimplemented information class: " #c "\n", ProcessHandle); \
98         ret = STATUS_INVALID_INFO_CLASS; \
99         break
100
101 /******************************************************************************
102 *  NtQueryInformationProcess            [NTDLL.@]
103 *  ZwQueryInformationProcess            [NTDLL.@]
104 *
105 */
106 NTSTATUS WINAPI NtQueryInformationProcess(
107         IN HANDLE ProcessHandle,
108         IN PROCESSINFOCLASS ProcessInformationClass,
109         OUT PVOID ProcessInformation,
110         IN ULONG ProcessInformationLength,
111         OUT PULONG ReturnLength)
112 {
113     NTSTATUS ret = STATUS_SUCCESS;
114     ULONG len = 0;
115
116     TRACE("(%p,0x%08x,%p,0x%08x,%p)\n",
117           ProcessHandle,ProcessInformationClass,
118           ProcessInformation,ProcessInformationLength,
119           ReturnLength);
120
121     switch (ProcessInformationClass) 
122     {
123     UNIMPLEMENTED_INFO_CLASS(ProcessQuotaLimits);
124     UNIMPLEMENTED_INFO_CLASS(ProcessBasePriority);
125     UNIMPLEMENTED_INFO_CLASS(ProcessRaisePriority);
126     UNIMPLEMENTED_INFO_CLASS(ProcessExceptionPort);
127     UNIMPLEMENTED_INFO_CLASS(ProcessAccessToken);
128     UNIMPLEMENTED_INFO_CLASS(ProcessLdtInformation);
129     UNIMPLEMENTED_INFO_CLASS(ProcessLdtSize);
130     UNIMPLEMENTED_INFO_CLASS(ProcessIoPortHandlers);
131     UNIMPLEMENTED_INFO_CLASS(ProcessPooledUsageAndLimits);
132     UNIMPLEMENTED_INFO_CLASS(ProcessWorkingSetWatch);
133     UNIMPLEMENTED_INFO_CLASS(ProcessUserModeIOPL);
134     UNIMPLEMENTED_INFO_CLASS(ProcessEnableAlignmentFaultFixup);
135     UNIMPLEMENTED_INFO_CLASS(ProcessPriorityClass);
136     UNIMPLEMENTED_INFO_CLASS(ProcessWx86Information);
137     UNIMPLEMENTED_INFO_CLASS(ProcessAffinityMask);
138     UNIMPLEMENTED_INFO_CLASS(ProcessPriorityBoost);
139     UNIMPLEMENTED_INFO_CLASS(ProcessDeviceMap);
140     UNIMPLEMENTED_INFO_CLASS(ProcessSessionInformation);
141     UNIMPLEMENTED_INFO_CLASS(ProcessForegroundInformation);
142     UNIMPLEMENTED_INFO_CLASS(ProcessLUIDDeviceMapsEnabled);
143     UNIMPLEMENTED_INFO_CLASS(ProcessBreakOnTermination);
144     UNIMPLEMENTED_INFO_CLASS(ProcessDebugFlags);
145     UNIMPLEMENTED_INFO_CLASS(ProcessHandleTracing);
146
147     case ProcessBasicInformation:
148         {
149             PROCESS_BASIC_INFORMATION pbi;
150             const ULONG_PTR affinity_mask = ((ULONG_PTR)1 << NtCurrentTeb()->Peb->NumberOfProcessors) - 1;
151
152             if (ProcessInformationLength >= sizeof(PROCESS_BASIC_INFORMATION))
153             {
154                 if (!ProcessInformation)
155                     ret = STATUS_ACCESS_VIOLATION;
156                 else if (!ProcessHandle)
157                     ret = STATUS_INVALID_HANDLE;
158                 else
159                 {
160                     SERVER_START_REQ(get_process_info)
161                     {
162                         req->handle = wine_server_obj_handle( ProcessHandle );
163                         if ((ret = wine_server_call( req )) == STATUS_SUCCESS)
164                         {
165                             pbi.ExitStatus = reply->exit_code;
166                             pbi.PebBaseAddress = wine_server_get_ptr( reply->peb );
167                             pbi.AffinityMask = reply->affinity & affinity_mask;
168                             pbi.BasePriority = reply->priority;
169                             pbi.UniqueProcessId = reply->pid;
170                             pbi.InheritedFromUniqueProcessId = reply->ppid;
171                         }
172                     }
173                     SERVER_END_REQ;
174
175                     memcpy(ProcessInformation, &pbi, sizeof(PROCESS_BASIC_INFORMATION));
176
177                     len = sizeof(PROCESS_BASIC_INFORMATION);
178                 }
179
180                 if (ProcessInformationLength > sizeof(PROCESS_BASIC_INFORMATION))
181                     ret = STATUS_INFO_LENGTH_MISMATCH;
182             }
183             else
184             {
185                 len = sizeof(PROCESS_BASIC_INFORMATION);
186                 ret = STATUS_INFO_LENGTH_MISMATCH;
187             }
188         }
189         break;
190     case ProcessIoCounters:
191         {
192             IO_COUNTERS pii;
193
194             if (ProcessInformationLength >= sizeof(IO_COUNTERS))
195             {
196                 if (!ProcessInformation)
197                     ret = STATUS_ACCESS_VIOLATION;
198                 else if (!ProcessHandle)
199                     ret = STATUS_INVALID_HANDLE;
200                 else
201                 {
202                     /* FIXME : real data */
203                     memset(&pii, 0 , sizeof(IO_COUNTERS));
204
205                     memcpy(ProcessInformation, &pii, sizeof(IO_COUNTERS));
206
207                     len = sizeof(IO_COUNTERS);
208                 }
209
210                 if (ProcessInformationLength > sizeof(IO_COUNTERS))
211                     ret = STATUS_INFO_LENGTH_MISMATCH;
212             }
213             else
214             {
215                 len = sizeof(IO_COUNTERS);
216                 ret = STATUS_INFO_LENGTH_MISMATCH;
217             }
218         }
219         break;
220     case ProcessVmCounters:
221         {
222             VM_COUNTERS pvmi;
223
224             /* older Windows versions don't have the PrivatePageCount field */
225             if (ProcessInformationLength >= FIELD_OFFSET(VM_COUNTERS,PrivatePageCount))
226             {
227                 if (!ProcessInformation)
228                     ret = STATUS_ACCESS_VIOLATION;
229                 else if (!ProcessHandle)
230                     ret = STATUS_INVALID_HANDLE;
231                 else
232                 {
233                     /* FIXME : real data */
234                     memset(&pvmi, 0 , sizeof(VM_COUNTERS));
235
236                     len = ProcessInformationLength;
237                     if (len != FIELD_OFFSET(VM_COUNTERS,PrivatePageCount)) len = sizeof(VM_COUNTERS);
238
239                     memcpy(ProcessInformation, &pvmi, min(ProcessInformationLength,sizeof(VM_COUNTERS)));
240                 }
241
242                 if (ProcessInformationLength != FIELD_OFFSET(VM_COUNTERS,PrivatePageCount) &&
243                     ProcessInformationLength != sizeof(VM_COUNTERS))
244                     ret = STATUS_INFO_LENGTH_MISMATCH;
245             }
246             else
247             {
248                 len = sizeof(pvmi);
249                 ret = STATUS_INFO_LENGTH_MISMATCH;
250             }
251         }
252         break;
253     case ProcessTimes:
254         {
255             KERNEL_USER_TIMES pti;
256
257             if (ProcessInformationLength >= sizeof(KERNEL_USER_TIMES))
258             {
259                 if (!ProcessInformation)
260                     ret = STATUS_ACCESS_VIOLATION;
261                 else if (!ProcessHandle)
262                     ret = STATUS_INVALID_HANDLE;
263                 else
264                 {
265                     /* FIXME : User- and KernelTime have to be implemented */
266                     memset(&pti, 0, sizeof(KERNEL_USER_TIMES));
267
268                     SERVER_START_REQ(get_process_info)
269                     {
270                       req->handle = wine_server_obj_handle( ProcessHandle );
271                       if ((ret = wine_server_call( req )) == STATUS_SUCCESS)
272                       {
273                           pti.CreateTime.QuadPart = reply->start_time;
274                           pti.ExitTime.QuadPart = reply->end_time;
275                       }
276                     }
277                     SERVER_END_REQ;
278
279                     memcpy(ProcessInformation, &pti, sizeof(KERNEL_USER_TIMES));
280                     len = sizeof(KERNEL_USER_TIMES);
281                 }
282
283                 if (ProcessInformationLength > sizeof(KERNEL_USER_TIMES))
284                     ret = STATUS_INFO_LENGTH_MISMATCH;
285             }
286             else
287             {
288                 len = sizeof(KERNEL_USER_TIMES);
289                 ret = STATUS_INFO_LENGTH_MISMATCH;
290             }
291         }
292         break;
293     case ProcessDebugPort:
294         len = sizeof(DWORD_PTR);
295         if (ProcessInformationLength == len)
296         {
297             if (!ProcessInformation)
298                 ret = STATUS_ACCESS_VIOLATION;
299             else if (!ProcessHandle)
300                 ret = STATUS_INVALID_HANDLE;
301             else
302             {
303                 SERVER_START_REQ(get_process_info)
304                 {
305                     req->handle = wine_server_obj_handle( ProcessHandle );
306                     if ((ret = wine_server_call( req )) == STATUS_SUCCESS)
307                     {
308                         *(DWORD_PTR *)ProcessInformation = reply->debugger_present ? ~(DWORD_PTR)0 : 0;
309                     }
310                 }
311                 SERVER_END_REQ;
312             }
313         }
314         else
315             ret = STATUS_INFO_LENGTH_MISMATCH;
316         break;
317     case ProcessDefaultHardErrorMode:
318         len = sizeof(process_error_mode);
319         if (ProcessInformationLength == len)
320             memcpy(ProcessInformation, &process_error_mode, len);
321         else
322             ret = STATUS_INFO_LENGTH_MISMATCH;
323         break;
324     case ProcessDebugObjectHandle:
325         /* "These are not the debuggers you are looking for." *
326          * set it to 0 aka "no debugger" to satisfy copy protections */
327         len = sizeof(HANDLE);
328         if (ProcessInformationLength == len)
329         {
330             if (!ProcessInformation)
331                 ret = STATUS_ACCESS_VIOLATION;
332             else if (!ProcessHandle)
333                 ret = STATUS_INVALID_HANDLE;
334             else
335             {
336                 memset(ProcessInformation, 0, ProcessInformationLength);
337                 ret = STATUS_PORT_NOT_SET;
338             }
339         }
340         else
341             ret = STATUS_INFO_LENGTH_MISMATCH;
342         break;
343     case ProcessHandleCount:
344         if (ProcessInformationLength >= 4)
345         {
346             if (!ProcessInformation)
347                 ret = STATUS_ACCESS_VIOLATION;
348             else if (!ProcessHandle)
349                 ret = STATUS_INVALID_HANDLE;
350             else
351             {
352                 memset(ProcessInformation, 0, 4);
353                 len = 4;
354             }
355
356             if (ProcessInformationLength > 4)
357                 ret = STATUS_INFO_LENGTH_MISMATCH;
358         }
359         else
360         {
361             len = 4;
362             ret = STATUS_INFO_LENGTH_MISMATCH;
363         }
364         break;
365     case ProcessWow64Information:
366         len = sizeof(DWORD);
367         if (ProcessInformationLength == len)
368         {
369             DWORD val = 0;
370
371             if (ProcessHandle == GetCurrentProcess()) val = is_wow64;
372             else if (server_cpus & (1 << CPU_x86_64))
373             {
374                 SERVER_START_REQ( get_process_info )
375                 {
376                     req->handle = wine_server_obj_handle( ProcessHandle );
377                     if (!(ret = wine_server_call( req ))) val = (reply->cpu != CPU_x86_64);
378                 }
379                 SERVER_END_REQ;
380             }
381             *(DWORD *)ProcessInformation = val;
382         }
383         else ret = STATUS_INFO_LENGTH_MISMATCH;
384         break;
385     case ProcessImageFileName:
386         /* FIXME: this will return a DOS path. Windows returns an NT path. Changing this would require also changing kernel32.QueryFullProcessImageName.
387          * The latter may be harder because of the lack of RtlNtPathNameToDosPathName. */
388         SERVER_START_REQ(get_dll_info)
389         {
390             UNICODE_STRING *image_file_name_str = ProcessInformation;
391
392             req->handle = wine_server_obj_handle( ProcessHandle );
393             req->base_address = 0; /* main module */
394             wine_server_set_reply( req, image_file_name_str ? image_file_name_str + 1 : NULL,
395                                    ProcessInformationLength > sizeof(UNICODE_STRING) ? ProcessInformationLength - sizeof(UNICODE_STRING) : 0 );
396             ret = wine_server_call( req );
397             if (ret == STATUS_BUFFER_TOO_SMALL) ret = STATUS_INFO_LENGTH_MISMATCH;
398
399             len = sizeof(UNICODE_STRING) + reply->filename_len;
400             if (ret == STATUS_SUCCESS)
401             {
402                 image_file_name_str->MaximumLength = image_file_name_str->Length = reply->filename_len;
403                 image_file_name_str->Buffer = (PWSTR)(image_file_name_str + 1);
404             }
405         }
406         SERVER_END_REQ;
407         break;
408     case ProcessExecuteFlags:
409         len = sizeof(ULONG);
410         if (ProcessInformationLength == len)
411             *(ULONG *)ProcessInformation = execute_flags;
412         else
413             ret = STATUS_INFO_LENGTH_MISMATCH;
414         break;
415     default:
416         FIXME("(%p,info_class=%d,%p,0x%08x,%p) Unknown information class\n",
417               ProcessHandle,ProcessInformationClass,
418               ProcessInformation,ProcessInformationLength,
419               ReturnLength);
420         ret = STATUS_INVALID_INFO_CLASS;
421         break;
422     }
423
424     if (ReturnLength) *ReturnLength = len;
425     
426     return ret;
427 }
428
429 /******************************************************************************
430  * NtSetInformationProcess [NTDLL.@]
431  * ZwSetInformationProcess [NTDLL.@]
432  */
433 NTSTATUS WINAPI NtSetInformationProcess(
434         IN HANDLE ProcessHandle,
435         IN PROCESSINFOCLASS ProcessInformationClass,
436         IN PVOID ProcessInformation,
437         IN ULONG ProcessInformationLength)
438 {
439     NTSTATUS ret = STATUS_SUCCESS;
440
441     switch (ProcessInformationClass)
442     {
443     case ProcessDefaultHardErrorMode:
444         if (ProcessInformationLength != sizeof(UINT)) return STATUS_INVALID_PARAMETER;
445         process_error_mode = *(UINT *)ProcessInformation;
446         break;
447     case ProcessAffinityMask:
448         if (ProcessInformationLength != sizeof(DWORD_PTR)) return STATUS_INVALID_PARAMETER;
449         if (*(PDWORD_PTR)ProcessInformation & ~(((DWORD_PTR)1 << NtCurrentTeb()->Peb->NumberOfProcessors) - 1))
450             return STATUS_INVALID_PARAMETER;
451         if (!*(PDWORD_PTR)ProcessInformation)
452             return STATUS_INVALID_PARAMETER;
453         SERVER_START_REQ( set_process_info )
454         {
455             req->handle   = wine_server_obj_handle( ProcessHandle );
456             req->affinity = *(PDWORD_PTR)ProcessInformation;
457             req->mask     = SET_PROCESS_INFO_AFFINITY;
458             ret = wine_server_call( req );
459         }
460         SERVER_END_REQ;
461         break;
462     case ProcessPriorityClass:
463         if (ProcessInformationLength != sizeof(PROCESS_PRIORITY_CLASS))
464             return STATUS_INVALID_PARAMETER;
465         else
466         {
467             PROCESS_PRIORITY_CLASS* ppc = ProcessInformation;
468
469             SERVER_START_REQ( set_process_info )
470             {
471                 req->handle   = wine_server_obj_handle( ProcessHandle );
472                 /* FIXME Foreground isn't used */
473                 req->priority = ppc->PriorityClass;
474                 req->mask     = SET_PROCESS_INFO_PRIORITY;
475                 ret = wine_server_call( req );
476             }
477             SERVER_END_REQ;
478         }
479         break;
480
481     case ProcessExecuteFlags:
482         if (ProcessInformationLength != sizeof(ULONG))
483             return STATUS_INVALID_PARAMETER;
484         else if (execute_flags & MEM_EXECUTE_OPTION_PERMANENT)
485             return STATUS_ACCESS_DENIED;
486         else
487         {
488             BOOL enable;
489             switch (*(ULONG *)ProcessInformation & (MEM_EXECUTE_OPTION_ENABLE|MEM_EXECUTE_OPTION_DISABLE))
490             {
491             case MEM_EXECUTE_OPTION_ENABLE:
492                 enable = TRUE;
493                 break;
494             case MEM_EXECUTE_OPTION_DISABLE:
495                 enable = FALSE;
496                 break;
497             default:
498                 return STATUS_INVALID_PARAMETER;
499             }
500             execute_flags = *(ULONG *)ProcessInformation;
501             VIRTUAL_SetForceExec( enable );
502         }
503         break;
504
505     default:
506         FIXME("(%p,0x%08x,%p,0x%08x) stub\n",
507               ProcessHandle,ProcessInformationClass,ProcessInformation,
508               ProcessInformationLength);
509         ret = STATUS_NOT_IMPLEMENTED;
510         break;
511     }
512     return ret;
513 }
514
515 /******************************************************************************
516  * NtFlushInstructionCache [NTDLL.@]
517  * ZwFlushInstructionCache [NTDLL.@]
518  */
519 NTSTATUS WINAPI NtFlushInstructionCache(
520         IN HANDLE ProcessHandle,
521         IN LPCVOID BaseAddress,
522         IN SIZE_T Size)
523 {
524 #ifdef __i386__
525     TRACE("%p %p %ld - no-op on x86\n", ProcessHandle, BaseAddress, Size );
526 #else
527     FIXME("%p %p %ld\n", ProcessHandle, BaseAddress, Size );
528 #endif
529     return STATUS_SUCCESS;
530 }
531
532 /******************************************************************
533  *              NtOpenProcess [NTDLL.@]
534  *              ZwOpenProcess [NTDLL.@]
535  */
536 NTSTATUS  WINAPI NtOpenProcess(PHANDLE handle, ACCESS_MASK access,
537                                const OBJECT_ATTRIBUTES* attr, const CLIENT_ID* cid)
538 {
539     NTSTATUS    status;
540
541     SERVER_START_REQ( open_process )
542     {
543         req->pid        = HandleToULong(cid->UniqueProcess);
544         req->access     = access;
545         req->attributes = attr ? attr->Attributes : 0;
546         status = wine_server_call( req );
547         if (!status) *handle = wine_server_ptr_handle( reply->handle );
548     }
549     SERVER_END_REQ;
550     return status;
551 }