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                         // WindowsImpersonationContext implements IDisposable only in 2.0
  83                         using (WindowsImpersonationContext context = WindowsIdentity.Impersonate (userToken)) {
  84                                 return InternalExecWaitWithCapture (cmd, currentDir, tempFiles, ref outputName, ref errorName);
  85                         }
  86                 }
  87
  88                 public static Int32 ExecWaitWithCapture (IntPtr userToken, string cmd, TempFileCollection tempFiles, ref string outputName, ref string errorName)
  89                 {
  90                         return ExecWaitWithCapture (userToken, cmd, Environment.CurrentDirectory, tempFiles, ref outputName, ref errorName);
  91                 }
  92
  93                 [SecurityPermission (SecurityAction.Demand, UnmanagedCode = true)]
  94                 public static Int32 ExecWaitWithCapture (string cmd, string currentDir, TempFileCollection tempFiles, ref string outputName, ref string errorName )
  95                 {
  96                         return InternalExecWaitWithCapture (cmd, currentDir, tempFiles, ref outputName, ref errorName);
  97                 }
  98
  99                 [SecurityPermission (SecurityAction.Demand, UnmanagedCode = true)]
 100                 public static Int32 ExecWaitWithCapture (string cmd, TempFileCollection tempFiles, ref string outputName, ref string errorName)
 101                 {
 102                         return InternalExecWaitWithCapture (cmd, Environment.CurrentDirectory, tempFiles, ref outputName, ref errorName);
 103                 }
 104
 105                 private static int InternalExecWaitWithCapture (string cmd, string currentDir, TempFileCollection tempFiles, ref string outputName, ref string errorName)
 106                 {
 107                         if ((cmd == null) || (cmd.Length == 0))
 108                                 throw new ExternalException (Locale.GetText ("No command provided for execution."));
 109
 110                         if (outputName == null)
 111                                 outputName = tempFiles.AddExtension ("out");
 112
 113                         if (errorName == null)
 114                                 errorName = tempFiles.AddExtension ("err");
 115
 116                         int exit_code = -1;
 117                         Process proc = new Process ();
 118                         proc.StartInfo.FileName = cmd;
 119                         proc.StartInfo.CreateNoWindow = true;
 120                         proc.StartInfo.UseShellExecute = false;
 121                         proc.StartInfo.RedirectStandardOutput = true;
 122                         proc.StartInfo.RedirectStandardError = true;
 123                         proc.StartInfo.WorkingDirectory = currentDir;
 124
 125                         try {
 126                                 proc.Start();
 127
 128                                 ProcessResultReader outReader = new ProcessResultReader (proc.StandardOutput, outputName);
 129                                 ProcessResultReader errReader = new ProcessResultReader (proc.StandardError, errorName);
 130
 131                                 Thread t = new Thread (new ThreadStart (errReader.Read));
 132                                 t.Start ();
 133
 134                                 outReader.Read ();
 135                                 t.Join ();
 136
 137                                 proc.WaitForExit();
 138                         }
 139                         finally  {
 140                                 exit_code = proc.ExitCode;
 141                                 // the handle is cleared in Close (so no ExitCode)
 142                                 proc.Close ();
 143                         }
 144                         return exit_code;
 145                 }
 146         }
 147 }