mono/metadata/process.c

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