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