mono/io-layer/processes.c

   1 /*
   2  * processes.c:  Process handles
   3  *
   4  * Author:
   5  *      Dick Porter (dick@ximian.com)
   6  *
   7  * (C) 2002-2011 Novell, Inc.
   8  * Copyright 2011 Xamarin Inc
   9  */
  10
  11 #include <config.h>
  12 #include <glib.h>
  13 #include <stdio.h>
  14 #include <string.h>
  15 #include <pthread.h>
  16 #include <sched.h>
  17 #include <sys/time.h>
  18 #include <errno.h>
  19 #include <sys/types.h>
  20 #include <sys/stat.h>
  21 #include <unistd.h>
  22 #ifdef HAVE_SIGNAL_H
  23 #include <signal.h>
  24 #endif
  25 #include <sys/time.h>
  26 #include <fcntl.h>
  27 #ifdef HAVE_SYS_PARAM_H
  28 #include <sys/param.h>
  29 #endif
  30 #include <ctype.h>
  31
  32 #ifdef HAVE_SYS_WAIT_H
  33 #include <sys/wait.h>
  34 #endif
  35 #ifdef HAVE_SYS_RESOURCE_H
  36 #include <sys/resource.h>
  37 #endif
  38
  39 #ifdef HAVE_SYS_MKDEV_H
  40 #include <sys/mkdev.h>
  41 #endif
  42
  43 #ifdef HAVE_UTIME_H
  44 #include <utime.h>
  45 #endif
  46
  47 /* sys/resource.h (for rusage) is required when using osx 10.3 (but not 10.4) */
  48 #ifdef __APPLE__
  49 #include <TargetConditionals.h>
  50 #include <sys/resource.h>
  51 #ifdef HAVE_LIBPROC_H
  52 /* proc_name */
  53 #include <libproc.h>
  54 #endif
  55 #endif
  56
  57 #if defined(PLATFORM_MACOSX) || defined(__OpenBSD__)
  58 #include <sys/proc.h>
  59 #include <sys/sysctl.h>
  60 #  if !defined(__OpenBSD__)
  61 #    include <sys/utsname.h>
  62 #  endif
  63 #endif
  64
  65 #ifdef PLATFORM_SOLARIS
  66 /* procfs.h cannot be included if this define is set, but it seems to work fine if it is undefined */
  67 #if _FILE_OFFSET_BITS == 64
  68 #undef _FILE_OFFSET_BITS
  69 #include <procfs.h>
  70 #define _FILE_OFFSET_BITS 64
  71 #else
  72 #include <procfs.h>
  73 #endif
  74 #endif
  75
  76 #ifdef __HAIKU__
  77 #include <KernelKit.h>
  78 #endif
  79
  80 #include <mono/io-layer/wapi.h>
  81 #include <mono/io-layer/wapi-private.h>
  82 #include <mono/io-layer/handles-private.h>
  83 #include <mono/io-layer/misc-private.h>
  84 #include <mono/io-layer/process-private.h>
  85 #include <mono/io-layer/threads.h>
  86 #include <mono/utils/strenc.h>
  87 #include <mono/utils/mono-path.h>
  88 #include <mono/io-layer/timefuncs-private.h>
  89 #include <mono/utils/mono-time.h>
  90 #include <mono/utils/mono-membar.h>
  91 #include <mono/utils/mono-mutex.h>
  92 #include <mono/utils/mono-signal-handler.h>
  93 #include <mono/utils/mono-proclib.h>
  94
  95 /* The process' environment strings */
  96 #if defined(__APPLE__) && !defined (__arm__) && !defined (__aarch64__)
  97 /* Apple defines this in crt_externs.h but doesn't provide that header for
  98  * arm-apple-darwin9.  We'll manually define the symbol on Apple as it does
  99  * in fact exist on all implementations (so far)
 100  */
 101 char ***_NSGetEnviron(void);
 102 #define environ (*_NSGetEnviron())
 103 #else
 104 extern char **environ;
 105 #endif
 106
 107 #if 0
 108 #define DEBUG(...) g_message(__VA_ARGS__)
 109 #define DEBUG_ENABLED 1
 110 #else
 111 #define DEBUG(...)
 112 #endif
 113
 114 static guint32 process_wait (gpointer handle, guint32 timeout, gboolean alertable);
 115 static void process_close (gpointer handle, gpointer data);
 116 static gboolean is_pid_valid (pid_t pid);
 117
 118 #if !(defined(PLATFORM_MACOSX) || defined(__OpenBSD__) || defined(__HAIKU__))
 119 static FILE *
 120 open_process_map (int pid, const char *mode);
 121 #endif
 122
 123 struct _WapiHandleOps _wapi_process_ops = {
 124         process_close,          /* close_shared */
 125         NULL,                           /* signal */
 126         NULL,                           /* own */
 127         NULL,                           /* is_owned */
 128         process_wait,                   /* special_wait */
 129         NULL                            /* prewait */
 130 };
 131
 132 #if HAVE_SIGACTION
 133 static struct sigaction previous_chld_sa;
 134 #endif
 135 static mono_once_t process_sig_chld_once = MONO_ONCE_INIT;
 136 static void process_add_sigchld_handler (void);
 137
 138 /* The signal-safe logic to use mono_processes goes like this:
 139  * - The list must be safe to traverse for the signal handler at all times.
 140  *   It's safe to: prepend an entry (which is a single store to 'mono_processes'),
 141  *   unlink an entry (assuming the unlinked entry isn't freed and doesn't
 142  *   change its 'next' pointer so that it can still be traversed).
 143  * When cleaning up we first unlink an entry, then we verify that
 144  * the read lock isn't locked. Then we can free the entry, since
 145  * we know that nobody is using the old version of the list (including
 146  * the unlinked entry).
 147  * We also need to lock when adding and cleaning up so that those two
 148  * operations don't mess with eachother. (This lock is not used in the
 149  * signal handler)
 150  */
 151 static struct MonoProcess *mono_processes = NULL;
 152 static volatile gint32 mono_processes_cleaning_up = 0;
 153 static mono_mutex_t mono_processes_mutex;
 154 static void mono_processes_cleanup (void);
 155
 156 static gpointer current_process;
 157 static char *cli_launcher;
 158
 159 static WapiHandle_process *
 160 lookup_process_handle (gpointer handle)
 161 {
 162         WapiHandle_process *process_data;
 163         gboolean ret;
 164
 165         ret = _wapi_lookup_handle (handle, WAPI_HANDLE_PROCESS,
 166                                                            (gpointer *)&process_data);
 167         if (!ret)
 168                 return NULL;
 169         return process_data;
 170 }
 171
 172 /* Check if a pid is valid - i.e. if a process exists with this pid. */
 173 static gboolean
 174 is_pid_valid (pid_t pid)
 175 {
 176         gboolean result = FALSE;
 177
 178 #if defined(PLATFORM_MACOSX) || defined(__OpenBSD__)
 179         if (((kill(pid, 0) == 0) || (errno == EPERM)) && pid != 0)
 180                 result = TRUE;
 181 #elif defined(__HAIKU__)
 182         team_info teamInfo;
 183         if (get_team_info ((team_id)pid, &teamInfo) == B_OK)
 184                 result = TRUE;
 185 #else
 186         char *dir = g_strdup_printf ("/proc/%d", pid);
 187         if (!access (dir, F_OK))
 188                 result = TRUE;
 189         g_free (dir);
 190 #endif
 191
 192         return result;
 193 }
 194
 195 static void
 196 process_set_defaults (WapiHandle_process *process_handle)
 197 {
 198         /* These seem to be the defaults on w2k */
 199         process_handle->min_working_set = 204800;
 200         process_handle->max_working_set = 1413120;
 201
 202         _wapi_time_t_to_filetime (time (NULL), &process_handle->create_time);
 203 }
 204
 205 static int
 206 len16 (const gunichar2 *str)
 207 {
 208         int len = 0;
 209
 210         while (*str++ != 0)
 211                 len++;
 212
 213         return len;
 214 }
 215
 216 static gunichar2 *
 217 utf16_concat (const gunichar2 *first, ...)
 218 {
 219         va_list args;
 220         int total = 0, i;
 221         const gunichar2 *s;
 222         gunichar2 *ret;
 223
 224         va_start (args, first);
 225         total += len16 (first);
 226         for (s = va_arg (args, gunichar2 *); s != NULL; s = va_arg(args, gunichar2 *)){
 227                 total += len16 (s);
 228         }
 229         va_end (args);
 230
 231         ret = g_new (gunichar2, total + 1);
 232         if (ret == NULL)
 233                 return NULL;
 234
 235         ret [total] = 0;
 236         i = 0;
 237         for (s = first; *s != 0; s++)
 238                 ret [i++] = *s;
 239         va_start (args, first);
 240         for (s = va_arg (args, gunichar2 *); s != NULL; s = va_arg (args, gunichar2 *)){
 241                 const gunichar2 *p;
 242
 243                 for (p = s; *p != 0; p++)
 244                         ret [i++] = *p;
 245         }
 246         va_end (args);
 247
 248         return ret;
 249 }
 250
 251 static const gunichar2 utf16_space_bytes [2] = { 0x20, 0 };
 252 static const gunichar2 *utf16_space = utf16_space_bytes;
 253 static const gunichar2 utf16_quote_bytes [2] = { 0x22, 0 };
 254 static const gunichar2 *utf16_quote = utf16_quote_bytes;
 255
 256 #ifdef DEBUG_ENABLED
 257 /* Useful in gdb */
 258 void
 259 print_utf16 (gunichar2 *str)
 260 {
 261         char *res;
 262
 263         res = g_utf16_to_utf8 (str, -1, NULL, NULL, NULL);
 264         g_print ("%s\n", res);
 265         g_free (res);
 266 }
 267 #endif
 268
 269 /* Implemented as just a wrapper around CreateProcess () */
 270 gboolean
 271 ShellExecuteEx (WapiShellExecuteInfo *sei)
 272 {
 273         gboolean ret;
 274         WapiProcessInformation process_info;
 275         gunichar2 *args;
 276
 277         if (sei == NULL) {
 278                 /* w2k just segfaults here, but we can do better than
 279                  * that
 280                  */
 281                 SetLastError (ERROR_INVALID_PARAMETER);
 282                 return FALSE;
 283         }
 284
 285         if (sei->lpFile == NULL)
 286                 /* w2k returns TRUE for this, for some reason. */
 287                 return TRUE;
 288
 289         /* Put both executable and parameters into the second argument
 290          * to CreateProcess (), so it searches $PATH.  The conversion
 291          * into and back out of utf8 is because there is no
 292          * g_strdup_printf () equivalent for gunichar2 :-(
 293          */
 294         args = utf16_concat (utf16_quote, sei->lpFile, utf16_quote, sei->lpParameters == NULL ? NULL : utf16_space, sei->lpParameters, NULL);
 295         if (args == NULL) {
 296                 SetLastError (ERROR_INVALID_DATA);
 297                 return FALSE;
 298         }
 299         ret = CreateProcess (NULL, args, NULL, NULL, TRUE,
 300                              CREATE_UNICODE_ENVIRONMENT, NULL,
 301                              sei->lpDirectory, NULL, &process_info);
 302         g_free (args);
 303
 304         if (!ret && GetLastError () == ERROR_OUTOFMEMORY)
 305                 return ret;
 306
 307         if (!ret) {
 308                 static char *handler;
 309                 static gunichar2 *handler_utf16;
 310
 311                 if (handler_utf16 == (gunichar2 *)-1)
 312                         return FALSE;
 313
 314 #ifdef PLATFORM_MACOSX
 315                 handler = g_strdup ("/usr/bin/open");
 316 #else
 317                 /*
 318                  * On Linux, try: xdg-open, the FreeDesktop standard way of doing it,
 319                  * if that fails, try to use gnome-open, then kfmclient
 320                  */
 321                 handler = g_find_program_in_path ("xdg-open");
 322                 if (handler == NULL){
 323                         handler = g_find_program_in_path ("gnome-open");
 324                         if (handler == NULL){
 325                                 handler = g_find_program_in_path ("kfmclient");
 326                                 if (handler == NULL){
 327                                         handler_utf16 = (gunichar2 *) -1;
 328                                         return FALSE;
 329                                 } else {
 330                                         /* kfmclient needs exec argument */
 331                                         char *old = handler;
 332                                         handler = g_strconcat (old, " exec",
 333                                                                NULL);
 334                                         g_free (old);
 335                                 }
 336                         }
 337                 }
 338 #endif
 339                 handler_utf16 = g_utf8_to_utf16 (handler, -1, NULL, NULL, NULL);
 340                 g_free (handler);
 341
 342                 /* Put quotes around the filename, in case it's a url
 343                  * that contains #'s (CreateProcess() calls
 344                  * g_shell_parse_argv(), which deliberately throws
 345                  * away anything after an unquoted #).  Fixes bug
 346                  * 371567.
 347                  */
 348                 args = utf16_concat (handler_utf16, utf16_space, utf16_quote,
 349                                      sei->lpFile, utf16_quote,
 350                                      sei->lpParameters == NULL ? NULL : utf16_space,
 351                                      sei->lpParameters, NULL);
 352                 if (args == NULL) {
 353                         SetLastError (ERROR_INVALID_DATA);
 354                         return FALSE;
 355                 }
 356                 ret = CreateProcess (NULL, args, NULL, NULL, TRUE,
 357                                      CREATE_UNICODE_ENVIRONMENT, NULL,
 358                                      sei->lpDirectory, NULL, &process_info);
 359                 g_free (args);
 360                 if (!ret) {
 361                         if (GetLastError () != ERROR_OUTOFMEMORY)
 362                                 SetLastError (ERROR_INVALID_DATA);
 363                         return FALSE;
 364                 }
 365                 /* Shell exec should not return a process handle when it spawned a GUI thing, like a browser. */
 366                 CloseHandle (process_info.hProcess);
 367                 process_info.hProcess = NULL;
 368         }
 369
 370         if (sei->fMask & SEE_MASK_NOCLOSEPROCESS)
 371                 sei->hProcess = process_info.hProcess;
 372         else
 373                 CloseHandle (process_info.hProcess);
 374
 375         return ret;
 376 }
 377
 378 static gboolean
 379 is_managed_binary (const char *filename)
 380 {
 381         int original_errno = errno;
 382 #if defined(HAVE_LARGE_FILE_SUPPORT) && defined(O_LARGEFILE)
 383         int file = open (filename, O_RDONLY | O_LARGEFILE);
 384 #else
 385         int file = open (filename, O_RDONLY);
 386 #endif
 387         off_t new_offset;
 388         unsigned char buffer[8];
 389         off_t file_size, optional_header_offset;
 390         off_t pe_header_offset;
 391         gboolean managed = FALSE;
 392         int num_read;
 393         guint32 first_word, second_word;
 394
 395         /* If we are unable to open the file, then we definitely
 396          * can't say that it is managed. The child mono process
 397          * probably wouldn't be able to open it anyway.
 398          */
 399         if (file < 0) {
 400                 errno = original_errno;
 401                 return FALSE;
 402         }
 403
 404         /* Retrieve the length of the file for future sanity checks. */
 405         file_size = lseek (file, 0, SEEK_END);
 406         lseek (file, 0, SEEK_SET);
 407
 408         /* We know we need to read a header field at offset 60. */
 409         if (file_size < 64)
 410                 goto leave;
 411
 412         num_read = read (file, buffer, 2);
 413
 414         if ((num_read != 2) || (buffer[0] != 'M') || (buffer[1] != 'Z'))
 415                 goto leave;
 416
 417         new_offset = lseek (file, 60, SEEK_SET);
 418
 419         if (new_offset != 60)
 420                 goto leave;
 421
 422         num_read = read (file, buffer, 4);
 423
 424         if (num_read != 4)
 425                 goto leave;
 426         pe_header_offset =  buffer[0]
 427                 | (buffer[1] <<  8)
 428                 | (buffer[2] << 16)
 429                 | (buffer[3] << 24);
 430
 431         if (pe_header_offset + 24 > file_size)
 432                 goto leave;
 433
 434         new_offset = lseek (file, pe_header_offset, SEEK_SET);
 435
 436         if (new_offset != pe_header_offset)
 437                 goto leave;
 438
 439         num_read = read (file, buffer, 4);
 440
 441         if ((num_read != 4) || (buffer[0] != 'P') || (buffer[1] != 'E') || (buffer[2] != 0) || (buffer[3] != 0))
 442                 goto leave;
 443
 444         /*
 445          * Verify that the header we want in the optional header data
 446          * is present in this binary.
 447          */
 448         new_offset = lseek (file, pe_header_offset + 20, SEEK_SET);
 449
 450         if (new_offset != pe_header_offset + 20)
 451                 goto leave;
 452
 453         num_read = read (file, buffer, 2);
 454
 455         if ((num_read != 2) || ((buffer[0] | (buffer[1] << 8)) < 216))
 456                 goto leave;
 457
 458         /* Read the CLR header address and size fields. These will be
 459          * zero if the binary is not managed.
 460          */
 461         optional_header_offset = pe_header_offset + 24;
 462         new_offset = lseek (file, optional_header_offset + 208, SEEK_SET);
 463
 464         if (new_offset != optional_header_offset + 208)
 465                 goto leave;
 466
 467         num_read = read (file, buffer, 8);
 468
 469         /* We are not concerned with endianness, only with
 470          * whether it is zero or not.
 471          */
 472         first_word = *(guint32 *)&buffer[0];
 473         second_word = *(guint32 *)&buffer[4];
 474
 475         if ((num_read != 8) || (first_word == 0) || (second_word == 0))
 476                 goto leave;
 477
 478         managed = TRUE;
 479
 480 leave:
 481         close (file);
 482         errno = original_errno;
 483         return managed;
 484 }
 485
 486 gboolean
 487 CreateProcessWithLogonW (const gunichar2 *username,
 488                                                  const gunichar2 *domain,
 489                                                  const gunichar2 *password,
 490                                                  const guint32 logonFlags,
 491                                                  const gunichar2 *appname,
 492                                                  const gunichar2 *cmdline,
 493                                                  guint32 create_flags,
 494                                                  gpointer env,
 495                                                  const gunichar2 *cwd,
 496                                                  WapiStartupInfo *startup,
 497                                                  WapiProcessInformation *process_info)
 498 {
 499         /* FIXME: use user information */
 500         return CreateProcess (appname, cmdline, NULL, NULL, FALSE, create_flags, env, cwd, startup, process_info);
 501 }
 502
 503 static gboolean
 504 is_readable_or_executable (const char *prog)
 505 {
 506         struct stat buf;
 507         int a = access (prog, R_OK);
 508         int b = access (prog, X_OK);
 509         if (a != 0 && b != 0)
 510                 return FALSE;
 511         if (stat (prog, &buf))
 512                 return FALSE;
 513         if (S_ISREG (buf.st_mode))
 514                 return TRUE;
 515         return FALSE;
 516 }
 517
 518 static gboolean
 519 is_executable (const char *prog)
 520 {
 521         struct stat buf;
 522         if (access (prog, X_OK) != 0)
 523                 return FALSE;
 524         if (stat (prog, &buf))
 525                 return FALSE;
 526         if (S_ISREG (buf.st_mode))
 527                 return TRUE;
 528         return FALSE;
 529 }
 530
 531 static void
 532 switch_dir_separators (char *path)
 533 {
 534         size_t i, pathLength = strlen(path);
 535
 536         /* Turn all the slashes round the right way, except for \' */
 537         /* There are probably other characters that need to be excluded as well. */
 538         for (i = 0; i < pathLength; i++) {
 539                 if (path[i] == '\\' && i < pathLength - 1 && path[i+1] != '\'' )
 540                         path[i] = '/';
 541         }
 542 }
 543
 544 gboolean CreateProcess (const gunichar2 *appname, const gunichar2 *cmdline,
 545                         WapiSecurityAttributes *process_attrs G_GNUC_UNUSED,
 546                         WapiSecurityAttributes *thread_attrs G_GNUC_UNUSED,
 547                         gboolean inherit_handles, guint32 create_flags,
 548                         gpointer new_environ, const gunichar2 *cwd,
 549                         WapiStartupInfo *startup,
 550                         WapiProcessInformation *process_info)
 551 {
 552         char *cmd = NULL, *prog = NULL, *full_prog = NULL, *args = NULL, *args_after_prog = NULL;
 553         char *dir = NULL, **env_strings = NULL, **argv = NULL;
 554         guint32 i, env_count = 0;
 555         gboolean ret = FALSE;
 556         gpointer handle;
 557         WapiHandle_process process_handle = {0}, *process_handle_data;
 558         GError *gerr = NULL;
 559         int in_fd, out_fd, err_fd;
 560         pid_t pid;
 561         int thr_ret;
 562         int startup_pipe [2] = {-1, -1};
 563         int dummy;
 564         struct MonoProcess *mono_process;
 565         gboolean fork_failed = FALSE;
 566
 567         mono_once (&process_sig_chld_once, process_add_sigchld_handler);
 568
 569         /* appname and cmdline specify the executable and its args:
 570          *
 571          * If appname is not NULL, it is the name of the executable.
 572          * Otherwise the executable is the first token in cmdline.
 573          *
 574          * Executable searching:
 575          *
 576          * If appname is not NULL, it can specify the full path and
 577          * file name, or else a partial name and the current directory
 578          * will be used.  There is no additional searching.
 579          *
 580          * If appname is NULL, the first whitespace-delimited token in
 581          * cmdline is used.  If the name does not contain a full
 582          * directory path, the search sequence is:
 583          *
 584          * 1) The directory containing the current process
 585          * 2) The current working directory
 586          * 3) The windows system directory  (Ignored)
 587          * 4) The windows directory (Ignored)
 588          * 5) $PATH
 589          *
 590          * Just to make things more interesting, tokens can contain
 591          * white space if they are surrounded by quotation marks.  I'm
 592          * beginning to understand just why windows apps are generally
 593          * so crap, with an API like this :-(
 594          */
 595         if (appname != NULL) {
 596                 cmd = mono_unicode_to_external (appname);
 597                 if (cmd == NULL) {
 598                         DEBUG ("%s: unicode conversion returned NULL",
 599                                    __func__);
 600
 601                         SetLastError (ERROR_PATH_NOT_FOUND);
 602                         goto free_strings;
 603                 }
 604
 605                 switch_dir_separators(cmd);
 606         }
 607
 608         if (cmdline != NULL) {
 609                 args = mono_unicode_to_external (cmdline);
 610                 if (args == NULL) {
 611                         DEBUG ("%s: unicode conversion returned NULL", __func__);
 612
 613                         SetLastError (ERROR_PATH_NOT_FOUND);
 614                         goto free_strings;
 615                 }
 616         }
 617
 618         if (cwd != NULL) {
 619                 dir = mono_unicode_to_external (cwd);
 620                 if (dir == NULL) {
 621                         DEBUG ("%s: unicode conversion returned NULL", __func__);
 622
 623                         SetLastError (ERROR_PATH_NOT_FOUND);
 624                         goto free_strings;
 625                 }
 626
 627                 /* Turn all the slashes round the right way */
 628                 switch_dir_separators(dir);
 629         }
 630
 631
 632         /* We can't put off locating the executable any longer :-( */
 633         if (cmd != NULL) {
 634                 char *unquoted;
 635                 if (g_ascii_isalpha (cmd[0]) && (cmd[1] == ':')) {
 636                         /* Strip off the drive letter.  I can't
 637                          * believe that CP/M holdover is still
 638                          * visible...
 639                          */
 640                         g_memmove (cmd, cmd+2, strlen (cmd)-2);
 641                         cmd[strlen (cmd)-2] = '\0';
 642                 }
 643
 644                 unquoted = g_shell_unquote (cmd, NULL);
 645                 if (unquoted[0] == '/') {
 646                         /* Assume full path given */
 647                         prog = g_strdup (unquoted);
 648
 649                         /* Executable existing ? */
 650                         if (!is_readable_or_executable (prog)) {
 651                                 DEBUG ("%s: Couldn't find executable %s",
 652                                            __func__, prog);
 653                                 g_free (unquoted);
 654                                 SetLastError (ERROR_FILE_NOT_FOUND);
 655                                 goto free_strings;
 656                         }
 657                 } else {
 658                         /* Search for file named by cmd in the current
 659                          * directory
 660                          */
 661                         char *curdir = g_get_current_dir ();
 662
 663                         prog = g_strdup_printf ("%s/%s", curdir, unquoted);
 664                         g_free (curdir);
 665
 666                         /* And make sure it's readable */
 667                         if (!is_readable_or_executable (prog)) {
 668                                 DEBUG ("%s: Couldn't find executable %s",
 669                                            __func__, prog);
 670                                 g_free (unquoted);
 671                                 SetLastError (ERROR_FILE_NOT_FOUND);
 672                                 goto free_strings;
 673                         }
 674                 }
 675                 g_free (unquoted);
 676
 677                 args_after_prog = args;
 678         } else {
 679                 char *token = NULL;
 680                 char quote;
 681
 682                 /* Dig out the first token from args, taking quotation
 683                  * marks into account
 684                  */
 685
 686                 /* First, strip off all leading whitespace */
 687                 args = g_strchug (args);
 688
 689                 /* args_after_prog points to the contents of args
 690                  * after token has been set (otherwise argv[0] is
 691                  * duplicated)
 692                  */
 693                 args_after_prog = args;
 694
 695                 /* Assume the opening quote will always be the first
 696                  * character
 697                  */
 698                 if (args[0] == '\"' || args [0] == '\'') {
 699                         quote = args [0];
 700                         for (i = 1; args[i] != '\0' && args[i] != quote; i++);
 701                         if (args [i + 1] == '\0' || g_ascii_isspace (args[i+1])) {
 702                                 /* We found the first token */
 703                                 token = g_strndup (args+1, i-1);
 704                                 args_after_prog = g_strchug (args + i + 1);
 705                         } else {
 706                                 /* Quotation mark appeared in the
 707                                  * middle of the token.  Just give the
 708                                  * whole first token, quotes and all,
 709                                  * to exec.
 710                                  */
 711                         }
 712                 }
 713
 714                 if (token == NULL) {
 715                         /* No quote mark, or malformed */
 716                         for (i = 0; args[i] != '\0'; i++) {
 717                                 if (g_ascii_isspace (args[i])) {
 718                                         token = g_strndup (args, i);
 719                                         args_after_prog = args + i + 1;
 720                                         break;
 721                                 }
 722                         }
 723                 }
 724
 725                 if (token == NULL && args[0] != '\0') {
 726                         /* Must be just one token in the string */
 727                         token = g_strdup (args);
 728                         args_after_prog = NULL;
 729                 }
 730
 731                 if (token == NULL) {
 732                         /* Give up */
 733                         DEBUG ("%s: Couldn't find what to exec", __func__);
 734
 735                         SetLastError (ERROR_PATH_NOT_FOUND);
 736                         goto free_strings;
 737                 }
 738
 739                 /* Turn all the slashes round the right way. Only for
 740                  * the prg. name
 741                  */
 742                 switch_dir_separators(token);
 743
 744                 if (g_ascii_isalpha (token[0]) && (token[1] == ':')) {
 745                         /* Strip off the drive letter.  I can't
 746                          * believe that CP/M holdover is still
 747                          * visible...
 748                          */
 749                         g_memmove (token, token+2, strlen (token)-2);
 750                         token[strlen (token)-2] = '\0';
 751                 }
 752
 753                 if (token[0] == '/') {
 754                         /* Assume full path given */
 755                         prog = g_strdup (token);
 756
 757                         /* Executable existing ? */
 758                         if (!is_readable_or_executable (prog)) {
 759                                 DEBUG ("%s: Couldn't find executable %s",
 760                                            __func__, token);
 761                                 g_free (token);
 762                                 SetLastError (ERROR_FILE_NOT_FOUND);
 763                                 goto free_strings;
 764                         }
 765                 } else {
 766                         char *curdir = g_get_current_dir ();
 767
 768                         /* FIXME: Need to record the directory
 769                          * containing the current process, and check
 770                          * that for the new executable as the first
 771                          * place to look
 772                          */
 773
 774                         prog = g_strdup_printf ("%s/%s", curdir, token);
 775                         g_free (curdir);
 776
 777                         /* I assume X_OK is the criterion to use,
 778                          * rather than F_OK
 779                          *
 780                          * X_OK is too strict *if* the target is a CLR binary
 781                          */
 782                         if (!is_readable_or_executable (prog)) {
 783                                 g_free (prog);
 784                                 prog = g_find_program_in_path (token);
 785                                 if (prog == NULL) {
 786                                         DEBUG ("%s: Couldn't find executable %s", __func__, token);
 787
 788                                         g_free (token);
 789                                         SetLastError (ERROR_FILE_NOT_FOUND);
 790                                         goto free_strings;
 791                                 }
 792                         }
 793                 }
 794
 795                 g_free (token);
 796         }
 797
 798         DEBUG ("%s: Exec prog [%s] args [%s]", __func__, prog,
 799                    args_after_prog);
 800
 801         /* Check for CLR binaries; if found, we will try to invoke
 802          * them using the same mono binary that started us.
 803          */
 804         if (is_managed_binary (prog)) {
 805                 gunichar2 *newapp = NULL, *newcmd;
 806                 gsize bytes_ignored;
 807
 808                 if (cli_launcher)
 809                         newapp = mono_unicode_from_external (cli_launcher, &bytes_ignored);
 810                 else
 811                         newapp = mono_unicode_from_external ("mono", &bytes_ignored);
 812
 813                 if (newapp != NULL) {
 814                         if (appname != NULL) {
 815                                 newcmd = utf16_concat (newapp, utf16_space,
 816                                                        appname, utf16_space,
 817                                                        cmdline, NULL);
 818                         } else {
 819                                 newcmd = utf16_concat (newapp, utf16_space,
 820                                                        cmdline, NULL);
 821                         }
 822
 823                         g_free ((gunichar2 *)newapp);
 824
 825                         if (newcmd != NULL) {
 826                                 ret = CreateProcess (NULL, newcmd,
 827                                                      process_attrs,
 828                                                      thread_attrs,
 829                                                      inherit_handles,
 830                                                      create_flags, new_environ,
 831                                                      cwd, startup,
 832                                                      process_info);
 833
 834                                 g_free ((gunichar2 *)newcmd);
 835
 836                                 goto free_strings;
 837                         }
 838                 }
 839         } else {
 840                 if (!is_executable (prog)) {
 841                         DEBUG ("%s: Executable permisson not set on %s", __func__, prog);
 842                         g_free (prog);
 843                         SetLastError (ERROR_ACCESS_DENIED);
 844                         goto free_strings;
 845                 }
 846         }
 847
 848         if (args_after_prog != NULL && *args_after_prog) {
 849                 char *qprog;
 850
 851                 qprog = g_shell_quote (prog);
 852                 full_prog = g_strconcat (qprog, " ", args_after_prog, NULL);
 853                 g_free (qprog);
 854         } else {
 855                 full_prog = g_shell_quote (prog);
 856         }
 857
 858         ret = g_shell_parse_argv (full_prog, NULL, &argv, &gerr);
 859         if (ret == FALSE) {
 860                 g_message ("CreateProcess: %s\n", gerr->message);
 861                 g_error_free (gerr);
 862                 gerr = NULL;
 863                 goto free_strings;
 864         }
 865
 866         if (startup != NULL && startup->dwFlags & STARTF_USESTDHANDLES) {
 867                 in_fd = GPOINTER_TO_UINT (startup->hStdInput);
 868                 out_fd = GPOINTER_TO_UINT (startup->hStdOutput);
 869                 err_fd = GPOINTER_TO_UINT (startup->hStdError);
 870         } else {
 871                 in_fd = GPOINTER_TO_UINT (GetStdHandle (STD_INPUT_HANDLE));
 872                 out_fd = GPOINTER_TO_UINT (GetStdHandle (STD_OUTPUT_HANDLE));
 873                 err_fd = GPOINTER_TO_UINT (GetStdHandle (STD_ERROR_HANDLE));
 874         }
 875
 876         process_handle.proc_name = g_strdup (prog);
 877
 878         process_set_defaults (&process_handle);
 879
 880         handle = _wapi_handle_new (WAPI_HANDLE_PROCESS, &process_handle);
 881         if (handle == _WAPI_HANDLE_INVALID) {
 882                 g_warning ("%s: error creating process handle", __func__);
 883
 884                 ret = FALSE;
 885                 SetLastError (ERROR_OUTOFMEMORY);
 886                 goto free_strings;
 887         }
 888
 889         /* new_environ is a block of NULL-terminated strings, which
 890          * is itself NULL-terminated. Of course, passing an array of
 891          * string pointers would have made things too easy :-(
 892          *
 893          * If new_environ is not NULL it specifies the entire set of
 894          * environment variables in the new process.  Otherwise the
 895          * new process inherits the same environment.
 896          */
 897         if (new_environ) {
 898                 gunichar2 *new_environp;
 899
 900                 /* Count the number of strings */
 901                 for (new_environp = (gunichar2 *)new_environ; *new_environp;
 902                      new_environp++) {
 903                         env_count++;
 904                         while (*new_environp) {
 905                                 new_environp++;
 906                         }
 907                 }
 908
 909                 /* +2: one for the process handle value, and the last
 910                  * one is NULL
 911                  */
 912                 env_strings = g_new0 (char *, env_count + 2);
 913
 914                 /* Copy each environ string into 'strings' turning it
 915                  * into utf8 (or the requested encoding) at the same
 916                  * time
 917                  */
 918                 env_count = 0;
 919                 for (new_environp = (gunichar2 *)new_environ; *new_environp;
 920                      new_environp++) {
 921                         env_strings[env_count] = mono_unicode_to_external (new_environp);
 922                         env_count++;
 923                         while (*new_environp) {
 924                                 new_environp++;
 925                         }
 926                 }
 927         } else {
 928                 for (i = 0; environ[i] != NULL; i++)
 929                         env_count++;
 930
 931                 /* +2: one for the process handle value, and the last
 932                  * one is NULL
 933                  */
 934                 env_strings = g_new0 (char *, env_count + 2);
 935
 936                 /* Copy each environ string into 'strings' turning it
 937                  * into utf8 (or the requested encoding) at the same
 938                  * time
 939                  */
 940                 env_count = 0;
 941                 for (i = 0; environ[i] != NULL; i++) {
 942                         env_strings[env_count] = g_strdup (environ[i]);
 943                         env_count++;
 944                 }
 945         }
 946
 947         /* Create a pipe to make sure the child doesn't exit before
 948          * we can add the process to the linked list of mono_processes */
 949         if (pipe (startup_pipe) == -1) {
 950                 /* Could not create the pipe to synchroniz process startup. We'll just not synchronize.
 951                  * This is just for a very hard to hit race condition in the first place */
 952                 startup_pipe [0] = startup_pipe [1] = -1;
 953                 DEBUG ("%s: new process startup not synchronized. We may not notice if the newly created process exits immediately.", __func__);
 954         }
 955
 956         thr_ret = _wapi_handle_lock_shared_handles ();
 957         g_assert (thr_ret == 0);
 958
 959         pid = fork ();
 960         if (pid == -1) {
 961                 /* Error */
 962                 SetLastError (ERROR_OUTOFMEMORY);
 963                 ret = FALSE;
 964                 fork_failed = TRUE;
 965                 goto cleanup;
 966         } else if (pid == 0) {
 967                 /* Child */
 968
 969                 if (startup_pipe [0] != -1) {
 970                         /* Wait until the parent has updated it's internal data */
 971                         ssize_t _i G_GNUC_UNUSED = read (startup_pipe [0], &dummy, 1);
 972                         DEBUG ("%s: child: parent has completed its setup", __func__);
 973                         close (startup_pipe [0]);
 974                         close (startup_pipe [1]);
 975                 }
 976
 977                 /* should we detach from the process group? */
 978
 979                 /* Connect stdin, stdout and stderr */
 980                 dup2 (in_fd, 0);
 981                 dup2 (out_fd, 1);
 982                 dup2 (err_fd, 2);
 983
 984                 if (inherit_handles != TRUE) {
 985                         /* FIXME: do something here */
 986                 }
 987
 988                 /* Close all file descriptors */
 989                 for (i = wapi_getdtablesize () - 1; i > 2; i--)
 990                         close (i);
 991
 992 #ifdef DEBUG_ENABLED
 993                 DEBUG ("%s: exec()ing [%s] in dir [%s]", __func__, cmd,
 994                            dir == NULL?".":dir);
 995                 for (i = 0; argv[i] != NULL; i++)
 996                         g_message ("arg %d: [%s]", i, argv[i]);
 997
 998                 for (i = 0; env_strings[i] != NULL; i++)
 999                         g_message ("env %d: [%s]", i, env_strings[i]);
1000 #endif
1001
1002                 /* set cwd */
1003                 if (dir != NULL && chdir (dir) == -1) {
1004                         /* set error */
1005                         _exit (-1);
1006                 }
1007
1008                 /* exec */
1009                 execve (argv[0], argv, env_strings);
1010
1011                 /* set error */
1012                 _exit (-1);
1013         }
1014         /* parent */
1015
1016         process_handle_data = lookup_process_handle (handle);
1017         if (!process_handle_data) {
1018                 g_warning ("%s: error looking up process handle %p", __func__,
1019                            handle);
1020                 _wapi_handle_unref (handle);
1021                 goto cleanup;
1022         }
1023
1024         process_handle_data->id = pid;
1025
1026         /* Add our mono_process into the linked list of mono_processes */
1027         mono_process = (struct MonoProcess *) g_malloc0 (sizeof (struct MonoProcess));
1028         mono_process->pid = pid;
1029         mono_process->handle_count = 1;
1030         if (MONO_SEM_INIT (&mono_process->exit_sem, 0) != 0) {
1031                 /* If we can't create the exit semaphore, we just don't add anything
1032                  * to our list of mono processes. Waiting on the process will return
1033                  * immediately. */
1034                 g_warning ("%s: could not create exit semaphore for process.", strerror (errno));
1035                 g_free (mono_process);
1036         } else {
1037                 /* Keep the process handle artificially alive until the process
1038                  * exits so that the information in the handle isn't lost. */
1039                 _wapi_handle_ref (handle);
1040                 mono_process->handle = handle;
1041
1042                 process_handle_data->mono_process = mono_process;
1043
1044                 mono_mutex_lock (&mono_processes_mutex);
1045                 mono_process->next = mono_processes;
1046                 mono_processes = mono_process;
1047                 mono_mutex_unlock (&mono_processes_mutex);
1048         }
1049
1050         if (process_info != NULL) {
1051                 process_info->hProcess = handle;
1052                 process_info->dwProcessId = pid;
1053
1054                 /* FIXME: we might need to handle the thread info some
1055                  * day
1056                  */
1057                 process_info->hThread = INVALID_HANDLE_VALUE;
1058                 process_info->dwThreadId = 0;
1059         }
1060
1061 cleanup:
1062         _wapi_handle_unlock_shared_handles ();
1063
1064         if (fork_failed)
1065                 _wapi_handle_unref (handle);
1066
1067         if (startup_pipe [1] != -1) {
1068                 /* Write 1 byte, doesn't matter what */
1069                 ssize_t _i G_GNUC_UNUSED = write (startup_pipe [1], startup_pipe, 1);
1070                 close (startup_pipe [0]);
1071                 close (startup_pipe [1]);
1072         }
1073
1074 free_strings:
1075         if (cmd)
1076                 g_free (cmd);
1077         if (full_prog)
1078                 g_free (full_prog);
1079         if (prog)
1080                 g_free (prog);
1081         if (args)
1082                 g_free (args);
1083         if (dir)
1084                 g_free (dir);
1085         if (env_strings)
1086                 g_strfreev (env_strings);
1087         if (argv)
1088                 g_strfreev (argv);
1089
1090         DEBUG ("%s: returning handle %p for pid %d", __func__, handle,
1091                    pid);
1092
1093         /* Check if something needs to be cleaned up. */
1094         mono_processes_cleanup ();
1095
1096         return ret;
1097 }
1098
1099 static void
1100 process_set_name (WapiHandle_process *process_handle)
1101 {
1102         char *progname, *utf8_progname, *slash;
1103
1104         progname = g_get_prgname ();
1105         utf8_progname = mono_utf8_from_external (progname);
1106
1107         DEBUG ("%s: using [%s] as prog name", __func__, progname);
1108
1109         if (utf8_progname) {
1110                 slash = strrchr (utf8_progname, '/');
1111                 if (slash)
1112                         process_handle->proc_name = g_strdup (slash+1);
1113                 else
1114                         process_handle->proc_name = g_strdup (utf8_progname);
1115                 g_free (utf8_progname);
1116         }
1117 }
1118
1119 void
1120 wapi_processes_init (void)
1121 {
1122         pid_t pid = _wapi_getpid ();
1123         WapiHandle_process process_handle = {0};
1124
1125         _wapi_handle_register_capabilities (WAPI_HANDLE_PROCESS,
1126                                             WAPI_HANDLE_CAP_WAIT |
1127                                             WAPI_HANDLE_CAP_SPECIAL_WAIT);
1128
1129         process_handle.id = pid;
1130
1131         process_set_defaults (&process_handle);
1132         process_set_name (&process_handle);
1133
1134         current_process = _wapi_handle_new (WAPI_HANDLE_PROCESS,
1135                                             &process_handle);
1136         g_assert (current_process);
1137 }
1138
1139 gpointer
1140 _wapi_process_duplicate (void)
1141 {
1142         _wapi_handle_ref (current_process);
1143
1144         return current_process;
1145 }
1146
1147 /* Returns a pseudo handle that doesn't need to be closed afterwards */
1148 gpointer
1149 GetCurrentProcess (void)
1150 {
1151         return _WAPI_PROCESS_CURRENT;
1152 }
1153
1154 guint32
1155 GetProcessId (gpointer handle)
1156 {
1157         WapiHandle_process *process_handle;
1158
1159         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (handle))
1160                 /* This is a pseudo handle */
1161                 return WAPI_HANDLE_TO_PID (handle);
1162
1163         process_handle = lookup_process_handle (handle);
1164         if (!process_handle) {
1165                 SetLastError (ERROR_INVALID_HANDLE);
1166                 return 0;
1167         }
1168
1169         return process_handle->id;
1170 }
1171
1172 static gboolean
1173 process_open_compare (gpointer handle, gpointer user_data)
1174 {
1175         pid_t wanted_pid;
1176         WapiHandle_process *process_handle;
1177         pid_t checking_pid;
1178
1179         g_assert (!WAPI_IS_PSEUDO_PROCESS_HANDLE (handle));
1180
1181         process_handle = lookup_process_handle (handle);
1182         g_assert (process_handle);
1183
1184         DEBUG ("%s: looking at process %d", __func__, process_handle->id);
1185
1186         checking_pid = process_handle->id;
1187
1188         if (checking_pid == 0)
1189                 return FALSE;
1190
1191         wanted_pid = GPOINTER_TO_UINT (user_data);
1192
1193         /* It's possible to have more than one process handle with the
1194          * same pid, but only the one running process can be
1195          * unsignalled
1196          */
1197         if (checking_pid == wanted_pid &&
1198             !_wapi_handle_issignalled (handle)) {
1199                 /* If the handle is blown away in the window between
1200                  * returning TRUE here and _wapi_search_handle pinging
1201                  * the timestamp, the search will continue
1202                  */
1203                 return TRUE;
1204         } else {
1205                 return FALSE;
1206         }
1207 }
1208
1209 gboolean
1210 CloseProcess (gpointer handle)
1211 {
1212         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (handle))
1213                 return TRUE;
1214         return CloseHandle (handle);
1215 }
1216
1217 /*
1218  * The caller owns the returned handle and must call CloseProcess () on it to clean it up.
1219  */
1220 gpointer
1221 OpenProcess (guint32 req_access G_GNUC_UNUSED, gboolean inherit G_GNUC_UNUSED, guint32 pid)
1222 {
1223         /* Find the process handle that corresponds to pid */
1224         gpointer handle = NULL;
1225
1226         DEBUG ("%s: looking for process %d", __func__, pid);
1227
1228         handle = _wapi_search_handle (WAPI_HANDLE_PROCESS,
1229                                       process_open_compare,
1230                                       GUINT_TO_POINTER (pid), NULL, TRUE);
1231         if (handle == 0) {
1232                 if (is_pid_valid (pid)) {
1233                         /* Return a pseudo handle for processes we
1234                          * don't have handles for
1235                          */
1236                         return WAPI_PID_TO_HANDLE (pid);
1237                 } else {
1238                         DEBUG ("%s: Can't find pid %d", __func__, pid);
1239
1240                         SetLastError (ERROR_PROC_NOT_FOUND);
1241
1242                         return NULL;
1243                 }
1244         }
1245
1246         /* _wapi_search_handle () already added a ref */
1247         return handle;
1248 }
1249
1250 gboolean
1251 GetExitCodeProcess (gpointer process, guint32 *code)
1252 {
1253         WapiHandle_process *process_handle;
1254         guint32 pid = -1;
1255
1256         if (!code)
1257                 return FALSE;
1258
1259         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (process)) {
1260                 pid = WAPI_HANDLE_TO_PID (process);
1261                 /* This is a pseudo handle, so we don't know what the
1262                  * exit code was, but we can check whether it's alive or not
1263                  */
1264                 if (is_pid_valid (pid)) {
1265                         *code = STILL_ACTIVE;
1266                         return TRUE;
1267                 } else {
1268                         return FALSE;
1269                 }
1270         }
1271
1272         process_handle = lookup_process_handle (process);
1273         if (!process_handle) {
1274                 DEBUG ("%s: Can't find process %p", __func__, process);
1275
1276                 return FALSE;
1277         }
1278
1279         if (process_handle->id == _wapi_getpid ()) {
1280                 *code = STILL_ACTIVE;
1281                 return TRUE;
1282         }
1283
1284         /* A process handle is only signalled if the process has exited
1285          * and has been waited for */
1286
1287         /* Make sure any process exit has been noticed, before
1288          * checking if the process is signalled.  Fixes bug 325463.
1289          */
1290         process_wait (process, 0, TRUE);
1291
1292         if (_wapi_handle_issignalled (process))
1293                 *code = process_handle->exitstatus;
1294         else
1295                 *code = STILL_ACTIVE;
1296
1297         return TRUE;
1298 }
1299
1300 gboolean
1301 GetProcessTimes (gpointer process, WapiFileTime *create_time,
1302                                  WapiFileTime *exit_time, WapiFileTime *kernel_time,
1303                                  WapiFileTime *user_time)
1304 {
1305         WapiHandle_process *process_handle;
1306         gboolean ku_times_set = FALSE;
1307
1308         if (create_time == NULL || exit_time == NULL || kernel_time == NULL ||
1309                 user_time == NULL)
1310                 /* Not sure if w32 allows NULLs here or not */
1311                 return FALSE;
1312
1313         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (process)) {
1314                 gpointer pid = GINT_TO_POINTER (WAPI_HANDLE_TO_PID(process));
1315                 gint64 start_ticks, user_ticks, kernel_ticks;
1316
1317                 mono_process_get_times (pid, &start_ticks, &user_ticks, &kernel_ticks);
1318
1319                 _wapi_guint64_to_filetime (start_ticks, create_time);
1320                 _wapi_guint64_to_filetime (user_ticks, kernel_time);
1321                 _wapi_guint64_to_filetime (kernel_ticks, user_time);
1322
1323                 return TRUE;
1324         }
1325
1326         process_handle = lookup_process_handle (process);
1327         if (!process_handle) {
1328                 DEBUG ("%s: Can't find process %p", __func__, process);
1329
1330                 return FALSE;
1331         }
1332
1333         *create_time = process_handle->create_time;
1334
1335         /* A process handle is only signalled if the process has
1336          * exited.  Otherwise exit_time isn't set
1337          */
1338         if (_wapi_handle_issignalled (process))
1339                 *exit_time = process_handle->exit_time;
1340
1341 #ifdef HAVE_GETRUSAGE
1342         if (process_handle->id == getpid ()) {
1343                 struct rusage time_data;
1344                 if (getrusage (RUSAGE_SELF, &time_data) == 0) {
1345                         guint64 tick_val;
1346                         ku_times_set = TRUE;
1347                         tick_val = (guint64)time_data.ru_utime.tv_sec * 10000000 + (guint64)time_data.ru_utime.tv_usec * 10;
1348                         _wapi_guint64_to_filetime (tick_val, user_time);
1349                         tick_val = (guint64)time_data.ru_stime.tv_sec * 10000000 + (guint64)time_data.ru_stime.tv_usec * 10;
1350                         _wapi_guint64_to_filetime (tick_val, kernel_time);
1351                 }
1352         }
1353 #endif
1354         if (!ku_times_set) {
1355                 memset (kernel_time, 0, sizeof (WapiFileTime));
1356                 memset (user_time, 0, sizeof (WapiFileTime));
1357         }
1358
1359         return TRUE;
1360 }
1361
1362 typedef struct
1363 {
1364         gpointer address_start;
1365         gpointer address_end;
1366         char *perms;
1367         gpointer address_offset;
1368         guint64 device;
1369         guint64 inode;
1370         char *filename;
1371 } WapiProcModule;
1372
1373 static void free_procmodule (WapiProcModule *mod)
1374 {
1375         if (mod->perms != NULL) {
1376                 g_free (mod->perms);
1377         }
1378         if (mod->filename != NULL) {
1379                 g_free (mod->filename);
1380         }
1381         g_free (mod);
1382 }
1383
1384 static gint find_procmodule (gconstpointer a, gconstpointer b)
1385 {
1386         WapiProcModule *want = (WapiProcModule *)a;
1387         WapiProcModule *compare = (WapiProcModule *)b;
1388
1389         if ((want->device == compare->device) &&
1390             (want->inode == compare->inode)) {
1391                 return(0);
1392         } else {
1393                 return(1);
1394         }
1395 }
1396
1397 #ifdef PLATFORM_MACOSX
1398 #include <mach-o/dyld.h>
1399 #include <mach-o/getsect.h>
1400
1401 static GSList *load_modules (void)
1402 {
1403         GSList *ret = NULL;
1404         WapiProcModule *mod;
1405         uint32_t count = _dyld_image_count ();
1406         int i = 0;
1407
1408         for (i = 0; i < count; i++) {
1409 #if SIZEOF_VOID_P == 8
1410                 const struct mach_header_64 *hdr;
1411                 const struct section_64 *sec;
1412 #else
1413                 const struct mach_header *hdr;
1414                 const struct section *sec;
1415 #endif
1416                 const char *name;
1417
1418                 name = _dyld_get_image_name (i);
1419 #if SIZEOF_VOID_P == 8
1420                 hdr = (const struct mach_header_64*)_dyld_get_image_header (i);
1421                 sec = getsectbynamefromheader_64 (hdr, SEG_DATA, SECT_DATA);
1422 #else
1423                 hdr = _dyld_get_image_header (i);
1424                 sec = getsectbynamefromheader (hdr, SEG_DATA, SECT_DATA);
1425 #endif
1426
1427                 /* Some dynlibs do not have data sections on osx (#533893) */
1428                 if (sec == 0) {
1429                         continue;
1430                 }
1431
1432                 mod = g_new0 (WapiProcModule, 1);
1433                 mod->address_start = GINT_TO_POINTER (sec->addr);
1434                 mod->address_end = GINT_TO_POINTER (sec->addr+sec->size);
1435                 mod->perms = g_strdup ("r--p");
1436                 mod->address_offset = 0;
1437                 mod->device = makedev (0, 0);
1438                 mod->inode = i;
1439                 mod->filename = g_strdup (name);
1440
1441                 if (g_slist_find_custom (ret, mod, find_procmodule) == NULL) {
1442                         ret = g_slist_prepend (ret, mod);
1443                 } else {
1444                         free_procmodule (mod);
1445                 }
1446         }
1447
1448         ret = g_slist_reverse (ret);
1449
1450         return(ret);
1451 }
1452 #elif defined(__OpenBSD__)
1453 #include <link.h>
1454 static int load_modules_callback (struct dl_phdr_info *info, size_t size, void *ptr)
1455 {
1456         if (size < offsetof (struct dl_phdr_info, dlpi_phnum)
1457             + sizeof (info->dlpi_phnum))
1458                 return (-1);
1459
1460         struct dl_phdr_info *cpy = calloc(1, sizeof(struct dl_phdr_info));
1461         if (!cpy)
1462                 return (-1);
1463
1464         memcpy(cpy, info, sizeof(*info));
1465
1466         g_ptr_array_add ((GPtrArray *)ptr, cpy);
1467
1468         return (0);
1469 }
1470
1471 static GSList *load_modules (void)
1472 {
1473         GSList *ret = NULL;
1474         WapiProcModule *mod;
1475         GPtrArray *dlarray = g_ptr_array_new();
1476         int i;
1477
1478         if (dl_iterate_phdr(load_modules_callback, dlarray) < 0)
1479                 return (ret);
1480
1481         for (i = 0; i < dlarray->len; i++) {
1482                 struct dl_phdr_info *info = g_ptr_array_index (dlarray, i);
1483
1484                 mod = g_new0 (WapiProcModule, 1);
1485                 mod->address_start = (gpointer)(info->dlpi_addr + info->dlpi_phdr[0].p_vaddr);
1486                 mod->address_end = (gpointer)(info->dlpi_addr +
1487                                        info->dlpi_phdr[info->dlpi_phnum - 1].p_vaddr);
1488                 mod->perms = g_strdup ("r--p");
1489                 mod->address_offset = 0;
1490                 mod->inode = i;
1491                 mod->filename = g_strdup (info->dlpi_name);
1492
1493                 DEBUG ("%s: inode=%d, filename=%s, address_start=%p, address_end=%p", __func__,
1494                                    mod->inode, mod->filename, mod->address_start, mod->address_end);
1495
1496                 free(info);
1497
1498                 if (g_slist_find_custom (ret, mod, find_procmodule) == NULL) {
1499                         ret = g_slist_prepend (ret, mod);
1500                 } else {
1501                         free_procmodule (mod);
1502                 }
1503         }
1504
1505         g_ptr_array_free (dlarray, TRUE);
1506
1507         ret = g_slist_reverse (ret);
1508
1509         return(ret);
1510 }
1511 #elif defined(__HAIKU__)
1512
1513 static GSList *load_modules (void)
1514 {
1515         GSList *ret = NULL;
1516         WapiProcModule *mod;
1517         int32 cookie = 0;
1518         image_info imageInfo;
1519
1520         while (get_next_image_info (B_CURRENT_TEAM, &cookie, &imageInfo) == B_OK) {
1521                 mod = g_new0 (WapiProcModule, 1);
1522                 mod->device = imageInfo.device;
1523                 mod->inode = imageInfo.node;
1524                 mod->filename = g_strdup (imageInfo.name);
1525                 mod->address_start = MIN (imageInfo.text, imageInfo.data);
1526                 mod->address_end = MAX ((uint8_t*)imageInfo.text + imageInfo.text_size,
1527                         (uint8_t*)imageInfo.data + imageInfo.data_size);
1528                 mod->perms = g_strdup ("r--p");
1529                 mod->address_offset = 0;
1530
1531                 if (g_slist_find_custom (ret, mod, find_procmodule) == NULL) {
1532                         ret = g_slist_prepend (ret, mod);
1533                 } else {
1534                         free_procmodule (mod);
1535                 }
1536         }
1537
1538         ret = g_slist_reverse (ret);
1539
1540         return ret;
1541 }
1542 #else
1543 static GSList *load_modules (FILE *fp)
1544 {
1545         GSList *ret = NULL;
1546         WapiProcModule *mod;
1547         char buf[MAXPATHLEN + 1], *p, *endp;
1548         char *start_start, *end_start, *prot_start, *offset_start;
1549         char *maj_dev_start, *min_dev_start, *inode_start, prot_buf[5];
1550         gpointer address_start, address_end, address_offset;
1551         guint32 maj_dev, min_dev;
1552         guint64 inode;
1553         guint64 device;
1554
1555         while (fgets (buf, sizeof(buf), fp)) {
1556                 p = buf;
1557                 while (g_ascii_isspace (*p)) ++p;
1558                 start_start = p;
1559                 if (!g_ascii_isxdigit (*start_start)) {
1560                         continue;
1561                 }
1562                 address_start = (gpointer)strtoul (start_start, &endp, 16);
1563                 p = endp;
1564                 if (*p != '-') {
1565                         continue;
1566                 }
1567
1568                 ++p;
1569                 end_start = p;
1570                 if (!g_ascii_isxdigit (*end_start)) {
1571                         continue;
1572                 }
1573                 address_end = (gpointer)strtoul (end_start, &endp, 16);
1574                 p = endp;
1575                 if (!g_ascii_isspace (*p)) {
1576                         continue;
1577                 }
1578
1579                 while (g_ascii_isspace (*p)) ++p;
1580                 prot_start = p;
1581                 if (*prot_start != 'r' && *prot_start != '-') {
1582                         continue;
1583                 }
1584                 memcpy (prot_buf, prot_start, 4);
1585                 prot_buf[4] = '\0';
1586                 while (!g_ascii_isspace (*p)) ++p;
1587
1588                 while (g_ascii_isspace (*p)) ++p;
1589                 offset_start = p;
1590                 if (!g_ascii_isxdigit (*offset_start)) {
1591                         continue;
1592                 }
1593                 address_offset = (gpointer)strtoul (offset_start, &endp, 16);
1594                 p = endp;
1595                 if (!g_ascii_isspace (*p)) {
1596                         continue;
1597                 }
1598
1599                 while(g_ascii_isspace (*p)) ++p;
1600                 maj_dev_start = p;
1601                 if (!g_ascii_isxdigit (*maj_dev_start)) {
1602                         continue;
1603                 }
1604                 maj_dev = strtoul (maj_dev_start, &endp, 16);
1605                 p = endp;
1606                 if (*p != ':') {
1607                         continue;
1608                 }
1609
1610                 ++p;
1611                 min_dev_start = p;
1612                 if (!g_ascii_isxdigit (*min_dev_start)) {
1613                         continue;
1614                 }
1615                 min_dev = strtoul (min_dev_start, &endp, 16);
1616                 p = endp;
1617                 if (!g_ascii_isspace (*p)) {
1618                         continue;
1619                 }
1620
1621                 while (g_ascii_isspace (*p)) ++p;
1622                 inode_start = p;
1623                 if (!g_ascii_isxdigit (*inode_start)) {
1624                         continue;
1625                 }
1626                 inode = (guint64)strtol (inode_start, &endp, 10);
1627                 p = endp;
1628                 if (!g_ascii_isspace (*p)) {
1629                         continue;
1630                 }
1631
1632                 device = makedev ((int)maj_dev, (int)min_dev);
1633                 if ((device == 0) &&
1634                     (inode == 0)) {
1635                         continue;
1636                 }
1637
1638                 while(g_ascii_isspace (*p)) ++p;
1639                 /* p now points to the filename */
1640
1641                 mod = g_new0 (WapiProcModule, 1);
1642                 mod->address_start = address_start;
1643                 mod->address_end = address_end;
1644                 mod->perms = g_strdup (prot_buf);
1645                 mod->address_offset = address_offset;
1646                 mod->device = device;
1647                 mod->inode = inode;
1648                 mod->filename = g_strdup (g_strstrip (p));
1649
1650                 if (g_slist_find_custom (ret, mod, find_procmodule) == NULL) {
1651                         ret = g_slist_prepend (ret, mod);
1652                 } else {
1653                         free_procmodule (mod);
1654                 }
1655         }
1656
1657         ret = g_slist_reverse (ret);
1658
1659         return(ret);
1660 }
1661 #endif
1662
1663 static gboolean match_procname_to_modulename (char *procname, char *modulename)
1664 {
1665         char* lastsep = NULL;
1666         char* lastsep2 = NULL;
1667         char* pname = NULL;
1668         char* mname = NULL;
1669         gboolean result = FALSE;
1670
1671         if (procname == NULL || modulename == NULL)
1672                 return (FALSE);
1673
1674         pname = mono_path_resolve_symlinks (procname);
1675         mname = mono_path_resolve_symlinks (modulename);
1676
1677         if (!strcmp (pname, mname))
1678                 result = TRUE;
1679
1680         if (!result) {
1681                 lastsep = strrchr (mname, '/');
1682                 if (lastsep)
1683                         if (!strcmp (lastsep+1, pname))
1684                                 result = TRUE;
1685                 if (!result) {
1686                         lastsep2 = strrchr (pname, '/');
1687                         if (lastsep2){
1688                                 if (lastsep) {
1689                                         if (!strcmp (lastsep+1, lastsep2+1))
1690                                                 result = TRUE;
1691                                 } else {
1692                                         if (!strcmp (mname, lastsep2+1))
1693                                                 result = TRUE;
1694                                 }
1695                         }
1696                 }
1697         }
1698
1699         g_free (pname);
1700         g_free (mname);
1701
1702         return result;
1703 }
1704
1705 #if !(defined(PLATFORM_MACOSX) || defined(__OpenBSD__) || defined(__HAIKU__))
1706 static FILE *
1707 open_process_map (int pid, const char *mode)
1708 {
1709         FILE *fp = NULL;
1710         const char *proc_path[] = {
1711                 "/proc/%d/maps",        /* GNU/Linux */
1712                 "/proc/%d/map",         /* FreeBSD */
1713                 NULL
1714         };
1715         int i;
1716         char *filename;
1717
1718         for (i = 0; fp == NULL && proc_path [i]; i++) {
1719                 filename = g_strdup_printf (proc_path[i], pid);
1720                 fp = fopen (filename, mode);
1721                 g_free (filename);
1722         }
1723
1724         return fp;
1725 }
1726 #endif
1727
1728 gboolean EnumProcessModules (gpointer process, gpointer *modules,
1729                              guint32 size, guint32 *needed)
1730 {
1731         WapiHandle_process *process_handle;
1732 #if !defined(__OpenBSD__) && !defined(PLATFORM_MACOSX)
1733         FILE *fp;
1734 #endif
1735         GSList *mods = NULL;
1736         WapiProcModule *module;
1737         guint32 count, avail = size / sizeof(gpointer);
1738         int i;
1739         pid_t pid;
1740         char *proc_name = NULL;
1741
1742         /* Store modules in an array of pointers (main module as
1743          * modules[0]), using the load address for each module as a
1744          * token.  (Use 'NULL' as an alternative for the main module
1745          * so that the simple implementation can just return one item
1746          * for now.)  Get the info from /proc/<pid>/maps on linux,
1747          * /proc/<pid>/map on FreeBSD, other systems will have to
1748          * implement /dev/kmem reading or whatever other horrid
1749          * technique is needed.
1750          */
1751         if (size < sizeof(gpointer))
1752                 return FALSE;
1753
1754         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (process)) {
1755                 pid = WAPI_HANDLE_TO_PID (process);
1756         } else {
1757                 process_handle = lookup_process_handle (process);
1758                 if (!process_handle) {
1759                         DEBUG ("%s: Can't find process %p", __func__, process);
1760
1761                         return FALSE;
1762                 }
1763                 pid = process_handle->id;
1764                 proc_name = process_handle->proc_name;
1765         }
1766
1767 #if defined(PLATFORM_MACOSX) || defined(__OpenBSD__) || defined(__HAIKU__)
1768         mods = load_modules ();
1769         if (!proc_name) {
1770                 modules[0] = NULL;
1771                 *needed = sizeof(gpointer);
1772                 return TRUE;
1773         }
1774 #else
1775         fp = open_process_map (pid, "r");
1776         if (!fp) {
1777                 /* No /proc/<pid>/maps so just return the main module
1778                  * shortcut for now
1779                  */
1780                 modules[0] = NULL;
1781                 *needed = sizeof(gpointer);
1782                 return TRUE;
1783         }
1784         mods = load_modules (fp);
1785         fclose (fp);
1786 #endif
1787         count = g_slist_length (mods);
1788
1789         /* count + 1 to leave slot 0 for the main module */
1790         *needed = sizeof(gpointer) * (count + 1);
1791
1792         /*
1793          * Use the NULL shortcut, as the first line in
1794          * /proc/<pid>/maps isn't the executable, and we need
1795          * that first in the returned list. Check the module name
1796          * to see if it ends with the proc name and substitute
1797          * the first entry with it.  FIXME if this turns out to
1798          * be a problem.
1799          */
1800         modules[0] = NULL;
1801         for (i = 0; i < (avail - 1) && i < count; i++) {
1802                 module = (WapiProcModule *)g_slist_nth_data (mods, i);
1803                 if (modules[0] != NULL)
1804                         modules[i] = module->address_start;
1805                 else if (match_procname_to_modulename (proc_name, module->filename))
1806                         modules[0] = module->address_start;
1807                 else
1808                         modules[i + 1] = module->address_start;
1809         }
1810
1811         for (i = 0; i < count; i++) {
1812                 free_procmodule (g_slist_nth_data (mods, i));
1813         }
1814         g_slist_free (mods);
1815
1816         return TRUE;
1817 }
1818
1819 static char *
1820 get_process_name_from_proc (pid_t pid)
1821 {
1822 #if defined(__OpenBSD__)
1823         int mib [6];
1824         size_t size;
1825         struct kinfo_proc *pi;
1826 #elif defined(PLATFORM_MACOSX)
1827 #if !(!defined (__mono_ppc__) && defined (TARGET_OSX))
1828         size_t size;
1829         struct kinfo_proc *pi;
1830         int mib[] = { CTL_KERN, KERN_PROC, KERN_PROC_PID, pid };
1831 #endif
1832 #else
1833         FILE *fp;
1834         char *filename = NULL;
1835 #endif
1836         char buf[256];
1837         char *ret = NULL;
1838
1839 #if defined(PLATFORM_SOLARIS)
1840         filename = g_strdup_printf ("/proc/%d/psinfo", pid);
1841         if ((fp = fopen (filename, "r")) != NULL) {
1842                 struct psinfo info;
1843                 int nread;
1844
1845                 nread = fread (&info, sizeof (info), 1, fp);
1846                 if (nread == 1) {
1847                         ret = g_strdup (info.pr_fname);
1848                 }
1849
1850                 fclose (fp);
1851         }
1852         g_free (filename);
1853 #elif defined(PLATFORM_MACOSX)
1854 #if !defined (__mono_ppc__) && defined (TARGET_OSX)
1855         /* No proc name on OSX < 10.5 nor ppc nor iOS */
1856         memset (buf, '\0', sizeof(buf));
1857         proc_name (pid, buf, sizeof(buf));
1858         if (strlen (buf) > 0)
1859                 ret = g_strdup (buf);
1860 #else
1861         if (sysctl(mib, 4, NULL, &size, NULL, 0) < 0)
1862                 return(ret);
1863
1864         if ((pi = malloc(size)) == NULL)
1865                 return(ret);
1866
1867         if (sysctl (mib, 4, pi, &size, NULL, 0) < 0) {
1868                 if (errno == ENOMEM) {
1869                         free(pi);
1870                         DEBUG ("%s: Didn't allocate enough memory for kproc info", __func__);
1871                 }
1872                 return(ret);
1873         }
1874
1875         if (strlen (pi->kp_proc.p_comm) > 0)
1876                 ret = g_strdup (pi->kp_proc.p_comm);
1877
1878         free(pi);
1879 #endif
1880 #elif defined(__OpenBSD__)
1881         mib [0] = CTL_KERN;
1882         mib [1] = KERN_PROC;
1883         mib [2] = KERN_PROC_PID;
1884         mib [3] = pid;
1885         mib [4] = sizeof(struct kinfo_proc);
1886         mib [5] = 0;
1887
1888 retry:
1889         if (sysctl(mib, 6, NULL, &size, NULL, 0) < 0)
1890                 return(ret);
1891
1892         if ((pi = malloc(size)) == NULL)
1893                 return(ret);
1894
1895         mib[5] = (int)(size / sizeof(struct kinfo_proc));
1896
1897         if ((sysctl (mib, 6, pi, &size, NULL, 0) < 0) ||
1898                 (size != sizeof (struct kinfo_proc))) {
1899                 if (errno == ENOMEM) {
1900                         free(pi);
1901                         goto retry;
1902                 }
1903                 return(ret);
1904         }
1905
1906         if (strlen (pi->p_comm) > 0)
1907                 ret = g_strdup (pi->p_comm);
1908
1909         free(pi);
1910 #elif defined(__HAIKU__)
1911         image_info imageInfo;
1912         int32 cookie = 0;
1913
1914         if (get_next_image_info ((team_id)pid, &cookie, &imageInfo) == B_OK) {
1915                 ret = g_strdup (imageInfo.name);
1916         }
1917 #else
1918         memset (buf, '\0', sizeof(buf));
1919         filename = g_strdup_printf ("/proc/%d/exe", pid);
1920         if (readlink (filename, buf, 255) > 0) {
1921                 ret = g_strdup (buf);
1922         }
1923         g_free (filename);
1924
1925         if (ret != NULL) {
1926                 return(ret);
1927         }
1928
1929         filename = g_strdup_printf ("/proc/%d/cmdline", pid);
1930         if ((fp = fopen (filename, "r")) != NULL) {
1931                 if (fgets (buf, 256, fp) != NULL) {
1932                         ret = g_strdup (buf);
1933                 }
1934
1935                 fclose (fp);
1936         }
1937         g_free (filename);
1938
1939         if (ret != NULL) {
1940                 return(ret);
1941         }
1942
1943         filename = g_strdup_printf ("/proc/%d/stat", pid);
1944         if ((fp = fopen (filename, "r")) != NULL) {
1945                 if (fgets (buf, 256, fp) != NULL) {
1946                         char *start, *end;
1947
1948                         start = strchr (buf, '(');
1949                         if (start != NULL) {
1950                                 end = strchr (start + 1, ')');
1951
1952                                 if (end != NULL) {
1953                                         ret = g_strndup (start + 1,
1954                                                          end - start - 1);
1955                                 }
1956                         }
1957                 }
1958
1959                 fclose (fp);
1960         }
1961         g_free (filename);
1962 #endif
1963
1964         return ret;
1965 }
1966
1967 /*
1968  * wapi_process_get_path:
1969  *
1970  *   Return the full path of the executable of the process PID, or NULL if it cannot be determined.
1971  * Returns malloc-ed memory.
1972  */
1973 char*
1974 wapi_process_get_path (pid_t pid)
1975 {
1976 #if defined(PLATFORM_MACOSX) && !defined(__mono_ppc__) && defined(TARGET_OSX)
1977         char buf [PROC_PIDPATHINFO_MAXSIZE];
1978         int res;
1979
1980         res = proc_pidpath (pid, buf, sizeof (buf));
1981         if (res <= 0)
1982                 return NULL;
1983         if (buf [0] == '\0')
1984                 return NULL;
1985         return g_strdup (buf);
1986 #else
1987         return get_process_name_from_proc (pid);
1988 #endif
1989 }
1990
1991 /*
1992  * wapi_process_set_cli_launcher:
1993  *
1994  *   Set the full path of the runtime executable used to launch managed exe's.
1995  */
1996 void
1997 wapi_process_set_cli_launcher (char *path)
1998 {
1999         g_free (cli_launcher);
2000         cli_launcher = path ? g_strdup (path) : NULL;
2001 }
2002
2003 static guint32
2004 get_module_name (gpointer process, gpointer module,
2005                                  gunichar2 *basename, guint32 size,
2006                                  gboolean base)
2007 {
2008         WapiHandle_process *process_handle;
2009         pid_t pid;
2010         gunichar2 *procname;
2011         char *procname_ext = NULL;
2012         glong len;
2013         gsize bytes;
2014 #if !defined(__OpenBSD__) && !defined(PLATFORM_MACOSX)
2015         FILE *fp;
2016 #endif
2017         GSList *mods = NULL;
2018         WapiProcModule *found_module;
2019         guint32 count;
2020         int i;
2021         char *proc_name = NULL;
2022
2023         DEBUG ("%s: Getting module base name, process handle %p module %p",
2024                    __func__, process, module);
2025
2026         size = size * sizeof (gunichar2); /* adjust for unicode characters */
2027
2028         if (basename == NULL || size == 0)
2029                 return 0;
2030
2031         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (process)) {
2032                 /* This is a pseudo handle */
2033                 pid = (pid_t)WAPI_HANDLE_TO_PID (process);
2034                 proc_name = get_process_name_from_proc (pid);
2035         } else {
2036                 process_handle = lookup_process_handle (process);
2037                 if (!process_handle) {
2038                         DEBUG ("%s: Can't find process %p", __func__,
2039                                    process);
2040
2041                         return 0;
2042                 }
2043                 pid = process_handle->id;
2044                 proc_name = g_strdup (process_handle->proc_name);
2045         }
2046
2047         /* Look up the address in /proc/<pid>/maps */
2048 #if defined(PLATFORM_MACOSX) || defined(__OpenBSD__) || defined(__HAIKU__)
2049         mods = load_modules ();
2050 #else
2051         fp = open_process_map (pid, "r");
2052         if (fp == NULL) {
2053                 if (errno == EACCES && module == NULL && base == TRUE) {
2054                         procname_ext = get_process_name_from_proc (pid);
2055                 } else {
2056                         /* No /proc/<pid>/maps, so just return failure
2057                          * for now
2058                          */
2059                         g_free (proc_name);
2060                         return 0;
2061                 }
2062         } else {
2063                 mods = load_modules (fp);
2064                 fclose (fp);
2065         }
2066 #endif
2067         count = g_slist_length (mods);
2068
2069         /* If module != NULL compare the address.
2070          * If module == NULL we are looking for the main module.
2071          * The best we can do for now check it the module name end with the process name.
2072          */
2073         for (i = 0; i < count; i++) {
2074                 found_module = (WapiProcModule *)g_slist_nth_data (mods, i);
2075                 if (procname_ext == NULL &&
2076                         ((module == NULL && match_procname_to_modulename (proc_name, found_module->filename)) ||
2077                          (module != NULL && found_module->address_start == module))) {
2078                         if (base)
2079                                 procname_ext = g_path_get_basename (found_module->filename);
2080                         else
2081                                 procname_ext = g_strdup (found_module->filename);
2082                 }
2083
2084                 free_procmodule (found_module);
2085         }
2086
2087         if (procname_ext == NULL) {
2088                 /* If it's *still* null, we might have hit the
2089                  * case where reading /proc/$pid/maps gives an
2090                  * empty file for this user.
2091                  */
2092                 procname_ext = get_process_name_from_proc (pid);
2093         }
2094
2095         g_slist_free (mods);
2096         g_free (proc_name);
2097
2098         if (procname_ext) {
2099                 DEBUG ("%s: Process name is [%s]", __func__,
2100                            procname_ext);
2101
2102                 procname = mono_unicode_from_external (procname_ext, &bytes);
2103                 if (procname == NULL) {
2104                         /* bugger */
2105                         g_free (procname_ext);
2106                         return 0;
2107                 }
2108
2109                 len = (bytes / 2);
2110
2111                 /* Add the terminator */
2112                 bytes += 2;
2113
2114                 if (size < bytes) {
2115                         DEBUG ("%s: Size %d smaller than needed (%ld); truncating", __func__, size, bytes);
2116
2117                         memcpy (basename, procname, size);
2118                 } else {
2119                         DEBUG ("%s: Size %d larger than needed (%ld)",
2120                                    __func__, size, bytes);
2121
2122                         memcpy (basename, procname, bytes);
2123                 }
2124
2125                 g_free (procname);
2126                 g_free (procname_ext);
2127
2128                 return len;
2129         }
2130
2131         return 0;
2132 }
2133
2134 guint32
2135 GetModuleBaseName (gpointer process, gpointer module,
2136                                    gunichar2 *basename, guint32 size)
2137 {
2138         return get_module_name (process, module, basename, size, TRUE);
2139 }
2140
2141 guint32
2142 GetModuleFileNameEx (gpointer process, gpointer module,
2143                                          gunichar2 *filename, guint32 size)
2144 {
2145         return get_module_name (process, module, filename, size, FALSE);
2146 }
2147
2148 gboolean
2149 GetModuleInformation (gpointer process, gpointer module,
2150                                           WapiModuleInfo *modinfo, guint32 size)
2151 {
2152         WapiHandle_process *process_handle;
2153         pid_t pid;
2154 #if !defined(__OpenBSD__) && !defined(PLATFORM_MACOSX)
2155         FILE *fp;
2156 #endif
2157         GSList *mods = NULL;
2158         WapiProcModule *found_module;
2159         guint32 count;
2160         int i;
2161         gboolean ret = FALSE;
2162         char *proc_name = NULL;
2163
2164         DEBUG ("%s: Getting module info, process handle %p module %p",
2165                    __func__, process, module);
2166
2167         if (modinfo == NULL || size < sizeof (WapiModuleInfo))
2168                 return FALSE;
2169
2170         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (process)) {
2171                 pid = (pid_t)WAPI_HANDLE_TO_PID (process);
2172                 proc_name = get_process_name_from_proc (pid);
2173         } else {
2174                 process_handle = lookup_process_handle (process);
2175                 if (!process_handle) {
2176                         DEBUG ("%s: Can't find process %p", __func__,
2177                                    process);
2178
2179                         return FALSE;
2180                 }
2181                 pid = process_handle->id;
2182                 proc_name = g_strdup (process_handle->proc_name);
2183         }
2184
2185 #if defined(PLATFORM_MACOSX) || defined(__OpenBSD__) || defined(__HAIKU__)
2186         mods = load_modules ();
2187 #else
2188         /* Look up the address in /proc/<pid>/maps */
2189         if ((fp = open_process_map (pid, "r")) == NULL) {
2190                 /* No /proc/<pid>/maps, so just return failure
2191                  * for now
2192                  */
2193                 g_free (proc_name);
2194                 return FALSE;
2195         }
2196         mods = load_modules (fp);
2197         fclose (fp);
2198 #endif
2199         count = g_slist_length (mods);
2200
2201         /* If module != NULL compare the address.
2202          * If module == NULL we are looking for the main module.
2203          * The best we can do for now check it the module name end with the process name.
2204          */
2205         for (i = 0; i < count; i++) {
2206                         found_module = (WapiProcModule *)g_slist_nth_data (mods, i);
2207                         if (ret == FALSE &&
2208                                 ((module == NULL && match_procname_to_modulename (proc_name, found_module->filename)) ||
2209                                  (module != NULL && found_module->address_start == module))) {
2210                                 modinfo->lpBaseOfDll = found_module->address_start;
2211                                 modinfo->SizeOfImage = (gsize)(found_module->address_end) - (gsize)(found_module->address_start);
2212                                 modinfo->EntryPoint = found_module->address_offset;
2213                                 ret = TRUE;
2214                         }
2215
2216                         free_procmodule (found_module);
2217         }
2218
2219         g_slist_free (mods);
2220         g_free (proc_name);
2221
2222         return ret;
2223 }
2224
2225 gboolean
2226 GetProcessWorkingSetSize (gpointer process, size_t *min, size_t *max)
2227 {
2228         WapiHandle_process *process_handle;
2229
2230         if (min == NULL || max == NULL)
2231                 /* Not sure if w32 allows NULLs here or not */
2232                 return FALSE;
2233
2234         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (process))
2235                 /* This is a pseudo handle, so just fail for now */
2236                 return FALSE;
2237
2238         process_handle = lookup_process_handle (process);
2239         if (!process_handle) {
2240                 DEBUG ("%s: Can't find process %p", __func__, process);
2241
2242                 return FALSE;
2243         }
2244
2245         *min = process_handle->min_working_set;
2246         *max = process_handle->max_working_set;
2247
2248         return TRUE;
2249 }
2250
2251 gboolean
2252 SetProcessWorkingSetSize (gpointer process, size_t min, size_t max)
2253 {
2254         WapiHandle_process *process_handle;
2255
2256         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (process))
2257                 /* This is a pseudo handle, so just fail for now
2258                  */
2259                 return FALSE;
2260
2261         process_handle = lookup_process_handle (process);
2262         if (!process_handle) {
2263                 DEBUG ("%s: Can't find process %p", __func__, process);
2264
2265                 return FALSE;
2266         }
2267
2268         process_handle->min_working_set = min;
2269         process_handle->max_working_set = max;
2270
2271         return TRUE;
2272 }
2273
2274
2275 gboolean
2276 TerminateProcess (gpointer process, gint32 exitCode)
2277 {
2278         WapiHandle_process *process_handle;
2279         int signo;
2280         int ret;
2281         pid_t pid;
2282
2283         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (process)) {
2284                 /* This is a pseudo handle */
2285                 pid = (pid_t)WAPI_HANDLE_TO_PID (process);
2286         } else {
2287                 process_handle = lookup_process_handle (process);
2288                 if (!process_handle) {
2289                         DEBUG ("%s: Can't find process %p", __func__, process);
2290                         SetLastError (ERROR_INVALID_HANDLE);
2291                         return FALSE;
2292                 }
2293                 pid = process_handle->id;
2294         }
2295
2296         signo = (exitCode == -1) ? SIGKILL : SIGTERM;
2297         ret = kill (pid, signo);
2298         if (ret == -1) {
2299                 switch (errno) {
2300                 case EINVAL:
2301                         SetLastError (ERROR_INVALID_PARAMETER);
2302                         break;
2303                 case EPERM:
2304                         SetLastError (ERROR_ACCESS_DENIED);
2305                         break;
2306                 case ESRCH:
2307                         SetLastError (ERROR_PROC_NOT_FOUND);
2308                         break;
2309                 default:
2310                         SetLastError (ERROR_GEN_FAILURE);
2311                 }
2312         }
2313
2314         return (ret == 0);
2315 }
2316
2317 guint32
2318 GetPriorityClass (gpointer process)
2319 {
2320 #ifdef HAVE_GETPRIORITY
2321         WapiHandle_process *process_handle;
2322         int ret;
2323         pid_t pid;
2324
2325         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (process)) {
2326                 /* This is a pseudo handle */
2327                 pid = (pid_t)WAPI_HANDLE_TO_PID (process);
2328         } else {
2329                 process_handle = lookup_process_handle (process);
2330                 if (!process_handle) {
2331                         SetLastError (ERROR_INVALID_HANDLE);
2332                         return FALSE;
2333                 }
2334                 pid = process_handle->id;
2335         }
2336
2337         errno = 0;
2338         ret = getpriority (PRIO_PROCESS, pid);
2339         if (ret == -1 && errno != 0) {
2340                 switch (errno) {
2341                 case EPERM:
2342                 case EACCES:
2343                         SetLastError (ERROR_ACCESS_DENIED);
2344                         break;
2345                 case ESRCH:
2346                         SetLastError (ERROR_PROC_NOT_FOUND);
2347                         break;
2348                 default:
2349                         SetLastError (ERROR_GEN_FAILURE);
2350                 }
2351                 return FALSE;
2352         }
2353
2354         if (ret == 0)
2355                 return NORMAL_PRIORITY_CLASS;
2356         else if (ret < -15)
2357                 return REALTIME_PRIORITY_CLASS;
2358         else if (ret < -10)
2359                 return HIGH_PRIORITY_CLASS;
2360         else if (ret < 0)
2361                 return ABOVE_NORMAL_PRIORITY_CLASS;
2362         else if (ret > 10)
2363                 return IDLE_PRIORITY_CLASS;
2364         else if (ret > 0)
2365                 return BELOW_NORMAL_PRIORITY_CLASS;
2366
2367         return NORMAL_PRIORITY_CLASS;
2368 #else
2369         SetLastError (ERROR_NOT_SUPPORTED);
2370         return 0;
2371 #endif
2372 }
2373
2374 gboolean
2375 SetPriorityClass (gpointer process, guint32  priority_class)
2376 {
2377 #ifdef HAVE_SETPRIORITY
2378         WapiHandle_process *process_handle;
2379         int ret;
2380         int prio;
2381         pid_t pid;
2382
2383         if (WAPI_IS_PSEUDO_PROCESS_HANDLE (process)) {
2384                 /* This is a pseudo handle */
2385                 pid = (pid_t)WAPI_HANDLE_TO_PID (process);
2386         } else {
2387                 process_handle = lookup_process_handle (process);
2388                 if (!process_handle) {
2389                         SetLastError (ERROR_INVALID_HANDLE);
2390                         return FALSE;
2391                 }
2392                 pid = process_handle->id;
2393         }
2394
2395         switch (priority_class) {
2396         case IDLE_PRIORITY_CLASS:
2397                 prio = 19;
2398                 break;
2399         case BELOW_NORMAL_PRIORITY_CLASS:
2400                 prio = 10;
2401                 break;
2402         case NORMAL_PRIORITY_CLASS:
2403                 prio = 0;
2404                 break;
2405         case ABOVE_NORMAL_PRIORITY_CLASS:
2406                 prio = -5;
2407                 break;
2408         case HIGH_PRIORITY_CLASS:
2409                 prio = -11;
2410                 break;
2411         case REALTIME_PRIORITY_CLASS:
2412                 prio = -20;
2413                 break;
2414         default:
2415                 SetLastError (ERROR_INVALID_PARAMETER);
2416                 return FALSE;
2417         }
2418
2419         ret = setpriority (PRIO_PROCESS, pid, prio);
2420         if (ret == -1) {
2421                 switch (errno) {
2422                 case EPERM:
2423                 case EACCES:
2424                         SetLastError (ERROR_ACCESS_DENIED);
2425                         break;
2426                 case ESRCH:
2427                         SetLastError (ERROR_PROC_NOT_FOUND);
2428                         break;
2429                 default:
2430                         SetLastError (ERROR_GEN_FAILURE);
2431                 }
2432         }
2433
2434         return ret == 0;
2435 #else
2436         SetLastError (ERROR_NOT_SUPPORTED);
2437         return FALSE;
2438 #endif
2439 }
2440
2441 static void
2442 mono_processes_cleanup (void)
2443 {
2444         struct MonoProcess *mp;
2445         struct MonoProcess *prev = NULL;
2446         GSList *finished = NULL;
2447         GSList *l;
2448         gpointer unref_handle;
2449
2450         DEBUG ("%s", __func__);
2451
2452         /* Ensure we're not in here in multiple threads at once, nor recursive. */
2453         if (InterlockedCompareExchange (&mono_processes_cleaning_up, 1, 0) != 0)
2454                 return;
2455
2456         for (mp = mono_processes; mp; mp = mp->next) {
2457                 if (mp->pid == 0 && mp->handle) {
2458                         /* This process has exited and we need to remove the artifical ref
2459                          * on the handle */
2460                         mono_mutex_lock (&mono_processes_mutex);
2461                         unref_handle = mp->handle;
2462                         mp->handle = NULL;
2463                         mono_mutex_unlock (&mono_processes_mutex);
2464                         if (unref_handle)
2465                                 _wapi_handle_unref (unref_handle);
2466                 }
2467         }
2468
2469         /*
2470          * Remove processes which exited from the mono_processes list.
2471          * We need to synchronize with the sigchld handler here, which runs
2472          * asynchronously. The handler requires that the mono_processes list
2473          * remain valid.
2474          */
2475         mono_mutex_lock (&mono_processes_mutex);
2476
2477         mp = mono_processes;
2478         while (mp) {
2479                 if (mp->handle_count == 0 && mp->freeable) {
2480                         /*
2481                          * Unlink the entry.
2482                          * This code can run parallel with the sigchld handler, but the
2483                          * modifications it makes are safe.
2484                          */
2485                         if (mp == mono_processes)
2486                                 mono_processes = mp->next;
2487                         else
2488                                 prev->next = mp->next;
2489                         finished = g_slist_prepend (finished, mp);
2490
2491                         mp = mp->next;
2492                 } else {
2493                         prev = mp;
2494                         mp = mp->next;
2495                 }
2496         }
2497
2498         mono_memory_barrier ();
2499
2500         for (l = finished; l; l = l->next) {
2501                 /*
2502                  * All the entries in the finished list are unlinked from mono_processes, and
2503                  * they have the 'finished' flag set, which means the sigchld handler is done
2504                  * accessing them.
2505                  */
2506                 mp = l->data;
2507                 MONO_SEM_DESTROY (&mp->exit_sem);
2508                 g_free (mp);
2509         }
2510         g_slist_free (finished);
2511
2512         mono_mutex_unlock (&mono_processes_mutex);
2513
2514         DEBUG ("%s done", __func__);
2515
2516         InterlockedDecrement (&mono_processes_cleaning_up);
2517 }
2518
2519 static void
2520 process_close (gpointer handle, gpointer data)
2521 {
2522         WapiHandle_process *process_handle;
2523
2524         DEBUG ("%s", __func__);
2525
2526         process_handle = (WapiHandle_process *) data;
2527         g_free (process_handle->proc_name);
2528         process_handle->proc_name = NULL;
2529         if (process_handle->mono_process)
2530                 InterlockedDecrement (&process_handle->mono_process->handle_count);
2531         mono_processes_cleanup ();
2532 }
2533
2534 #if HAVE_SIGACTION
2535 MONO_SIGNAL_HANDLER_FUNC (static, mono_sigchld_signal_handler, (int _dummy, siginfo_t *info, void *context))
2536 {
2537         int status;
2538         int pid;
2539         struct MonoProcess *p;
2540
2541         DEBUG ("SIG CHILD handler for pid: %i\n", info->si_pid);
2542
2543         do {
2544                 do {
2545                         pid = waitpid (-1, &status, WNOHANG);
2546                 } while (pid == -1 && errno == EINTR);
2547
2548                 if (pid <= 0)
2549                         break;
2550
2551                 DEBUG ("child ended: %i", pid);
2552
2553                 /*
2554                  * This can run concurrently with the code in the rest of this module.
2555                  */
2556                 for (p = mono_processes; p; p = p->next) {
2557                         if (p->pid == pid) {
2558                                 break;
2559                         }
2560                 }
2561                 if (p) {
2562                         p->pid = 0; /* this pid doesn't exist anymore, clear it */
2563                         p->status = status;
2564                         MONO_SEM_POST (&p->exit_sem);
2565                         mono_memory_barrier ();
2566                         /* Mark this as freeable, the pointer becomes invalid afterwards */
2567                         p->freeable = TRUE;
2568                 }
2569         } while (1);
2570
2571         DEBUG ("SIG CHILD handler: done looping.");
2572 }
2573
2574 #endif
2575
2576 static void
2577 process_add_sigchld_handler (void)
2578 {
2579 #if HAVE_SIGACTION
2580         struct sigaction sa;
2581
2582         sa.sa_sigaction = mono_sigchld_signal_handler;
2583         sigemptyset (&sa.sa_mask);
2584         sa.sa_flags = SA_NOCLDSTOP | SA_SIGINFO;
2585         g_assert (sigaction (SIGCHLD, &sa, &previous_chld_sa) != -1);
2586         DEBUG ("Added SIGCHLD handler");
2587 #endif
2588 }
2589
2590 static guint32
2591 process_wait (gpointer handle, guint32 timeout, gboolean alertable)
2592 {
2593         WapiHandle_process *process_handle;
2594         pid_t pid G_GNUC_UNUSED, ret;
2595         int status;
2596         guint32 start;
2597         guint32 now;
2598         struct MonoProcess *mp;
2599
2600         /* FIXME: We can now easily wait on processes that aren't our own children,
2601          * but WaitFor*Object won't call us for pseudo handles. */
2602         g_assert ((GPOINTER_TO_UINT (handle) & _WAPI_PROCESS_UNHANDLED) != _WAPI_PROCESS_UNHANDLED);
2603
2604         DEBUG ("%s (%p, %u)", __func__, handle, timeout);
2605
2606         process_handle = lookup_process_handle (handle);
2607         if (!process_handle) {
2608                 g_warning ("%s: error looking up process handle %p", __func__, handle);
2609                 return WAIT_FAILED;
2610         }
2611
2612         if (process_handle->exited) {
2613                 /* We've already done this one */
2614                 DEBUG ("%s (%p, %u): Process already exited", __func__, handle, timeout);
2615                 return WAIT_OBJECT_0;
2616         }
2617
2618         pid = process_handle->id;
2619
2620         DEBUG ("%s (%p, %u): PID: %d", __func__, handle, timeout, pid);
2621
2622         /* We don't need to lock mono_processes here, the entry
2623          * has a handle_count > 0 which means it will not be freed. */
2624         mp = process_handle->mono_process;
2625         g_assert (mp);
2626
2627         start = mono_msec_ticks ();
2628         now = start;
2629
2630         while (1) {
2631                 if (timeout != INFINITE) {
2632                         DEBUG ("%s (%p, %u): waiting on semaphore for %li ms...",
2633                                    __func__, handle, timeout, (timeout - (now - start)));
2634
2635                         ret = MONO_SEM_TIMEDWAIT_ALERTABLE (&mp->exit_sem, (timeout - (now - start)), alertable);
2636                 } else {
2637                         DEBUG ("%s (%p, %u): waiting on semaphore forever...",
2638                                    __func__, handle, timeout);
2639                         ret = MONO_SEM_WAIT_ALERTABLE (&mp->exit_sem, alertable);
2640                 }
2641
2642                 if (ret == -1 && errno != EINTR && errno != ETIMEDOUT) {
2643                         DEBUG ("%s (%p, %u): sem_timedwait failure: %s",
2644                                    __func__, handle, timeout, g_strerror (errno));
2645                         /* Should we return a failure here? */
2646                 }
2647
2648                 if (ret == 0) {
2649                         /* Success, process has exited */
2650                         MONO_SEM_POST (&mp->exit_sem);
2651                         break;
2652                 }
2653
2654                 if (timeout == 0) {
2655                         DEBUG ("%s (%p, %u): WAIT_TIMEOUT (timeout = 0)", __func__, handle, timeout);
2656                         return WAIT_TIMEOUT;
2657                 }
2658
2659                 now = mono_msec_ticks ();
2660                 if (now - start >= timeout) {
2661                         DEBUG ("%s (%p, %u): WAIT_TIMEOUT", __func__, handle, timeout);
2662                         return WAIT_TIMEOUT;
2663                 }
2664
2665                 if (alertable && _wapi_thread_cur_apc_pending ()) {
2666                         DEBUG ("%s (%p, %u): WAIT_IO_COMPLETION", __func__, handle, timeout);
2667                         return WAIT_IO_COMPLETION;
2668                 }
2669         }
2670
2671         /* Process must have exited */
2672         DEBUG ("%s (%p, %u): Waited successfully", __func__, handle, timeout);
2673
2674         ret = _wapi_handle_lock_shared_handles ();
2675         g_assert (ret == 0);
2676
2677         status = mp ? mp->status : 0;
2678         if (WIFSIGNALED (status))
2679                 process_handle->exitstatus = 128 + WTERMSIG (status);
2680         else
2681                 process_handle->exitstatus = WEXITSTATUS (status);
2682         _wapi_time_t_to_filetime (time (NULL), &process_handle->exit_time);
2683
2684         process_handle->exited = TRUE;
2685
2686         DEBUG ("%s (%p, %u): Setting pid %d signalled, exit status %d",
2687                    __func__, handle, timeout, process_handle->id, process_handle->exitstatus);
2688
2689         _wapi_handle_set_signal_state (handle, TRUE, TRUE);
2690
2691         _wapi_handle_unlock_shared_handles ();
2692
2693         return WAIT_OBJECT_0;
2694 }
2695
2696 void
2697 wapi_processes_cleanup (void)
2698 {
2699         g_free (cli_launcher);
2700 }