krnl386.exe: Fix size calculation in GetSystemDirectory16.
[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(ProcessPriorityBoost);
138     UNIMPLEMENTED_INFO_CLASS(ProcessDeviceMap);
139     UNIMPLEMENTED_INFO_CLASS(ProcessSessionInformation);
140     UNIMPLEMENTED_INFO_CLASS(ProcessForegroundInformation);
141     UNIMPLEMENTED_INFO_CLASS(ProcessLUIDDeviceMapsEnabled);
142     UNIMPLEMENTED_INFO_CLASS(ProcessBreakOnTermination);
143     UNIMPLEMENTED_INFO_CLASS(ProcessHandleTracing);
144
145     case ProcessBasicInformation:
146         {
147             PROCESS_BASIC_INFORMATION pbi;
148             const ULONG_PTR affinity_mask = ((ULONG_PTR)1 << NtCurrentTeb()->Peb->NumberOfProcessors) - 1;
149
150             if (ProcessInformationLength >= sizeof(PROCESS_BASIC_INFORMATION))
151             {
152                 if (!ProcessInformation)
153                     ret = STATUS_ACCESS_VIOLATION;
154                 else if (!ProcessHandle)
155                     ret = STATUS_INVALID_HANDLE;
156                 else
157                 {
158                     SERVER_START_REQ(get_process_info)
159                     {
160                         req->handle = wine_server_obj_handle( ProcessHandle );
161                         if ((ret = wine_server_call( req )) == STATUS_SUCCESS)
162                         {
163                             pbi.ExitStatus = reply->exit_code;
164                             pbi.PebBaseAddress = wine_server_get_ptr( reply->peb );
165                             pbi.AffinityMask = reply->affinity & affinity_mask;
166                             pbi.BasePriority = reply->priority;
167                             pbi.UniqueProcessId = reply->pid;
168                             pbi.InheritedFromUniqueProcessId = reply->ppid;
169                         }
170                     }
171                     SERVER_END_REQ;
172
173                     memcpy(ProcessInformation, &pbi, sizeof(PROCESS_BASIC_INFORMATION));
174
175                     len = sizeof(PROCESS_BASIC_INFORMATION);
176                 }
177
178                 if (ProcessInformationLength > sizeof(PROCESS_BASIC_INFORMATION))
179                     ret = STATUS_INFO_LENGTH_MISMATCH;
180             }
181             else
182             {
183                 len = sizeof(PROCESS_BASIC_INFORMATION);
184                 ret = STATUS_INFO_LENGTH_MISMATCH;
185             }
186         }
187         break;
188     case ProcessIoCounters:
189         {
190             IO_COUNTERS pii;
191
192             if (ProcessInformationLength >= sizeof(IO_COUNTERS))
193             {
194                 if (!ProcessInformation)
195                     ret = STATUS_ACCESS_VIOLATION;
196                 else if (!ProcessHandle)
197                     ret = STATUS_INVALID_HANDLE;
198                 else
199                 {
200                     /* FIXME : real data */
201                     memset(&pii, 0 , sizeof(IO_COUNTERS));
202
203                     memcpy(ProcessInformation, &pii, sizeof(IO_COUNTERS));
204
205                     len = sizeof(IO_COUNTERS);
206                 }
207
208                 if (ProcessInformationLength > sizeof(IO_COUNTERS))
209                     ret = STATUS_INFO_LENGTH_MISMATCH;
210             }
211             else
212             {
213                 len = sizeof(IO_COUNTERS);
214                 ret = STATUS_INFO_LENGTH_MISMATCH;
215             }
216         }
217         break;
218     case ProcessVmCounters:
219         {
220             VM_COUNTERS pvmi;
221
222             /* older Windows versions don't have the PrivatePageCount field */
223             if (ProcessInformationLength >= FIELD_OFFSET(VM_COUNTERS,PrivatePageCount))
224             {
225                 if (!ProcessInformation)
226                     ret = STATUS_ACCESS_VIOLATION;
227                 else if (!ProcessHandle)
228                     ret = STATUS_INVALID_HANDLE;
229                 else
230                 {
231                     /* FIXME : real data */
232                     memset(&pvmi, 0 , sizeof(VM_COUNTERS));
233
234                     len = ProcessInformationLength;
235                     if (len != FIELD_OFFSET(VM_COUNTERS,PrivatePageCount)) len = sizeof(VM_COUNTERS);
236
237                     memcpy(ProcessInformation, &pvmi, min(ProcessInformationLength,sizeof(VM_COUNTERS)));
238                 }
239
240                 if (ProcessInformationLength != FIELD_OFFSET(VM_COUNTERS,PrivatePageCount) &&
241                     ProcessInformationLength != sizeof(VM_COUNTERS))
242                     ret = STATUS_INFO_LENGTH_MISMATCH;
243             }
244             else
245             {
246                 len = sizeof(pvmi);
247                 ret = STATUS_INFO_LENGTH_MISMATCH;
248             }
249         }
250         break;
251     case ProcessTimes:
252         {
253             KERNEL_USER_TIMES pti;
254
255             if (ProcessInformationLength >= sizeof(KERNEL_USER_TIMES))
256             {
257                 if (!ProcessInformation)
258                     ret = STATUS_ACCESS_VIOLATION;
259                 else if (!ProcessHandle)
260                     ret = STATUS_INVALID_HANDLE;
261                 else
262                 {
263                     /* FIXME : User- and KernelTime have to be implemented */
264                     memset(&pti, 0, sizeof(KERNEL_USER_TIMES));
265
266                     SERVER_START_REQ(get_process_info)
267                     {
268                       req->handle = wine_server_obj_handle( ProcessHandle );
269                       if ((ret = wine_server_call( req )) == STATUS_SUCCESS)
270                       {
271                           pti.CreateTime.QuadPart = reply->start_time;
272                           pti.ExitTime.QuadPart = reply->end_time;
273                       }
274                     }
275                     SERVER_END_REQ;
276
277                     memcpy(ProcessInformation, &pti, sizeof(KERNEL_USER_TIMES));
278                     len = sizeof(KERNEL_USER_TIMES);
279                 }
280
281                 if (ProcessInformationLength > sizeof(KERNEL_USER_TIMES))
282                     ret = STATUS_INFO_LENGTH_MISMATCH;
283             }
284             else
285             {
286                 len = sizeof(KERNEL_USER_TIMES);
287                 ret = STATUS_INFO_LENGTH_MISMATCH;
288             }
289         }
290         break;
291     case ProcessDebugPort:
292         len = sizeof(DWORD_PTR);
293         if (ProcessInformationLength == len)
294         {
295             if (!ProcessInformation)
296                 ret = STATUS_ACCESS_VIOLATION;
297             else if (!ProcessHandle)
298                 ret = STATUS_INVALID_HANDLE;
299             else
300             {
301                 SERVER_START_REQ(get_process_info)
302                 {
303                     req->handle = wine_server_obj_handle( ProcessHandle );
304                     if ((ret = wine_server_call( req )) == STATUS_SUCCESS)
305                     {
306                         *(DWORD_PTR *)ProcessInformation = reply->debugger_present ? ~(DWORD_PTR)0 : 0;
307                     }
308                 }
309                 SERVER_END_REQ;
310             }
311         }
312         else
313             ret = STATUS_INFO_LENGTH_MISMATCH;
314         break;
315     case ProcessDebugFlags:
316         len = sizeof(DWORD);
317         if (ProcessInformationLength == len)
318         {
319             if (!ProcessInformation)
320                 ret = STATUS_ACCESS_VIOLATION;
321             else if (!ProcessHandle)
322                 ret = STATUS_INVALID_HANDLE;
323             else
324             {
325                 SERVER_START_REQ(get_process_info)
326                 {
327                     req->handle = wine_server_obj_handle( ProcessHandle );
328                     if ((ret = wine_server_call( req )) == STATUS_SUCCESS)
329                     {
330                         *(DWORD *)ProcessInformation = !reply->debugger_present;
331                     }
332                 }
333                 SERVER_END_REQ;
334             }
335         }
336         else
337             ret = STATUS_INFO_LENGTH_MISMATCH;
338         break;
339     case ProcessDefaultHardErrorMode:
340         len = sizeof(process_error_mode);
341         if (ProcessInformationLength == len)
342             memcpy(ProcessInformation, &process_error_mode, len);
343         else
344             ret = STATUS_INFO_LENGTH_MISMATCH;
345         break;
346     case ProcessDebugObjectHandle:
347         /* "These are not the debuggers you are looking for." *
348          * set it to 0 aka "no debugger" to satisfy copy protections */
349         len = sizeof(HANDLE);
350         if (ProcessInformationLength == len)
351         {
352             if (!ProcessInformation)
353                 ret = STATUS_ACCESS_VIOLATION;
354             else if (!ProcessHandle)
355                 ret = STATUS_INVALID_HANDLE;
356             else
357             {
358                 memset(ProcessInformation, 0, ProcessInformationLength);
359                 ret = STATUS_PORT_NOT_SET;
360             }
361         }
362         else
363             ret = STATUS_INFO_LENGTH_MISMATCH;
364         break;
365     case ProcessHandleCount:
366         if (ProcessInformationLength >= 4)
367         {
368             if (!ProcessInformation)
369                 ret = STATUS_ACCESS_VIOLATION;
370             else if (!ProcessHandle)
371                 ret = STATUS_INVALID_HANDLE;
372             else
373             {
374                 memset(ProcessInformation, 0, 4);
375                 len = 4;
376             }
377
378             if (ProcessInformationLength > 4)
379                 ret = STATUS_INFO_LENGTH_MISMATCH;
380         }
381         else
382         {
383             len = 4;
384             ret = STATUS_INFO_LENGTH_MISMATCH;
385         }
386         break;
387
388     case ProcessAffinityMask:
389         len = sizeof(ULONG_PTR);
390         if (ProcessInformationLength == len)
391         {
392             const ULONG_PTR system_mask = ((ULONG_PTR)1 << NtCurrentTeb()->Peb->NumberOfProcessors) - 1;
393
394             SERVER_START_REQ(get_process_info)
395             {
396                 req->handle = wine_server_obj_handle( ProcessHandle );
397                 if (!(ret = wine_server_call( req )))
398                     *(ULONG_PTR *)ProcessInformation = reply->affinity & system_mask;
399             }
400             SERVER_END_REQ;
401         }
402         else ret = STATUS_INFO_LENGTH_MISMATCH;
403         break;
404
405     case ProcessWow64Information:
406         len = sizeof(DWORD);
407         if (ProcessInformationLength == len)
408         {
409             DWORD val = 0;
410
411             if (ProcessHandle == GetCurrentProcess()) val = is_wow64;
412             else if (server_cpus & (1 << CPU_x86_64))
413             {
414                 SERVER_START_REQ( get_process_info )
415                 {
416                     req->handle = wine_server_obj_handle( ProcessHandle );
417                     if (!(ret = wine_server_call( req ))) val = (reply->cpu != CPU_x86_64);
418                 }
419                 SERVER_END_REQ;
420             }
421             *(DWORD *)ProcessInformation = val;
422         }
423         else ret = STATUS_INFO_LENGTH_MISMATCH;
424         break;
425     case ProcessImageFileName:
426         /* FIXME: this will return a DOS path. Windows returns an NT path. Changing this would require also changing kernel32.QueryFullProcessImageName.
427          * The latter may be harder because of the lack of RtlNtPathNameToDosPathName. */
428         SERVER_START_REQ(get_dll_info)
429         {
430             UNICODE_STRING *image_file_name_str = ProcessInformation;
431
432             req->handle = wine_server_obj_handle( ProcessHandle );
433             req->base_address = 0; /* main module */
434             wine_server_set_reply( req, image_file_name_str ? image_file_name_str + 1 : NULL,
435                                    ProcessInformationLength > sizeof(UNICODE_STRING) ? ProcessInformationLength - sizeof(UNICODE_STRING) : 0 );
436             ret = wine_server_call( req );
437             if (ret == STATUS_BUFFER_TOO_SMALL) ret = STATUS_INFO_LENGTH_MISMATCH;
438
439             len = sizeof(UNICODE_STRING) + reply->filename_len;
440             if (ret == STATUS_SUCCESS)
441             {
442                 image_file_name_str->MaximumLength = image_file_name_str->Length = reply->filename_len;
443                 image_file_name_str->Buffer = (PWSTR)(image_file_name_str + 1);
444             }
445         }
446         SERVER_END_REQ;
447         break;
448     case ProcessExecuteFlags:
449         len = sizeof(ULONG);
450         if (ProcessInformationLength == len)
451             *(ULONG *)ProcessInformation = execute_flags;
452         else
453             ret = STATUS_INFO_LENGTH_MISMATCH;
454         break;
455     default:
456         FIXME("(%p,info_class=%d,%p,0x%08x,%p) Unknown information class\n",
457               ProcessHandle,ProcessInformationClass,
458               ProcessInformation,ProcessInformationLength,
459               ReturnLength);
460         ret = STATUS_INVALID_INFO_CLASS;
461         break;
462     }
463
464     if (ReturnLength) *ReturnLength = len;
465     
466     return ret;
467 }
468
469 /******************************************************************************
470  * NtSetInformationProcess [NTDLL.@]
471  * ZwSetInformationProcess [NTDLL.@]
472  */
473 NTSTATUS WINAPI NtSetInformationProcess(
474         IN HANDLE ProcessHandle,
475         IN PROCESSINFOCLASS ProcessInformationClass,
476         IN PVOID ProcessInformation,
477         IN ULONG ProcessInformationLength)
478 {
479     NTSTATUS ret = STATUS_SUCCESS;
480
481     switch (ProcessInformationClass)
482     {
483     case ProcessDefaultHardErrorMode:
484         if (ProcessInformationLength != sizeof(UINT)) return STATUS_INVALID_PARAMETER;
485         process_error_mode = *(UINT *)ProcessInformation;
486         break;
487     case ProcessAffinityMask:
488         if (ProcessInformationLength != sizeof(DWORD_PTR)) return STATUS_INVALID_PARAMETER;
489         if (*(PDWORD_PTR)ProcessInformation & ~(((DWORD_PTR)1 << NtCurrentTeb()->Peb->NumberOfProcessors) - 1))
490             return STATUS_INVALID_PARAMETER;
491         if (!*(PDWORD_PTR)ProcessInformation)
492             return STATUS_INVALID_PARAMETER;
493         SERVER_START_REQ( set_process_info )
494         {
495             req->handle   = wine_server_obj_handle( ProcessHandle );
496             req->affinity = *(PDWORD_PTR)ProcessInformation;
497             req->mask     = SET_PROCESS_INFO_AFFINITY;
498             ret = wine_server_call( req );
499         }
500         SERVER_END_REQ;
501         break;
502     case ProcessPriorityClass:
503         if (ProcessInformationLength != sizeof(PROCESS_PRIORITY_CLASS))
504             return STATUS_INVALID_PARAMETER;
505         else
506         {
507             PROCESS_PRIORITY_CLASS* ppc = ProcessInformation;
508
509             SERVER_START_REQ( set_process_info )
510             {
511                 req->handle   = wine_server_obj_handle( ProcessHandle );
512                 /* FIXME Foreground isn't used */
513                 req->priority = ppc->PriorityClass;
514                 req->mask     = SET_PROCESS_INFO_PRIORITY;
515                 ret = wine_server_call( req );
516             }
517             SERVER_END_REQ;
518         }
519         break;
520
521     case ProcessExecuteFlags:
522         if (ProcessInformationLength != sizeof(ULONG))
523             return STATUS_INVALID_PARAMETER;
524         else if (execute_flags & MEM_EXECUTE_OPTION_PERMANENT)
525             return STATUS_ACCESS_DENIED;
526         else
527         {
528             BOOL enable;
529             switch (*(ULONG *)ProcessInformation & (MEM_EXECUTE_OPTION_ENABLE|MEM_EXECUTE_OPTION_DISABLE))
530             {
531             case MEM_EXECUTE_OPTION_ENABLE:
532                 enable = TRUE;
533                 break;
534             case MEM_EXECUTE_OPTION_DISABLE:
535                 enable = FALSE;
536                 break;
537             default:
538                 return STATUS_INVALID_PARAMETER;
539             }
540             execute_flags = *(ULONG *)ProcessInformation;
541             VIRTUAL_SetForceExec( enable );
542         }
543         break;
544
545     default:
546         FIXME("(%p,0x%08x,%p,0x%08x) stub\n",
547               ProcessHandle,ProcessInformationClass,ProcessInformation,
548               ProcessInformationLength);
549         ret = STATUS_NOT_IMPLEMENTED;
550         break;
551     }
552     return ret;
553 }
554
555 /******************************************************************************
556  * NtFlushInstructionCache [NTDLL.@]
557  * ZwFlushInstructionCache [NTDLL.@]
558  */
559 NTSTATUS WINAPI NtFlushInstructionCache(
560         IN HANDLE ProcessHandle,
561         IN LPCVOID BaseAddress,
562         IN SIZE_T Size)
563 {
564 #ifdef __i386__
565     TRACE("%p %p %ld - no-op on x86\n", ProcessHandle, BaseAddress, Size );
566 #else
567     FIXME("%p %p %ld\n", ProcessHandle, BaseAddress, Size );
568 #endif
569     return STATUS_SUCCESS;
570 }
571
572 /******************************************************************
573  *              NtOpenProcess [NTDLL.@]
574  *              ZwOpenProcess [NTDLL.@]
575  */
576 NTSTATUS  WINAPI NtOpenProcess(PHANDLE handle, ACCESS_MASK access,
577                                const OBJECT_ATTRIBUTES* attr, const CLIENT_ID* cid)
578 {
579     NTSTATUS    status;
580
581     SERVER_START_REQ( open_process )
582     {
583         req->pid        = HandleToULong(cid->UniqueProcess);
584         req->access     = access;
585         req->attributes = attr ? attr->Attributes : 0;
586         status = wine_server_call( req );
587         if (!status) *handle = wine_server_ptr_handle( reply->handle );
588     }
589     SERVER_END_REQ;
590     return status;
591 }