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