2008-03-26 Dick Porter <dick@ximian.com>
[mono.git] / mono / metadata / process.c
1 /*
2  * process.c: System.Diagnostics.Process support
3  *
4  * Author:
5  *      Dick Porter (dick@ximian.com)
6  *
7  * (C) 2002 Ximian, Inc.
8  * Copyright (c) 2002-2006 Novell, Inc.
9  */
10
11 #include <config.h>
12
13 #include <glib.h>
14 #include <string.h>
15
16 #include <mono/metadata/object.h>
17 #include <mono/metadata/process.h>
18 #include <mono/metadata/assembly.h>
19 #include <mono/metadata/appdomain.h>
20 #include <mono/metadata/image.h>
21 #include <mono/metadata/cil-coff.h>
22 #include <mono/metadata/exception.h>
23 #include <mono/utils/strenc.h>
24 #include <mono/io-layer/io-layer.h>
25 /* FIXME: fix this code to not depend so much on the inetrnals */
26 #include <mono/metadata/class-internals.h>
27
28 #undef DEBUG
29
30 HANDLE ves_icall_System_Diagnostics_Process_GetProcess_internal (guint32 pid)
31 {
32         HANDLE handle;
33         
34         MONO_ARCH_SAVE_REGS;
35
36         /* GetCurrentProcess returns a pseudo-handle, so use
37          * OpenProcess instead
38          */
39         handle=OpenProcess (PROCESS_ALL_ACCESS, TRUE, pid);
40         
41         if(handle==NULL) {
42                 /* FIXME: Throw an exception */
43                 return(NULL);
44         }
45         
46         return(handle);
47 }
48
49 guint32 ves_icall_System_Diagnostics_Process_GetPid_internal (void)
50 {
51         MONO_ARCH_SAVE_REGS;
52
53         return(GetCurrentProcessId ());
54 }
55
56 void ves_icall_System_Diagnostics_Process_Process_free_internal (MonoObject *this,
57                                                                  HANDLE process)
58 {
59         MONO_ARCH_SAVE_REGS;
60
61 #ifdef THREAD_DEBUG
62         g_message (G_GNUC_PRETTY_FUNCTION ": Closing process %p, handle %p",
63                    this, process);
64 #endif
65
66         CloseHandle (process);
67 }
68
69 #define STASH_SYS_ASS(this) \
70         if(system_assembly == NULL) { \
71                 system_assembly=this->vtable->klass->image; \
72         }
73
74 static MonoImage *system_assembly=NULL;
75
76 static guint32 unicode_chars (const gunichar2 *str)
77 {
78         guint32 len=0;
79         
80         do {
81                 if(str[len]=='\0') {
82                         return(len);
83                 }
84                 len++;
85         } while(1);
86 }
87
88 static void process_set_field_object (MonoObject *obj, const gchar *fieldname,
89                                       MonoObject *data)
90 {
91         MonoClassField *field;
92
93 #ifdef DEBUG
94         g_message (G_GNUC_PRETTY_FUNCTION ": Setting field %s to object at %p",
95                    fieldname, data);
96 #endif
97
98         field=mono_class_get_field_from_name (mono_object_class (obj),
99                                               fieldname);
100         /* FIXME: moving GC */
101         *(MonoObject **)(((char *)obj) + field->offset)=data;
102 }
103
104 static void process_set_field_string (MonoObject *obj, const gchar *fieldname,
105                                       const gunichar2 *val, guint32 len)
106 {
107         MonoClassField *field;
108         MonoString *string;
109
110 #ifdef DEBUG
111         g_message (G_GNUC_PRETTY_FUNCTION ": Setting field %s to [%s]",
112                    fieldname, g_utf16_to_utf8 (val, len, NULL, NULL, NULL));
113 #endif
114
115         string=mono_string_new_utf16 (mono_object_domain (obj), val, len);
116         
117         field=mono_class_get_field_from_name (mono_object_class (obj),
118                                               fieldname);
119         /* FIXME: moving GC */
120         *(MonoString **)(((char *)obj) + field->offset)=string;
121 }
122
123 static void process_set_field_int (MonoObject *obj, const gchar *fieldname,
124                                    guint32 val)
125 {
126         MonoClassField *field;
127
128 #ifdef DEBUG
129         g_message (G_GNUC_PRETTY_FUNCTION ": Setting field %s to %d",
130                    fieldname, val);
131 #endif
132         
133         field=mono_class_get_field_from_name (mono_object_class (obj),
134                                               fieldname);
135         *(guint32 *)(((char *)obj) + field->offset)=val;
136 }
137
138 static void process_set_field_intptr (MonoObject *obj, const gchar *fieldname,
139                                       gpointer val)
140 {
141         MonoClassField *field;
142
143 #ifdef DEBUG
144         g_message ("%s: Setting field %s to %p", __func__, fieldname, val);
145 #endif
146         
147         field=mono_class_get_field_from_name (mono_object_class (obj),
148                                               fieldname);
149         *(gpointer *)(((char *)obj) + field->offset)=val;
150 }
151
152 static void process_set_field_bool (MonoObject *obj, const gchar *fieldname,
153                                     gboolean val)
154 {
155         MonoClassField *field;
156
157 #ifdef DEBUG
158         g_message (G_GNUC_PRETTY_FUNCTION ": Setting field %s to %s",
159                    fieldname, val?"TRUE":"FALSE");
160 #endif
161         
162         field=mono_class_get_field_from_name (mono_object_class (obj),
163                                               fieldname);
164         *(guint8 *)(((char *)obj) + field->offset)=val;
165 }
166
167 #define SFI_COMMENTS            "\\StringFileInfo\\%02X%02X%02X%02X\\Comments"
168 #define SFI_COMPANYNAME         "\\StringFileInfo\\%02X%02X%02X%02X\\CompanyName"
169 #define SFI_FILEDESCRIPTION     "\\StringFileInfo\\%02X%02X%02X%02X\\FileDescription"
170 #define SFI_FILEVERSION         "\\StringFileInfo\\%02X%02X%02X%02X\\FileVersion"
171 #define SFI_INTERNALNAME        "\\StringFileInfo\\%02X%02X%02X%02X\\InternalName"
172 #define SFI_LEGALCOPYRIGHT      "\\StringFileInfo\\%02X%02X%02X%02X\\LegalCopyright"
173 #define SFI_LEGALTRADEMARKS     "\\StringFileInfo\\%02X%02X%02X%02X\\LegalTrademarks"
174 #define SFI_ORIGINALFILENAME    "\\StringFileInfo\\%02X%02X%02X%02X\\OriginalFilename"
175 #define SFI_PRIVATEBUILD        "\\StringFileInfo\\%02X%02X%02X%02X\\PrivateBuild"
176 #define SFI_PRODUCTNAME         "\\StringFileInfo\\%02X%02X%02X%02X\\ProductName"
177 #define SFI_PRODUCTVERSION      "\\StringFileInfo\\%02X%02X%02X%02X\\ProductVersion"
178 #define SFI_SPECIALBUILD        "\\StringFileInfo\\%02X%02X%02X%02X\\SpecialBuild"
179 #define EMPTY_STRING            (gunichar2*)"\000\000"
180
181 static void process_module_string_read (MonoObject *filever, gpointer data,
182                                         const gchar *fieldname,
183                                         guchar lang_hi, guchar lang_lo,
184                                         const gchar *key)
185 {
186         gchar *lang_key_utf8;
187         gunichar2 *lang_key, *buffer;
188         UINT chars;
189
190         lang_key_utf8 = g_strdup_printf (key, lang_lo, lang_hi, 0x04, 0xb0);
191
192 #ifdef DEBUG
193         g_message ("%s: asking for [%s]", __func__, lang_key_utf8);
194 #endif
195
196         lang_key = g_utf8_to_utf16 (lang_key_utf8, -1, NULL, NULL, NULL);
197
198         if (VerQueryValue (data, lang_key, (gpointer *)&buffer, &chars) && chars > 0) {
199 #ifdef DEBUG
200                 g_message ("%s: found %d chars of [%s]", __func__, chars,
201                            g_utf16_to_utf8 (buffer, chars, NULL, NULL, NULL));
202 #endif
203                 /* chars includes trailing null */
204                 process_set_field_string (filever, fieldname, buffer, chars - 1);
205         } else {
206                 process_set_field_string (filever, fieldname, EMPTY_STRING, 0);
207         }
208
209         g_free (lang_key);
210         g_free (lang_key_utf8);
211 }
212
213 static void process_module_stringtable (MonoObject *filever, gpointer data,
214                                         guchar lang_hi, guchar lang_lo)
215 {
216         process_module_string_read (filever, data, "comments", lang_hi, lang_lo,
217                                     SFI_COMMENTS);
218         process_module_string_read (filever, data, "companyname", lang_hi,
219                                     lang_lo, SFI_COMPANYNAME);
220         process_module_string_read (filever, data, "filedescription", lang_hi,
221                                     lang_lo, SFI_FILEDESCRIPTION);
222         process_module_string_read (filever, data, "fileversion", lang_hi,
223                                     lang_lo, SFI_FILEVERSION);
224         process_module_string_read (filever, data, "internalname", lang_hi,
225                                     lang_lo, SFI_INTERNALNAME);
226         process_module_string_read (filever, data, "legalcopyright", lang_hi,
227                                     lang_lo, SFI_LEGALCOPYRIGHT);
228         process_module_string_read (filever, data, "legaltrademarks", lang_hi,
229                                     lang_lo, SFI_LEGALTRADEMARKS);
230         process_module_string_read (filever, data, "originalfilename", lang_hi,
231                                     lang_lo, SFI_ORIGINALFILENAME);
232         process_module_string_read (filever, data, "privatebuild", lang_hi,
233                                     lang_lo, SFI_PRIVATEBUILD);
234         process_module_string_read (filever, data, "productname", lang_hi,
235                                     lang_lo, SFI_PRODUCTNAME);
236         process_module_string_read (filever, data, "productversion", lang_hi,
237                                     lang_lo, SFI_PRODUCTVERSION);
238         process_module_string_read (filever, data, "specialbuild", lang_hi,
239                                     lang_lo, SFI_SPECIALBUILD);
240 }
241
242 static void process_get_fileversion (MonoObject *filever, gunichar2 *filename)
243 {
244         DWORD verinfohandle;
245         VS_FIXEDFILEINFO *ffi;
246         gpointer data;
247         DWORD datalen;
248         guchar *trans_data;
249         gunichar2 *query;
250         UINT ffi_size, trans_size;
251         BOOL ok;
252         int i;
253
254         datalen = GetFileVersionInfoSize (filename, &verinfohandle);
255         if (datalen) {
256                 data = g_malloc0 (datalen);
257                 ok = GetFileVersionInfo (filename, verinfohandle, datalen,
258                                          data);
259                 if (ok) {
260                         query = g_utf8_to_utf16 ("\\", -1, NULL, NULL, NULL);
261                         if (query == NULL) {
262                                 g_free (data);
263                                 return;
264                         }
265                         
266                         if (VerQueryValue (data, query, (gpointer *)&ffi,
267                             &ffi_size)) {
268 #ifdef DEBUG
269                                 g_message (G_GNUC_PRETTY_FUNCTION ": recording assembly: FileName [%s] FileVersionInfo [%d.%d.%d.%d]", g_utf16_to_utf8 (filename, -1, NULL, NULL, NULL), HIWORD (ffi->dwFileVersionMS), LOWORD (ffi->dwFileVersionMS), HIWORD (ffi->dwFileVersionLS), LOWORD (ffi->dwFileVersionLS));
270 #endif
271         
272                                 process_set_field_int (filever, "filemajorpart", HIWORD (ffi->dwFileVersionMS));
273                                 process_set_field_int (filever, "fileminorpart", LOWORD (ffi->dwFileVersionMS));
274                                 process_set_field_int (filever, "filebuildpart", HIWORD (ffi->dwFileVersionLS));
275                                 process_set_field_int (filever, "fileprivatepart", LOWORD (ffi->dwFileVersionLS));
276
277                                 process_set_field_int (filever, "productmajorpart", HIWORD (ffi->dwProductVersionMS));
278                                 process_set_field_int (filever, "productminorpart", LOWORD (ffi->dwProductVersionMS));
279                                 process_set_field_int (filever, "productbuildpart", HIWORD (ffi->dwProductVersionLS));
280                                 process_set_field_int (filever, "productprivatepart", LOWORD (ffi->dwProductVersionLS));
281
282                                 process_set_field_bool (filever, "isdebug", (ffi->dwFileFlags & ffi->dwFileFlagsMask) & VS_FF_DEBUG);
283                                 process_set_field_bool (filever, "isprerelease", (ffi->dwFileFlags & ffi->dwFileFlagsMask) & VS_FF_PRERELEASE);
284                                 process_set_field_bool (filever, "ispatched", (ffi->dwFileFlags & ffi->dwFileFlagsMask) & VS_FF_PATCHED);
285                                 process_set_field_bool (filever, "isprivatebuild", (ffi->dwFileFlags & ffi->dwFileFlagsMask) & VS_FF_PRIVATEBUILD);
286                                 process_set_field_bool (filever, "isspecialbuild", (ffi->dwFileFlags & ffi->dwFileFlagsMask) & VS_FF_SPECIALBUILD);
287                         }
288                         g_free (query);
289
290                         query = g_utf8_to_utf16 ("\\VarFileInfo\\Translation", -1, NULL, NULL, NULL);
291                         if (query == NULL) {
292                                 g_free (data);
293                                 return;
294                         }
295                         
296                         if (VerQueryValue (data, query,
297                                            (gpointer *)&trans_data,
298                                            &trans_size)) {
299                                 /* Look for neutral or en_US language data
300                                  * (or should we use the first language ID we
301                                  * see?)
302                                  */
303                                 for (i = 0; i < trans_size; i += 4) {
304 #ifdef DEBUG
305                                         g_message("%s: %s has 0x%0x 0x%0x 0x%0x 0x%0x", __func__, g_utf16_to_utf8 (filename, -1, NULL, NULL, NULL), trans_data[i], trans_data[i+1], trans_data[i+2], trans_data[i+3]);
306 #endif
307
308                                         if ((trans_data[i] == 0x09 &&
309                                              trans_data[i+1] == 0x04 &&
310                                              trans_data[i+2] == 0xb0 &&
311                                              trans_data[i+3] == 0x04) ||
312                                             (trans_data[i] == 0x00 &&
313                                              trans_data[i+1] == 0x00 &&
314                                              trans_data[i+2] == 0xb0 &&
315                                              trans_data[i+3] == 0x04) ||
316                                             (trans_data[i] == 0x7f &&
317                                              trans_data[i+1] == 0x00 &&
318                                              trans_data[i+2] == 0xb0 &&
319                                              trans_data[i+3] == 0x04)) {
320                                                 gunichar2 lang_buf[128];
321                                                 guint32 lang, lang_count;
322
323                                                 lang = (trans_data[i]) |
324                                                        (trans_data[i+1] << 8) |
325                                                        (trans_data[i+2] << 16) |
326                                                        (trans_data[i+3] << 24);
327                                                 lang_count = VerLanguageName (lang, lang_buf, 128);
328                                                 if (lang_count) {
329                                                         process_set_field_string (filever, "language", lang_buf, lang_count);
330                                                 }
331                                                 process_module_stringtable (filever, data, trans_data[i], trans_data[i+1]);
332                                         }
333                                 }
334                         } else {
335                                 /* No strings, so set every field to
336                                  * the empty string
337                                  */
338                                 process_set_field_string (filever,
339                                                           "comments",
340                                                           EMPTY_STRING, 0);
341                                 process_set_field_string (filever,
342                                                           "companyname",
343                                                           EMPTY_STRING, 0);
344                                 process_set_field_string (filever,
345                                                           "filedescription",
346                                                           EMPTY_STRING, 0);
347                                 process_set_field_string (filever,
348                                                           "fileversion",
349                                                           EMPTY_STRING, 0);
350                                 process_set_field_string (filever,
351                                                           "internalname",
352                                                           EMPTY_STRING, 0);
353                                 process_set_field_string (filever,
354                                                           "legalcopyright",
355                                                           EMPTY_STRING, 0);
356                                 process_set_field_string (filever,
357                                                           "legaltrademarks",
358                                                           EMPTY_STRING, 0);
359                                 process_set_field_string (filever,
360                                                           "originalfilename",
361                                                           EMPTY_STRING, 0);
362                                 process_set_field_string (filever,
363                                                           "privatebuild",
364                                                           EMPTY_STRING, 0);
365                                 process_set_field_string (filever,
366                                                           "productname",
367                                                           EMPTY_STRING, 0);
368                                 process_set_field_string (filever,
369                                                           "productversion",
370                                                           EMPTY_STRING, 0);
371                                 process_set_field_string (filever,
372                                                           "specialbuild",
373                                                           EMPTY_STRING, 0);
374                         }
375                         
376                         g_free (query);
377                 }
378                 g_free (data);
379         }
380 }
381
382 static void process_add_module (GPtrArray *modules, HANDLE process, HMODULE mod,
383                                 gunichar2 *filename, gunichar2 *modulename)
384 {
385         MonoClass *proc_class, *filever_class;
386         MonoObject *item, *filever;
387         MonoDomain *domain=mono_domain_get ();
388         MODULEINFO modinfo;
389         BOOL ok;
390         
391         /* Build a System.Diagnostics.ProcessModule with the data.
392          */
393         proc_class=mono_class_from_name (system_assembly, "System.Diagnostics",
394                                          "ProcessModule");
395         item=mono_object_new (domain, proc_class);
396
397         filever_class=mono_class_from_name (system_assembly,
398                                             "System.Diagnostics",
399                                             "FileVersionInfo");
400         filever=mono_object_new (domain, filever_class);
401
402         process_get_fileversion (filever, filename);
403
404         process_set_field_string (filever, "filename", filename,
405                                   unicode_chars (filename));
406
407         ok = GetModuleInformation (process, mod, &modinfo, sizeof(MODULEINFO));
408         if (ok) {
409                 process_set_field_intptr (item, "baseaddr",
410                                           modinfo.lpBaseOfDll);
411                 process_set_field_intptr (item, "entryaddr",
412                                           modinfo.EntryPoint);
413                 process_set_field_int (item, "memory_size",
414                                        modinfo.SizeOfImage);
415         }
416         process_set_field_string (item, "filename", filename,
417                                   unicode_chars (filename));
418         process_set_field_string (item, "modulename", modulename,
419                                   unicode_chars (modulename));
420         process_set_field_object (item, "version_info", filever);
421
422         /* FIXME: moving GC */
423         g_ptr_array_add (modules, item);
424 }
425
426 /* Returns an array of System.Diagnostics.ProcessModule */
427 MonoArray *ves_icall_System_Diagnostics_Process_GetModules_internal (MonoObject *this, HANDLE process)
428 {
429         GPtrArray *modules_list=g_ptr_array_new ();
430         MonoArray *arr;
431         HMODULE mods[1024];
432         gunichar2 filename[MAX_PATH];
433         gunichar2 modname[MAX_PATH];
434         DWORD needed;
435         guint32 count;
436         guint32 i;
437         
438         MONO_ARCH_SAVE_REGS;
439
440         STASH_SYS_ASS (this);
441
442         if (EnumProcessModules (process, mods, sizeof(mods), &needed)) {
443                 count = needed / sizeof(HMODULE);
444                 for (i = 0; i < count; i++) {
445                         if (GetModuleBaseName (process, mods[i], modname,
446                                                MAX_PATH) &&
447                             GetModuleFileNameEx (process, mods[i], filename,
448                                                  MAX_PATH)) {
449                                 process_add_module (modules_list, process,
450                                                     mods[i], filename, modname);
451                         }
452                 }
453         }
454
455         /* Build a MonoArray out of modules_list */
456         arr=mono_array_new (mono_domain_get (), mono_get_object_class (),
457                             modules_list->len);
458         
459         for(i=0; i<modules_list->len; i++) {
460                 mono_array_setref (arr, i, g_ptr_array_index (modules_list, i));
461         }
462         
463         g_ptr_array_free (modules_list, TRUE);
464         
465         return(arr);
466 }
467
468 void ves_icall_System_Diagnostics_FileVersionInfo_GetVersionInfo_internal (MonoObject *this, MonoString *filename)
469 {
470         MONO_ARCH_SAVE_REGS;
471
472         STASH_SYS_ASS (this);
473         
474         process_get_fileversion (this, mono_string_chars (filename));
475         process_set_field_string (this, "filename",
476                                   mono_string_chars (filename),
477                                   mono_string_length (filename));
478 }
479
480 /* Only used when UseShellExecute is false */
481 static gchar *
482 quote_path (const gchar *path)
483 {
484         gchar *res = g_shell_quote (path);
485 #ifdef PLATFORM_WIN32
486         {
487         gchar *q = res;
488         while (*q) {
489                 if (*q == '\'')
490                         *q = '\"';
491                 q++;
492         }
493         }
494 #endif
495         return res;
496 }
497
498 /* Only used when UseShellExecute is false */
499 static gboolean
500 complete_path (const gunichar2 *appname, gchar **completed)
501 {
502         gchar *utf8app;
503         gchar *found;
504
505         utf8app = g_utf16_to_utf8 (appname, -1, NULL, NULL, NULL);
506         if (g_path_is_absolute (utf8app)) {
507                 *completed = quote_path (utf8app);
508                 g_free (utf8app);
509                 return TRUE;
510         }
511
512         if (g_file_test (utf8app, G_FILE_TEST_IS_EXECUTABLE) && !g_file_test (utf8app, G_FILE_TEST_IS_DIR)) {
513                 *completed = quote_path (utf8app);
514                 g_free (utf8app);
515                 return TRUE;
516         }
517         
518         found = g_find_program_in_path (utf8app);
519         if (found == NULL) {
520                 *completed = NULL;
521                 g_free (utf8app);
522                 return FALSE;
523         }
524
525         *completed = quote_path (found);
526         g_free (found);
527         g_free (utf8app);
528         return TRUE;
529 }
530
531 #ifndef HAVE_GETPROCESSID
532 /* Run-time GetProcessId detection for Windows */
533 #ifdef PLATFORM_WIN32
534 #define HAVE_GETPROCESSID
535
536 typedef DWORD (WINAPI *GETPROCESSID_PROC) (HANDLE);
537 typedef DWORD (WINAPI *NTQUERYINFORMATIONPROCESS_PROC) (HANDLE, PROCESSINFOCLASS, PVOID, ULONG, PULONG);
538 typedef DWORD (WINAPI *RTLNTSTATUSTODOSERROR_PROC) (NTSTATUS);
539
540 static DWORD WINAPI GetProcessId_detect (HANDLE process);
541
542 static GETPROCESSID_PROC GetProcessId = &GetProcessId_detect;
543 static NTQUERYINFORMATIONPROCESS_PROC NtQueryInformationProcess_proc = NULL;
544 static RTLNTSTATUSTODOSERROR_PROC RtlNtStatusToDosError_proc = NULL;
545
546 static DWORD WINAPI GetProcessId_ntdll (HANDLE process)
547 {
548         PROCESS_BASIC_INFORMATION pi;
549         NTSTATUS status;
550
551         status = NtQueryInformationProcess_proc (process, ProcessBasicInformation, &pi, sizeof (pi), NULL);
552         if (NT_SUCCESS (status)) {
553                 return pi.UniqueProcessId;
554         } else {
555                 SetLastError (RtlNtStatusToDosError_proc (status));
556                 return 0;
557         }
558 }
559
560 static DWORD WINAPI GetProcessId_stub (HANDLE process)
561 {
562         SetLastError (ERROR_CALL_NOT_IMPLEMENTED);
563         return 0;
564 }
565
566 static DWORD WINAPI GetProcessId_detect (HANDLE process)
567 {
568         HMODULE module_handle;
569         GETPROCESSID_PROC GetProcessId_kernel;
570
571         /* Windows XP SP1 and above have GetProcessId API */
572         module_handle = GetModuleHandle (L"kernel32.dll");
573         if (module_handle != NULL) {
574                 GetProcessId_kernel = (GETPROCESSID_PROC) GetProcAddress (module_handle, "GetProcessId");
575                 if (GetProcessId_kernel != NULL) {
576                         GetProcessId = GetProcessId_kernel;
577                         return GetProcessId (process);
578                 }
579         }
580
581         /* Windows 2000 and above have deprecated NtQueryInformationProcess API */
582         module_handle = GetModuleHandle (L"ntdll.dll");
583         if (module_handle != NULL) {
584                 NtQueryInformationProcess_proc = (NTQUERYINFORMATIONPROCESS_PROC) GetProcAddress (module_handle, "NtQueryInformationProcess");
585                 if (NtQueryInformationProcess_proc != NULL) {
586                         RtlNtStatusToDosError_proc = (RTLNTSTATUSTODOSERROR_PROC) GetProcAddress (module_handle, "RtlNtStatusToDosError");
587                         if (RtlNtStatusToDosError_proc != NULL) {
588                                 GetProcessId = &GetProcessId_ntdll;
589                                 return GetProcessId (process);
590                         }
591                 }
592         }
593
594         /* Fall back to ERROR_CALL_NOT_IMPLEMENTED */
595         GetProcessId = &GetProcessId_stub;
596         return GetProcessId (process);
597 }
598 #endif /* PLATFORM_WIN32 */
599 #endif /* !HAVE_GETPROCESSID */
600
601 MonoBoolean ves_icall_System_Diagnostics_Process_ShellExecuteEx_internal (MonoProcessStartInfo *proc_start_info, MonoProcInfo *process_info)
602 {
603         SHELLEXECUTEINFO shellex = {0};
604         gboolean ret;
605
606         shellex.cbSize = sizeof(SHELLEXECUTEINFO);
607         shellex.fMask = SEE_MASK_FLAG_DDEWAIT | SEE_MASK_NOCLOSEPROCESS | SEE_MASK_UNICODE;
608         shellex.nShow = SW_SHOWNORMAL;
609
610         
611         
612         if (proc_start_info->filename != NULL) {
613                 shellex.lpFile = mono_string_chars (proc_start_info->filename);
614         }
615
616         if (proc_start_info->arguments != NULL) {
617                 shellex.lpParameters = mono_string_chars (proc_start_info->arguments);
618         }
619
620         if (proc_start_info->verb != NULL &&
621             mono_string_length (proc_start_info->verb) != 0) {
622                 shellex.lpVerb = mono_string_chars (proc_start_info->verb);
623         }
624
625         if (proc_start_info->working_directory != NULL &&
626             mono_string_length (proc_start_info->working_directory) != 0) {
627                 shellex.lpDirectory = mono_string_chars (proc_start_info->working_directory);
628         }
629
630         if (proc_start_info->error_dialog) {    
631                 shellex.hwnd = proc_start_info->error_dialog_parent_handle;
632         } else {
633                 shellex.fMask |= SEE_MASK_FLAG_NO_UI;
634         }
635
636         ret = ShellExecuteEx (&shellex);
637         if (ret == FALSE) {
638                 process_info->pid = -GetLastError ();
639         } else {
640                 process_info->process_handle = shellex.hProcess;
641                 process_info->thread_handle = NULL;
642                 /* It appears that there's no way to get the pid from a
643                  * process handle before windows xp.  Really.
644                  */
645 #ifdef HAVE_GETPROCESSID
646                 process_info->pid = GetProcessId (shellex.hProcess);
647 #else
648                 process_info->pid = 0;
649 #endif
650                 process_info->tid = 0;
651         }
652
653         return (ret);
654 }
655
656 MonoBoolean ves_icall_System_Diagnostics_Process_CreateProcess_internal (MonoProcessStartInfo *proc_start_info, HANDLE stdin_handle, HANDLE stdout_handle, HANDLE stderr_handle, MonoProcInfo *process_info)
657 {
658         gboolean ret;
659         gunichar2 *dir;
660         STARTUPINFO startinfo={0};
661         PROCESS_INFORMATION procinfo;
662         gunichar2 *shell_path = NULL;
663         gchar *env_vars = NULL;
664         gboolean free_shell_path = TRUE;
665         gchar *spath = NULL;
666         MonoString *cmd = proc_start_info->arguments;
667         guint32 creation_flags, logon_flags;
668         
669         startinfo.cb=sizeof(STARTUPINFO);
670         startinfo.dwFlags=STARTF_USESTDHANDLES;
671         startinfo.hStdInput=stdin_handle;
672         startinfo.hStdOutput=stdout_handle;
673         startinfo.hStdError=stderr_handle;
674
675         creation_flags = CREATE_UNICODE_ENVIRONMENT;
676         if (proc_start_info->create_no_window)
677                 creation_flags |= CREATE_NO_WINDOW;
678         
679         shell_path = mono_string_chars (proc_start_info->filename);
680         complete_path (shell_path, &spath);
681         if (spath == NULL) {
682                 process_info->pid = -ERROR_FILE_NOT_FOUND;
683                 return FALSE;
684         }
685 #ifdef PLATFORM_WIN32
686         /* Seems like our CreateProcess does not work as the windows one.
687          * This hack is needed to deal with paths containing spaces */
688         shell_path = NULL;
689         free_shell_path = FALSE;
690         if (cmd) {
691                 gchar *newcmd, *tmp;
692                 tmp = mono_string_to_utf8 (cmd);
693                 newcmd = g_strdup_printf ("%s %s", spath, tmp);
694                 cmd = mono_string_new_wrapper (newcmd);
695                 g_free (tmp);
696                 g_free (newcmd);
697         }
698         else {
699                 cmd = mono_string_new_wrapper (spath);
700         }
701 #else
702         shell_path = g_utf8_to_utf16 (spath, -1, NULL, NULL, NULL);
703 #endif
704         g_free (spath);
705
706         if (process_info->env_keys != NULL) {
707                 gint i, len; 
708                 MonoString *ms;
709                 MonoString *key, *value;
710                 gunichar2 *str, *ptr;
711                 gunichar2 *equals16;
712
713                 for (len = 0, i = 0; i < mono_array_length (process_info->env_keys); i++) {
714                         ms = mono_array_get (process_info->env_values, MonoString *, i);
715                         if (ms == NULL)
716                                 continue;
717
718                         len += mono_string_length (ms) * sizeof (gunichar2);
719                         ms = mono_array_get (process_info->env_keys, MonoString *, i);
720                         len += mono_string_length (ms) * sizeof (gunichar2);
721                         len += 2 * sizeof (gunichar2);
722                 }
723
724                 equals16 = g_utf8_to_utf16 ("=", 1, NULL, NULL, NULL);
725                 ptr = str = g_new0 (gunichar2, len + 1);
726                 for (i = 0; i < mono_array_length (process_info->env_keys); i++) {
727                         value = mono_array_get (process_info->env_values, MonoString *, i);
728                         if (value == NULL)
729                                 continue;
730
731                         key = mono_array_get (process_info->env_keys, MonoString *, i);
732                         memcpy (ptr, mono_string_chars (key), mono_string_length (key) * sizeof (gunichar2));
733                         ptr += mono_string_length (key);
734
735                         memcpy (ptr, equals16, sizeof (gunichar2));
736                         ptr++;
737
738                         memcpy (ptr, mono_string_chars (value), mono_string_length (value) * sizeof (gunichar2));
739                         ptr += mono_string_length (value);
740                         ptr++;
741                 }
742
743                 g_free (equals16);
744                 env_vars = (gchar *) str;
745         }
746         
747         /* The default dir name is "".  Turn that into NULL to mean
748          * "current directory"
749          */
750         if(mono_string_length (proc_start_info->working_directory)==0) {
751                 dir=NULL;
752         } else {
753                 dir=mono_string_chars (proc_start_info->working_directory);
754         }
755
756         if (process_info->username) {
757                 logon_flags = process_info->load_user_profile ? LOGON_WITH_PROFILE : 0;
758                 ret=CreateProcessWithLogonW (mono_string_chars (process_info->username), process_info->domain ? mono_string_chars (process_info->domain) : NULL, process_info->password, logon_flags, shell_path, cmd? mono_string_chars (cmd): NULL, creation_flags, env_vars, dir, &startinfo, &procinfo);
759         } else {
760                 ret=CreateProcess (shell_path, cmd? mono_string_chars (cmd): NULL, NULL, NULL, TRUE, creation_flags, env_vars, dir, &startinfo, &procinfo);
761         }
762
763         g_free (env_vars);
764         if (free_shell_path)
765                 g_free (shell_path);
766
767         if(ret) {
768                 process_info->process_handle=procinfo.hProcess;
769                 /*process_info->thread_handle=procinfo.hThread;*/
770                 process_info->thread_handle=NULL;
771                 if (procinfo.hThread != NULL && procinfo.hThread != INVALID_HANDLE_VALUE)
772                         CloseHandle(procinfo.hThread);
773                 process_info->pid=procinfo.dwProcessId;
774                 process_info->tid=procinfo.dwThreadId;
775         } else {
776                 process_info->pid = -GetLastError ();
777         }
778         
779         return(ret);
780 }
781
782 MonoBoolean ves_icall_System_Diagnostics_Process_WaitForExit_internal (MonoObject *this, HANDLE process, gint32 ms)
783 {
784         guint32 ret;
785         
786         MONO_ARCH_SAVE_REGS;
787
788         if(ms<0) {
789                 /* Wait forever */
790                 ret=WaitForSingleObjectEx (process, INFINITE, TRUE);
791         } else {
792                 ret=WaitForSingleObjectEx (process, ms, TRUE);
793         }
794         if(ret==WAIT_OBJECT_0) {
795                 return(TRUE);
796         } else {
797                 return(FALSE);
798         }
799 }
800
801 gint64 ves_icall_System_Diagnostics_Process_ExitTime_internal (HANDLE process)
802 {
803         gboolean ret;
804         gint64 ticks;
805         FILETIME create_time, exit_time, kernel_time, user_time;
806         
807         MONO_ARCH_SAVE_REGS;
808
809         ret=GetProcessTimes (process, &create_time, &exit_time, &kernel_time,
810                              &user_time);
811         if(ret==TRUE) {
812                 ticks=((guint64)exit_time.dwHighDateTime << 32) +
813                         exit_time.dwLowDateTime;
814                 
815                 return(ticks);
816         } else {
817                 return(0);
818         }
819 }
820
821 gint64 ves_icall_System_Diagnostics_Process_StartTime_internal (HANDLE process)
822 {
823         gboolean ret;
824         gint64 ticks;
825         FILETIME create_time, exit_time, kernel_time, user_time;
826         
827         MONO_ARCH_SAVE_REGS;
828
829         ret=GetProcessTimes (process, &create_time, &exit_time, &kernel_time,
830                              &user_time);
831         if(ret==TRUE) {
832                 ticks=((guint64)create_time.dwHighDateTime << 32) +
833                         create_time.dwLowDateTime;
834                 
835                 return(ticks);
836         } else {
837                 return(0);
838         }
839 }
840
841 gint32 ves_icall_System_Diagnostics_Process_ExitCode_internal (HANDLE process)
842 {
843         DWORD code;
844         
845         MONO_ARCH_SAVE_REGS;
846
847         GetExitCodeProcess (process, &code);
848         
849 #ifdef DEBUG
850         g_message (G_GNUC_PRETTY_FUNCTION ": process exit code is %d", code);
851 #endif
852         
853         return(code);
854 }
855
856 MonoString *ves_icall_System_Diagnostics_Process_ProcessName_internal (HANDLE process)
857 {
858         MonoString *string;
859         gboolean ok;
860         HMODULE mod;
861         gunichar2 name[MAX_PATH];
862         DWORD needed;
863         guint32 len;
864         
865         MONO_ARCH_SAVE_REGS;
866
867         ok=EnumProcessModules (process, &mod, sizeof(mod), &needed);
868         if(ok==FALSE) {
869                 return(NULL);
870         }
871         
872         len=GetModuleBaseName (process, mod, name, MAX_PATH);
873         if(len==0) {
874                 return(NULL);
875         }
876         
877 #ifdef DEBUG
878         g_message (G_GNUC_PRETTY_FUNCTION ": process name is [%s]",
879                    g_utf16_to_utf8 (name, -1, NULL, NULL, NULL));
880 #endif
881         
882         string=mono_string_new_utf16 (mono_domain_get (), name, len);
883         
884         return(string);
885 }
886
887 /* Returns an array of pids */
888 MonoArray *ves_icall_System_Diagnostics_Process_GetProcesses_internal (void)
889 {
890         MonoArray *procs;
891         gboolean ret;
892         DWORD needed;
893         guint32 count, i;
894         DWORD pids[1024];
895
896         MONO_ARCH_SAVE_REGS;
897
898         ret=EnumProcesses (pids, sizeof(pids), &needed);
899         if(ret==FALSE) {
900                 /* FIXME: throw an exception */
901                 return(NULL);
902         }
903         
904         count=needed/sizeof(DWORD);
905         procs=mono_array_new (mono_domain_get (), mono_get_int32_class (),
906                               count);
907         for(i=0; i<count; i++) {
908                 mono_array_set (procs, guint32, i, pids[i]);
909         }
910         
911         return(procs);
912 }
913
914 MonoBoolean ves_icall_System_Diagnostics_Process_GetWorkingSet_internal (HANDLE process, guint32 *min, guint32 *max)
915 {
916         gboolean ret;
917         SIZE_T ws_min, ws_max;
918         
919         MONO_ARCH_SAVE_REGS;
920
921         ret=GetProcessWorkingSetSize (process, &ws_min, &ws_max);
922         *min=(guint32)ws_min;
923         *max=(guint32)ws_max;
924         
925         return(ret);
926 }
927
928 MonoBoolean ves_icall_System_Diagnostics_Process_SetWorkingSet_internal (HANDLE process, guint32 min, guint32 max, MonoBoolean use_min)
929 {
930         gboolean ret;
931         SIZE_T ws_min;
932         SIZE_T ws_max;
933         
934         MONO_ARCH_SAVE_REGS;
935
936         ret=GetProcessWorkingSetSize (process, &ws_min, &ws_max);
937         if(ret==FALSE) {
938                 return(FALSE);
939         }
940         
941         if(use_min==TRUE) {
942                 ws_min=(SIZE_T)min;
943         } else {
944                 ws_max=(SIZE_T)max;
945         }
946         
947         ret=SetProcessWorkingSetSize (process, ws_min, ws_max);
948
949         return(ret);
950 }
951
952 MonoBoolean
953 ves_icall_System_Diagnostics_Process_Kill_internal (HANDLE process, gint32 sig)
954 {
955         MONO_ARCH_SAVE_REGS;
956
957         /* sig == 1 -> Kill, sig == 2 -> CloseMainWindow */
958
959         return TerminateProcess (process, -sig);
960 }
961
962 gint64
963 ves_icall_System_Diagnostics_Process_Times (HANDLE process, gint32 type)
964 {
965         FILETIME create_time, exit_time, kernel_time, user_time;
966         
967         if (GetProcessTimes (process, &create_time, &exit_time, &kernel_time, &user_time)) {
968                 if (type == 0)
969                         return *(gint64*)&user_time;
970                 else if (type == 1)
971                         return *(gint64*)&kernel_time;
972                 /* system + user time: FILETIME can be (memory) cast to a 64 bit int */
973                 return *(gint64*)&kernel_time + *(gint64*)&user_time;
974         }
975         return 0;
976 }
977
978 gint32
979 ves_icall_System_Diagnostics_Process_GetPriorityClass (HANDLE process, gint32 *error)
980 {
981         gint32 ret = GetPriorityClass (process);
982         *error = ret == 0 ? GetLastError () : 0;
983         return ret;
984 }
985
986 MonoBoolean
987 ves_icall_System_Diagnostics_Process_SetPriorityClass (HANDLE process, gint32 priority_class, gint32 *error)
988 {
989         gboolean ret = SetPriorityClass (process, priority_class);
990         *error = ret == 0 ? GetLastError () : 0;
991         return ret;
992 }