Merge pull request #216 from ilkerde/master

[mono.git] / mcs / class / System / System.CodeDom.Compiler / TempFileCollection.cs

diff --git a/mcs/class/System/System.CodeDom.Compiler/TempFileCollection.cs b/mcs/class/System/System.CodeDom.Compiler/TempFileCollection.cs
index 947e6faa656d9c22e1585bb8d2aa132a864d0a89..da056d15336030f5a974515f73f1cbf02cb7ed4c 100644 (file)
--- a/mcs/class/System/System.CodeDom.Compiler/TempFileCollection.cs
+++ b/mcs/class/System/System.CodeDom.Compiler/TempFileCollection.cs
@@ -31,12 +31,11 @@ using System.Collections;
 using System.IO;
 using System.Security;
 using System.Security.Permissions;
+using System.Runtime.InteropServices;
 
 namespace System.CodeDom.Compiler {
 
-#if NET_2_0
        [Serializable]
-#endif
        [PermissionSet (SecurityAction.LinkDemand, Unrestricted = true)]
        public class TempFileCollection:ICollection, IEnumerable, IDisposable
        {
@@ -45,6 +44,7 @@ namespace System.CodeDom.Compiler {
                bool keepfiles;
                string basepath;
                Random rnd;
+               string ownTempDir;
                
                public TempFileCollection ()
                        : this (String.Empty, false)
@@ -67,16 +67,15 @@ namespace System.CodeDom.Compiler {
                {
                        get {
                                if(basepath==null) {
-                                       // note: this property *cannot* change TempDir property
-                                       string temp = tempdir;
-                                       if (temp.Length == 0) {
-                                               // this call ensure the Environment permissions check
-                                               temp = Path.GetTempPath ();
-                                       }
-
+                               
                                        if (rnd == null)
                                                rnd = new Random ();
 
+                                       // note: this property *cannot* change TempDir property
+                                       string temp = tempdir;
+                                       if (temp.Length == 0)
+                                               temp = GetOwnTempDir ();
+
                                        // Create a temporary file at the target directory. This ensures
                                        // that the generated file name is unique.
                                        FileStream f = null;
@@ -111,6 +110,46 @@ namespace System.CodeDom.Compiler {
                                return(basepath);
                        }
                }
+               
+               string GetOwnTempDir ()
+               {
+                       if (ownTempDir != null)
+                               return ownTempDir;
+
+                       // this call ensure the Environment permissions check
+                       string basedir = Path.GetTempPath ();
+                       
+                       // Create a subdirectory with the correct user permissions
+                       int res = -1;
+                       bool win32 = false;
+                       switch (Environment.OSVersion.Platform) {
+                       case PlatformID.Win32S:
+                       case PlatformID.Win32Windows:
+                       case PlatformID.Win32NT:
+                       case PlatformID.WinCE:
+                               win32 = true;
+                               res = 0;
+                               break;
+                       }
+
+                       do {
+                               int num = rnd.Next ();
+                               num++;
+                               ownTempDir = Path.Combine (basedir, num.ToString("x"));
+                               if (Directory.Exists (ownTempDir))
+                                       continue;
+                               if (win32)
+                                       Directory.CreateDirectory (ownTempDir);
+                               else
+                                       res = mkdir (ownTempDir, 0x1c0);
+                               if (res != 0) {
+                                       if (!Directory.Exists (ownTempDir))
+                                               throw new IOException ();
+                                       // Somebody already created the dir, keep trying
+                               }
+                       } while (res != 0);
+                       return ownTempDir;
+               }
 
                int ICollection.Count {
                        get {
@@ -190,18 +229,25 @@ namespace System.CodeDom.Compiler {
                
                public void Delete()
                {
-                       string[] filenames=new string[filehash.Count];
-                       filehash.Keys.CopyTo(filenames, 0);
+                       bool allDeleted = true;
+                       string[] filenames = new string[filehash.Count];
+                       filehash.Keys.CopyTo (filenames, 0);
 
                        foreach(string file in filenames) {
                                if((bool)filehash[file]==false) {
                                        File.Delete(file);
                                        filehash.Remove(file);
-                               }
+                               } else
+                                       allDeleted = false;
                        }
                        if (basepath != null) {
                                string tmpFile = basepath + ".tmp";
                                File.Delete (tmpFile);
+                               basepath = null;
+                       }
+                       if (allDeleted && ownTempDir != null) {
+                               Directory.Delete (ownTempDir, true);
+                               ownTempDir = null;
                        }
                }
 
@@ -228,5 +274,6 @@ namespace System.CodeDom.Compiler {
                        Dispose(false);
                }
                
+               [DllImport ("libc")] private static extern int mkdir (string olpath, uint mode);
        }
 }