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-internals.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 #if defined (MINGW_CROSS_COMPILE) && defined (HAVE_GETPROCESSID)
 514 #undef HAVE_GETPROCESSID
 515 #endif
 516
 517 #ifndef HAVE_GETPROCESSID
 518 /* Run-time GetProcessId detection for Windows */
 519 #ifdef TARGET_WIN32
 520 #define HAVE_GETPROCESSID
 521
 522 typedef DWORD (WINAPI *GETPROCESSID_PROC) (HANDLE);
 523 typedef DWORD (WINAPI *NTQUERYINFORMATIONPROCESS_PROC) (HANDLE, PROCESSINFOCLASS, PVOID, ULONG, PULONG);
 524 typedef DWORD (WINAPI *RTLNTSTATUSTODOSERROR_PROC) (NTSTATUS);
 525
 526 static DWORD WINAPI GetProcessId_detect (HANDLE process);
 527
 528 static GETPROCESSID_PROC GetProcessId = &GetProcessId_detect;
 529 static NTQUERYINFORMATIONPROCESS_PROC NtQueryInformationProcess_proc = NULL;
 530 static RTLNTSTATUSTODOSERROR_PROC RtlNtStatusToDosError_proc = NULL;
 531
 532 static DWORD WINAPI GetProcessId_ntdll (HANDLE process)
 533 {
 534         PROCESS_BASIC_INFORMATION pi;
 535         NTSTATUS status;
 536
 537         status = NtQueryInformationProcess_proc (process, ProcessBasicInformation, &pi, sizeof (pi), NULL);
 538         if (NT_SUCCESS (status)) {
 539                 return pi.UniqueProcessId;
 540         } else {
 541                 SetLastError (RtlNtStatusToDosError_proc (status));
 542                 return 0;
 543         }
 544 }
 545
 546 static DWORD WINAPI GetProcessId_stub (HANDLE process)
 547 {
 548         SetLastError (ERROR_CALL_NOT_IMPLEMENTED);
 549         return 0;
 550 }
 551
 552 static DWORD WINAPI GetProcessId_detect (HANDLE process)
 553 {
 554         HMODULE module_handle;
 555         GETPROCESSID_PROC GetProcessId_kernel;
 556
 557         /* Windows XP SP1 and above have GetProcessId API */
 558         module_handle = GetModuleHandle (L"kernel32.dll");
 559         if (module_handle != NULL) {
 560                 GetProcessId_kernel = (GETPROCESSID_PROC) GetProcAddress (module_handle, "GetProcessId");
 561                 if (GetProcessId_kernel != NULL) {
 562                         GetProcessId = GetProcessId_kernel;
 563                         return GetProcessId (process);
 564                 }
 565         }
 566
 567         /* Windows 2000 and above have deprecated NtQueryInformationProcess API */
 568         module_handle = GetModuleHandle (L"ntdll.dll");
 569         if (module_handle != NULL) {
 570                 NtQueryInformationProcess_proc = (NTQUERYINFORMATIONPROCESS_PROC) GetProcAddress (module_handle, "NtQueryInformationProcess");
 571                 if (NtQueryInformationProcess_proc != NULL) {
 572                         RtlNtStatusToDosError_proc = (RTLNTSTATUSTODOSERROR_PROC) GetProcAddress (module_handle, "RtlNtStatusToDosError");
 573                         if (RtlNtStatusToDosError_proc != NULL) {
 574                                 GetProcessId = &GetProcessId_ntdll;
 575                                 return GetProcessId (process);
 576                         }
 577                 }
 578         }
 579
 580         /* Fall back to ERROR_CALL_NOT_IMPLEMENTED */
 581         GetProcessId = &GetProcessId_stub;
 582         return GetProcessId (process);
 583 }
 584 #endif /* HOST_WIN32 */
 585 #endif /* !HAVE_GETPROCESSID */
 586
 587 MonoBoolean ves_icall_System_Diagnostics_Process_ShellExecuteEx_internal (MonoProcessStartInfo *proc_start_info, MonoProcInfo *process_info)
 588 {
 589         SHELLEXECUTEINFO shellex = {0};
 590         gboolean ret;
 591
 592         shellex.cbSize = sizeof(SHELLEXECUTEINFO);
 593         shellex.fMask = SEE_MASK_FLAG_DDEWAIT | SEE_MASK_NOCLOSEPROCESS | SEE_MASK_UNICODE;
 594         shellex.nShow = proc_start_info->window_style;
 595         shellex.nShow = (shellex.nShow == 0) ? 1 : (shellex.nShow == 1 ? 0 : shellex.nShow);
 596
 597
 598         if (proc_start_info->filename != NULL) {
 599                 shellex.lpFile = mono_string_chars (proc_start_info->filename);
 600         }
 601
 602         if (proc_start_info->arguments != NULL) {
 603                 shellex.lpParameters = mono_string_chars (proc_start_info->arguments);
 604         }
 605
 606         if (proc_start_info->verb != NULL &&
 607             mono_string_length (proc_start_info->verb) != 0) {
 608                 shellex.lpVerb = mono_string_chars (proc_start_info->verb);
 609         }
 610
 611         if (proc_start_info->working_directory != NULL &&
 612             mono_string_length (proc_start_info->working_directory) != 0) {
 613                 shellex.lpDirectory = mono_string_chars (proc_start_info->working_directory);
 614         }
 615
 616         if (proc_start_info->error_dialog) {
 617                 shellex.hwnd = proc_start_info->error_dialog_parent_handle;
 618         } else {
 619                 shellex.fMask |= SEE_MASK_FLAG_NO_UI;
 620         }
 621
 622         ret = ShellExecuteEx (&shellex);
 623         if (ret == FALSE) {
 624                 process_info->pid = -GetLastError ();
 625         } else {
 626                 process_info->process_handle = shellex.hProcess;
 627                 process_info->thread_handle = NULL;
 628                 /* It appears that there's no way to get the pid from a
 629                  * process handle before windows xp.  Really.
 630                  */
 631 #if defined(HAVE_GETPROCESSID) && !defined(MONO_CROSS_COMPILE)
 632                 process_info->pid = GetProcessId (shellex.hProcess);
 633 #else
 634                 process_info->pid = 0;
 635 #endif
 636                 process_info->tid = 0;
 637         }
 638
 639         return (ret);
 640 }
 641
 642 MonoBoolean ves_icall_System_Diagnostics_Process_CreateProcess_internal (MonoProcessStartInfo *proc_start_info, HANDLE stdin_handle, HANDLE stdout_handle, HANDLE stderr_handle, MonoProcInfo *process_info)
 643 {
 644         gboolean ret;
 645         gunichar2 *dir;
 646         STARTUPINFO startinfo={0};
 647         PROCESS_INFORMATION procinfo;
 648         gunichar2 *shell_path = NULL;
 649         gchar *env_vars = NULL;
 650         gboolean free_shell_path = TRUE;
 651         gchar *spath = NULL;
 652         MonoString *cmd = proc_start_info->arguments;
 653         guint32 creation_flags, logon_flags;
 654
 655         startinfo.cb=sizeof(STARTUPINFO);
 656         startinfo.dwFlags=STARTF_USESTDHANDLES;
 657         startinfo.hStdInput=stdin_handle;
 658         startinfo.hStdOutput=stdout_handle;
 659         startinfo.hStdError=stderr_handle;
 660
 661         creation_flags = CREATE_UNICODE_ENVIRONMENT;
 662         if (proc_start_info->create_no_window)
 663                 creation_flags |= CREATE_NO_WINDOW;
 664
 665         shell_path = mono_string_chars (proc_start_info->filename);
 666         complete_path (shell_path, &spath);
 667         if (spath == NULL) {
 668                 process_info->pid = -ERROR_FILE_NOT_FOUND;
 669                 return FALSE;
 670         }
 671 #ifdef TARGET_WIN32
 672         /* Seems like our CreateProcess does not work as the windows one.
 673          * This hack is needed to deal with paths containing spaces */
 674         shell_path = NULL;
 675         free_shell_path = FALSE;
 676         if (cmd) {
 677                 gchar *newcmd, *tmp;
 678                 tmp = mono_string_to_utf8 (cmd);
 679                 newcmd = g_strdup_printf ("%s %s", spath, tmp);
 680                 cmd = mono_string_new_wrapper (newcmd);
 681                 g_free (tmp);
 682                 g_free (newcmd);
 683         }
 684         else {
 685                 cmd = mono_string_new_wrapper (spath);
 686         }
 687 #else
 688         shell_path = g_utf8_to_utf16 (spath, -1, NULL, NULL, NULL);
 689 #endif
 690         g_free (spath);
 691
 692         if (process_info->env_keys != NULL) {
 693                 gint i, len;
 694                 MonoString *ms;
 695                 MonoString *key, *value;
 696                 gunichar2 *str, *ptr;
 697                 gunichar2 *equals16;
 698
 699                 for (len = 0, i = 0; i < mono_array_length (process_info->env_keys); i++) {
 700                         ms = mono_array_get (process_info->env_values, MonoString *, i);
 701                         if (ms == NULL)
 702                                 continue;
 703
 704                         len += mono_string_length (ms) * sizeof (gunichar2);
 705                         ms = mono_array_get (process_info->env_keys, MonoString *, i);
 706                         len += mono_string_length (ms) * sizeof (gunichar2);
 707                         len += 2 * sizeof (gunichar2);
 708                 }
 709
 710                 equals16 = g_utf8_to_utf16 ("=", 1, NULL, NULL, NULL);
 711                 ptr = str = g_new0 (gunichar2, len + 1);
 712                 for (i = 0; i < mono_array_length (process_info->env_keys); i++) {
 713                         value = mono_array_get (process_info->env_values, MonoString *, i);
 714                         if (value == NULL)
 715                                 continue;
 716
 717                         key = mono_array_get (process_info->env_keys, MonoString *, i);
 718                         memcpy (ptr, mono_string_chars (key), mono_string_length (key) * sizeof (gunichar2));
 719                         ptr += mono_string_length (key);
 720
 721                         memcpy (ptr, equals16, sizeof (gunichar2));
 722                         ptr++;
 723
 724                         memcpy (ptr, mono_string_chars (value), mono_string_length (value) * sizeof (gunichar2));
 725                         ptr += mono_string_length (value);
 726                         ptr++;
 727                 }
 728
 729                 g_free (equals16);
 730                 env_vars = (gchar *) str;
 731         }
 732
 733         /* The default dir name is "".  Turn that into NULL to mean
 734          * "current directory"
 735          */
 736         if(mono_string_length (proc_start_info->working_directory)==0) {
 737                 dir=NULL;
 738         } else {
 739                 dir=mono_string_chars (proc_start_info->working_directory);
 740         }
 741
 742         if (process_info->username) {
 743                 logon_flags = process_info->load_user_profile ? LOGON_WITH_PROFILE : 0;
 744                 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);
 745         } else {
 746                 ret=CreateProcess (shell_path, cmd? mono_string_chars (cmd): NULL, NULL, NULL, TRUE, creation_flags, env_vars, dir, &startinfo, &procinfo);
 747         }
 748
 749         g_free (env_vars);
 750         if (free_shell_path)
 751                 g_free (shell_path);
 752
 753         if(ret) {
 754                 process_info->process_handle=procinfo.hProcess;
 755                 /*process_info->thread_handle=procinfo.hThread;*/
 756                 process_info->thread_handle=NULL;
 757                 if (procinfo.hThread != NULL && procinfo.hThread != INVALID_HANDLE_VALUE)
 758                         CloseHandle(procinfo.hThread);
 759                 process_info->pid=procinfo.dwProcessId;
 760                 process_info->tid=procinfo.dwThreadId;
 761         } else {
 762                 process_info->pid = -GetLastError ();
 763         }
 764
 765         return(ret);
 766 }
 767
 768 MonoBoolean ves_icall_System_Diagnostics_Process_WaitForExit_internal (MonoObject *this, HANDLE process, gint32 ms)
 769 {
 770         guint32 ret;
 771
 772         MONO_ARCH_SAVE_REGS;
 773
 774         if(ms<0) {
 775                 /* Wait forever */
 776                 ret=WaitForSingleObjectEx (process, INFINITE, TRUE);
 777         } else {
 778                 ret=WaitForSingleObjectEx (process, ms, TRUE);
 779         }
 780         if(ret==WAIT_OBJECT_0) {
 781                 return(TRUE);
 782         } else {
 783                 return(FALSE);
 784         }
 785 }
 786
 787 MonoBoolean ves_icall_System_Diagnostics_Process_WaitForInputIdle_internal (MonoObject *this, HANDLE process, gint32 ms)
 788 {
 789         guint32 ret;
 790
 791         MONO_ARCH_SAVE_REGS;
 792
 793         if(ms<0) {
 794                 /* Wait forever */
 795                 ret=WaitForInputIdle (process, INFINITE);
 796         } else {
 797                 ret=WaitForInputIdle (process, ms);
 798         }
 799
 800         return (ret) ? FALSE : TRUE;
 801 }
 802
 803 gint64 ves_icall_System_Diagnostics_Process_ExitTime_internal (HANDLE process)
 804 {
 805         gboolean ret;
 806         gint64 ticks;
 807         FILETIME create_time, exit_time, kernel_time, user_time;
 808
 809         MONO_ARCH_SAVE_REGS;
 810
 811         ret=GetProcessTimes (process, &create_time, &exit_time, &kernel_time,
 812                              &user_time);
 813         if(ret==TRUE) {
 814                 ticks=((guint64)exit_time.dwHighDateTime << 32) +
 815                         exit_time.dwLowDateTime;
 816
 817                 return(ticks);
 818         } else {
 819                 return(0);
 820         }
 821 }
 822
 823 gint64 ves_icall_System_Diagnostics_Process_StartTime_internal (HANDLE process)
 824 {
 825         gboolean ret;
 826         gint64 ticks;
 827         FILETIME create_time, exit_time, kernel_time, user_time;
 828
 829         MONO_ARCH_SAVE_REGS;
 830
 831         ret=GetProcessTimes (process, &create_time, &exit_time, &kernel_time,
 832                              &user_time);
 833         if(ret==TRUE) {
 834                 ticks=((guint64)create_time.dwHighDateTime << 32) +
 835                         create_time.dwLowDateTime;
 836
 837                 return(ticks);
 838         } else {
 839                 return(0);
 840         }
 841 }
 842
 843 gint32 ves_icall_System_Diagnostics_Process_ExitCode_internal (HANDLE process)
 844 {
 845         DWORD code;
 846
 847         MONO_ARCH_SAVE_REGS;
 848
 849         GetExitCodeProcess (process, &code);
 850
 851         LOGDEBUG (g_message ("%s: process exit code is %d", __func__, code));
 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         LOGDEBUG (g_message ("%s: process name is [%s]", __func__, g_utf16_to_utf8 (name, -1, NULL, NULL, NULL)));
 878
 879         string=mono_string_new_utf16 (mono_domain_get (), name, len);
 880
 881         return(string);
 882 }
 883
 884 /* Returns an array of pids */
 885 MonoArray *ves_icall_System_Diagnostics_Process_GetProcesses_internal (void)
 886 {
 887         MonoArray *procs;
 888         gboolean ret;
 889         DWORD needed;
 890         guint32 count, i;
 891         DWORD pids[1024];
 892
 893         MONO_ARCH_SAVE_REGS;
 894
 895         ret=EnumProcesses (pids, sizeof(pids), &needed);
 896         if(ret==FALSE) {
 897                 /* FIXME: throw an exception */
 898                 return(NULL);
 899         }
 900
 901         count=needed/sizeof(DWORD);
 902         procs=mono_array_new (mono_domain_get (), mono_get_int32_class (),
 903                               count);
 904         for(i=0; i<count; i++) {
 905                 mono_array_set (procs, guint32, i, pids[i]);
 906         }
 907
 908         return(procs);
 909 }
 910
 911 MonoBoolean ves_icall_System_Diagnostics_Process_GetWorkingSet_internal (HANDLE process, guint32 *min, guint32 *max)
 912 {
 913         gboolean ret;
 914         SIZE_T ws_min, ws_max;
 915
 916         MONO_ARCH_SAVE_REGS;
 917
 918         ret=GetProcessWorkingSetSize (process, &ws_min, &ws_max);
 919         *min=(guint32)ws_min;
 920         *max=(guint32)ws_max;
 921
 922         return(ret);
 923 }
 924
 925 MonoBoolean ves_icall_System_Diagnostics_Process_SetWorkingSet_internal (HANDLE process, guint32 min, guint32 max, MonoBoolean use_min)
 926 {
 927         gboolean ret;
 928         SIZE_T ws_min;
 929         SIZE_T ws_max;
 930
 931         MONO_ARCH_SAVE_REGS;
 932
 933         ret=GetProcessWorkingSetSize (process, &ws_min, &ws_max);
 934         if(ret==FALSE) {
 935                 return(FALSE);
 936         }
 937
 938         if(use_min==TRUE) {
 939                 ws_min=(SIZE_T)min;
 940         } else {
 941                 ws_max=(SIZE_T)max;
 942         }
 943
 944         ret=SetProcessWorkingSetSize (process, ws_min, ws_max);
 945
 946         return(ret);
 947 }
 948
 949 MonoBoolean
 950 ves_icall_System_Diagnostics_Process_Kill_internal (HANDLE process, gint32 sig)
 951 {
 952         MONO_ARCH_SAVE_REGS;
 953
 954         /* sig == 1 -> Kill, sig == 2 -> CloseMainWindow */
 955
 956         return TerminateProcess (process, -sig);
 957 }
 958
 959 gint64
 960 ves_icall_System_Diagnostics_Process_Times (HANDLE process, gint32 type)
 961 {
 962         FILETIME create_time, exit_time, kernel_time, user_time;
 963
 964         if (GetProcessTimes (process, &create_time, &exit_time, &kernel_time, &user_time)) {
 965                 if (type == 0)
 966                         return *(gint64*)&user_time;
 967                 else if (type == 1)
 968                         return *(gint64*)&kernel_time;
 969                 /* system + user time: FILETIME can be (memory) cast to a 64 bit int */
 970                 return *(gint64*)&kernel_time + *(gint64*)&user_time;
 971         }
 972         return 0;
 973 }
 974
 975 gint32
 976 ves_icall_System_Diagnostics_Process_GetPriorityClass (HANDLE process, gint32 *error)
 977 {
 978         gint32 ret = GetPriorityClass (process);
 979         *error = ret == 0 ? GetLastError () : 0;
 980         return ret;
 981 }
 982
 983 MonoBoolean
 984 ves_icall_System_Diagnostics_Process_SetPriorityClass (HANDLE process, gint32 priority_class, gint32 *error)
 985 {
 986         gboolean ret = SetPriorityClass (process, priority_class);
 987         *error = ret == 0 ? GetLastError () : 0;
 988         return ret;
 989 }
 990
 991 HANDLE
 992 ves_icall_System_Diagnostics_Process_ProcessHandle_duplicate (HANDLE process)
 993 {
 994         HANDLE ret;
 995
 996         MONO_ARCH_SAVE_REGS;
 997
 998         LOGDEBUG (g_message ("%s: Duplicating process handle %p", __func__, process));
 999
1000         DuplicateHandle (GetCurrentProcess (), process, GetCurrentProcess (),
1001                          &ret, THREAD_ALL_ACCESS, TRUE, 0);
1002
1003         return ret;
1004 }
1005
1006 void
1007 ves_icall_System_Diagnostics_Process_ProcessHandle_close (HANDLE process)
1008 {
1009         MONO_ARCH_SAVE_REGS;
1010
1011         LOGDEBUG (g_message ("%s: Closing process handle %p", __func__, process));
1012
1013         CloseHandle (process);
1014 }
1015
1016 gint64
1017 ves_icall_System_Diagnostics_Process_GetProcessData (int pid, gint32 data_type, gint32 *error)
1018 {
1019         MonoProcessError perror;
1020         guint64 res;
1021
1022         res = mono_process_get_data_with_error (GINT_TO_POINTER (pid), data_type, &perror);
1023         if (error)
1024                 *error = perror;
1025         return res;
1026 }
1027