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