mcs/class/Microsoft.Build.Utilities/Microsoft.Build.Utilities/ProcessService.cs

   1 #if NET_2_0
   2
   3 using System;
   4 using System.IO;
   5 using System.Collections;
   6 using System.Collections.Specialized;
   7 using System.Diagnostics;
   8
   9 using SCS = System.Collections.Specialized;
  10
  11 namespace Microsoft.Build.Utilities
  12 {
  13         internal static class ProcessService {
  14                 static SCS.ProcessStringDictionary globalEnvironmentVariablesOverride;
  15
  16                 public static StringDictionary GlobalEnvironmentVariblesOverride {
  17                         get {
  18                                 if (globalEnvironmentVariablesOverride == null)
  19                                         globalEnvironmentVariablesOverride = new SCS.ProcessStringDictionary ();
  20                                 return globalEnvironmentVariablesOverride;
  21                         }
  22                 }
  23
  24                 public static ProcessWrapper StartProcess (string command, string arguments, string workingDirectory, EventHandler exited)
  25                 {
  26                         return StartProcess (command, arguments, workingDirectory, (ProcessEventHandler)null, (ProcessEventHandler)null, exited);
  27                 }
  28
  29                 public static ProcessWrapper StartProcess (string command, string arguments, string workingDirectory, ProcessEventHandler outputStreamChanged, ProcessEventHandler errorStreamChanged)
  30                 {
  31                         return StartProcess (command, arguments, workingDirectory, outputStreamChanged, errorStreamChanged, null);
  32                 }
  33
  34                 public static ProcessWrapper StartProcess (string command, string arguments, string workingDirectory, TextWriter outWriter, TextWriter errorWriter, EventHandler exited)
  35                 {
  36                         return StartProcess (command, arguments, workingDirectory, outWriter, errorWriter, exited, false);
  37                 }
  38
  39                 public static ProcessWrapper StartProcess (string command, string arguments, string workingDirectory, TextWriter outWriter, TextWriter errorWriter, EventHandler exited, bool redirectStandardInput)
  40                 {
  41                         ProcessEventHandler wout = OutWriter.GetWriteHandler (outWriter);
  42                         ProcessEventHandler werr = OutWriter.GetWriteHandler (errorWriter);
  43                         return StartProcess (command, arguments, workingDirectory, wout, werr, exited, redirectStandardInput);
  44                 }
  45
  46                 public static ProcessWrapper StartProcess (string command, string arguments, string workingDirectory, ProcessEventHandler outputStreamChanged, ProcessEventHandler errorStreamChanged, EventHandler exited)
  47                 {
  48                         return StartProcess (command, arguments, workingDirectory, outputStreamChanged, errorStreamChanged, exited, false);
  49                 }
  50
  51                 public static ProcessWrapper StartProcess (string command, string arguments, string workingDirectory, ProcessEventHandler outputStreamChanged, ProcessEventHandler errorStreamChanged, EventHandler exited, bool redirectStandardInput)
  52                 {
  53                         return StartProcess (CreateProcessStartInfo (command, arguments, workingDirectory, redirectStandardInput),
  54                                 outputStreamChanged, errorStreamChanged, exited, null);
  55                 }
  56
  57                 public static ProcessWrapper StartProcess (ProcessStartInfo startInfo, TextWriter outWriter, TextWriter errorWriter, EventHandler exited)
  58                 {
  59                         return StartProcess (startInfo, outWriter, errorWriter, exited, null);
  60                 }
  61
  62                 public static ProcessWrapper StartProcess (ProcessStartInfo startInfo, TextWriter outWriter, TextWriter errorWriter, EventHandler exited, StringDictionary environmentOverride)
  63                 {
  64                         ProcessEventHandler wout = OutWriter.GetWriteHandler (outWriter);
  65                         ProcessEventHandler werr = OutWriter.GetWriteHandler (errorWriter);
  66                         return StartProcess (startInfo, wout, werr, exited, environmentOverride);
  67                 }
  68
  69                 // @environmentOverride overrides even the global override values
  70                 public static ProcessWrapper StartProcess (ProcessStartInfo startInfo, ProcessEventHandler outputStreamChanged, ProcessEventHandler errorStreamChanged, EventHandler exited, StringDictionary environmentOverride)
  71                 {
  72                         if (startInfo == null)
  73                                 throw new ArgumentException ("startInfo");
  74
  75                         ProcessWrapper p = new ProcessWrapper();
  76
  77                         if (outputStreamChanged != null) {
  78                                 p.OutputStreamChanged += outputStreamChanged;
  79                         }
  80
  81                         if (errorStreamChanged != null)
  82                                 p.ErrorStreamChanged += errorStreamChanged;
  83
  84                         if (exited != null)
  85                                 p.Exited += exited;
  86
  87                         p.StartInfo = startInfo;
  88                         ProcessEnvironmentVariableOverrides (p.StartInfo, environmentOverride);
  89
  90                         // WORKAROUND for "Bug 410743 - wapi leak in System.Diagnostic.Process"
  91                         // Process leaks when an exit event is registered
  92                         // instead we use another thread to monitor I/O and wait for exit
  93                         // p.EnableRaisingEvents = true;
  94
  95                         p.Start ();
  96                         return p;
  97                 }
  98
  99                 public static ProcessStartInfo CreateProcessStartInfo (string command, string arguments, string workingDirectory, bool redirectStandardInput)
 100                 {
 101                         if (command == null)
 102                                 throw new ArgumentNullException("command");
 103
 104                         if (command.Length == 0)
 105                                 throw new ArgumentException("command");
 106
 107                         ProcessStartInfo startInfo = null;
 108                         if(String.IsNullOrEmpty (arguments))
 109                                 startInfo = new ProcessStartInfo (command);
 110                         else
 111                                 startInfo = new ProcessStartInfo (command, arguments);
 112
 113                         if(workingDirectory != null && workingDirectory.Length > 0)
 114                                 startInfo.WorkingDirectory = workingDirectory;
 115
 116                         startInfo.RedirectStandardOutput = true;
 117                         startInfo.RedirectStandardError = true;
 118                         startInfo.RedirectStandardInput = redirectStandardInput;
 119                         startInfo.UseShellExecute = false;
 120
 121                         return startInfo;
 122                 }
 123
 124                 public static void ProcessEnvironmentVariableOverrides (ProcessStartInfo info, StringDictionary environmentOverride)
 125                 {
 126                         if (globalEnvironmentVariablesOverride != null)
 127                                 foreach (DictionaryEntry entry in globalEnvironmentVariablesOverride)
 128                                         ProcessEnvironmentVariable (info, (string)entry.Key, (string)entry.Value);
 129
 130                         if (environmentOverride != null)
 131                                 foreach (DictionaryEntry entry in environmentOverride)
 132                                         ProcessEnvironmentVariable (info, (string)entry.Key, (string)entry.Value);
 133                 }
 134
 135                 static void ProcessEnvironmentVariable (ProcessStartInfo info, string name, string value)
 136                 {
 137                         if (value == null && info.EnvironmentVariables.ContainsKey (name))
 138                                 info.EnvironmentVariables.Remove (name);
 139                         else
 140                                 info.EnvironmentVariables[name] = value;
 141                 }
 142
 143         }
 144
 145         class OutWriter
 146         {
 147                 TextWriter writer;
 148
 149                 public OutWriter (TextWriter writer)
 150                 {
 151                         this.writer = writer;
 152                 }
 153
 154                 public void WriteOut (object sender, string s)
 155                 {
 156                         writer.Write (s);
 157                 }
 158
 159                 public static ProcessEventHandler GetWriteHandler (TextWriter tw)
 160                 {
 161                         return tw != null ? new ProcessEventHandler(new OutWriter (tw).WriteOut) : null;
 162                 }
 163         }
 164 }
 165
 166 #endif