mcs/class/System/System.CodeDom.Compiler/Executor.cs

   1 //
   2 // System.CodeDom.Compiler.Executor.cs
   3 //
   4 // Authors:
   5 //      Andreas Nahr (ClassDevelopment@A-SoftTech.com)
   6 //      Sebastien Pouliot  <sebastien@ximian.com>
   7 //
   8 // (C) 2003 Andreas Nahr
   9 // Copyright (C) 2005 Novell, Inc (http://www.novell.com)
  10 //
  11 // Permission is hereby granted, free of charge, to any person obtaining
  12 // a copy of this software and associated documentation files (the
  13 // "Software"), to deal in the Software without restriction, including
  14 // without limitation the rights to use, copy, modify, merge, publish,
  15 // distribute, sublicense, and/or sell copies of the Software, and to
  16 // permit persons to whom the Software is furnished to do so, subject to
  17 // the following conditions:
  18 //
  19 // The above copyright notice and this permission notice shall be
  20 // included in all copies or substantial portions of the Software.
  21 //
  22 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
  23 // EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  24 // MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  25 // NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
  26 // LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
  27 // OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
  28 // WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  29 //
  30
  31 using System.Diagnostics;
  32 using System.IO;
  33 using System.Runtime.InteropServices;
  34 using System.Security.Permissions;
  35 using System.Security.Principal;
  36 using System.Threading;
  37
  38 namespace System.CodeDom.Compiler {
  39
  40         [PermissionSet (SecurityAction.LinkDemand, Unrestricted = true)]
  41         public static class Executor {
  42
  43                 class ProcessResultReader
  44                 {
  45                         StreamReader reader;
  46                         string file;
  47
  48                         public ProcessResultReader (StreamReader reader, string file)
  49                         {
  50                                 this.reader = reader;
  51                                 this.file = file;
  52                         }
  53
  54                         public void Read ()
  55                         {
  56                                 StreamWriter sw = new StreamWriter (file);
  57
  58                                 try
  59                                 {
  60                                         string line;
  61                                         while ((line = reader.ReadLine()) != null)
  62                                                 sw.WriteLine (line);
  63                                 }
  64                                 finally
  65                                 {
  66                                         sw.Close ();
  67                                 }
  68                         }
  69                 }
  70
  71                 public static void ExecWait (string cmd, TempFileCollection tempFiles)
  72                 {
  73                         string outputName = null;
  74                         string errorName = null;
  75                         ExecWaitWithCapture (cmd, Environment.CurrentDirectory, tempFiles, ref outputName, ref errorName);
  76                 }
  77
  78                 [SecurityPermission (SecurityAction.Demand, UnmanagedCode = true)]
  79                 [SecurityPermission (SecurityAction.Assert, ControlPrincipal = true)] // UnmanagedCode "covers" more than ControlPrincipal
  80                 public static Int32 ExecWaitWithCapture (IntPtr userToken, string cmd, string currentDir, TempFileCollection tempFiles, ref string outputName, ref string errorName)
  81                 {
  82 #if NET_2_0
  83                         // WindowsImpersonationContext implements IDisposable only in 2.0
  84                         using (WindowsImpersonationContext context = WindowsIdentity.Impersonate (userToken)) {
  85                                 return InternalExecWaitWithCapture (cmd, currentDir, tempFiles, ref outputName, ref errorName);
  86                         }
  87 #else
  88                         int result = -1;
  89                         WindowsImpersonationContext context = WindowsIdentity.Impersonate (userToken);
  90                         try {
  91                                 result = InternalExecWaitWithCapture (cmd, currentDir, tempFiles, ref outputName, ref errorName);
  92                         }
  93                         finally {
  94                                 context.Undo ();
  95                                 context = null;
  96                         }
  97                         return result;
  98 #endif
  99                 }
 100
 101                 public static Int32 ExecWaitWithCapture (IntPtr userToken, string cmd, TempFileCollection tempFiles, ref string outputName, ref string errorName)
 102                 {
 103                         return ExecWaitWithCapture (userToken, cmd, Environment.CurrentDirectory, tempFiles, ref outputName, ref errorName);
 104                 }
 105
 106                 [SecurityPermission (SecurityAction.Demand, UnmanagedCode = true)]
 107                 public static Int32 ExecWaitWithCapture (string cmd, string currentDir, TempFileCollection tempFiles, ref string outputName, ref string errorName )
 108                 {
 109                         return InternalExecWaitWithCapture (cmd, currentDir, tempFiles, ref outputName, ref errorName);
 110                 }
 111
 112                 [SecurityPermission (SecurityAction.Demand, UnmanagedCode = true)]
 113                 public static Int32 ExecWaitWithCapture (string cmd, TempFileCollection tempFiles, ref string outputName, ref string errorName)
 114                 {
 115                         return InternalExecWaitWithCapture (cmd, Environment.CurrentDirectory, tempFiles, ref outputName, ref errorName);
 116                 }
 117
 118                 private static int InternalExecWaitWithCapture (string cmd, string currentDir, TempFileCollection tempFiles, ref string outputName, ref string errorName)
 119                 {
 120                         if ((cmd == null) || (cmd.Length == 0))
 121                                 throw new ExternalException (Locale.GetText ("No command provided for execution."));
 122
 123                         if (outputName == null)
 124                                 outputName = tempFiles.AddExtension ("out");
 125
 126                         if (errorName == null)
 127                                 errorName = tempFiles.AddExtension ("err");
 128
 129                         int exit_code = -1;
 130                         Process proc = new Process ();
 131                         proc.StartInfo.FileName = cmd;
 132                         proc.StartInfo.CreateNoWindow = true;
 133                         proc.StartInfo.UseShellExecute = false;
 134                         proc.StartInfo.RedirectStandardOutput = true;
 135                         proc.StartInfo.RedirectStandardError = true;
 136                         proc.StartInfo.WorkingDirectory = currentDir;
 137
 138                         try {
 139                                 proc.Start();
 140
 141                                 ProcessResultReader outReader = new ProcessResultReader (proc.StandardOutput, outputName);
 142                                 ProcessResultReader errReader = new ProcessResultReader (proc.StandardError, errorName);
 143
 144                                 Thread t = new Thread (new ThreadStart (errReader.Read));
 145                                 t.Start ();
 146
 147                                 outReader.Read ();
 148                                 t.Join ();
 149
 150                                 proc.WaitForExit();
 151                         }
 152                         finally  {
 153                                 exit_code = proc.ExitCode;
 154                                 // the handle is cleared in Close (so no ExitCode)
 155                                 proc.Close ();
 156                         }
 157                         return exit_code;
 158                 }
 159         }
 160 }