Changeset 7115

Timestamp:

Nov 15, 2008, 7:59:18 PM (12 years ago)

Author:

charles

Message:

(libT) get file preallocation working on Windows by copying how fsutil does it. (source: ​http://social.msdn.microsoft.com/forums/en-US/vclanguage/thread/4dabec8e-2909-40b3-b398-66bd4c587b9b/)

File:

• trunk/libtransmission/fdlimit.c (modified) (3 diffs)

trunk/libtransmission/fdlimit.c

@@ -96 +96 @@
 ***/
 
+#ifndef O_LARGEFILE
+#define O_LARGEFILE 0
+#endif
+
 static int
-preallocateFile( int fd UNUSED, uint64_t length UNUSED )
-{
-#ifdef HAVE_FALLOCATE
-
-    return fallocate( fd, FALLOC_FL_KEEP_SIZE, 0, length );
-
-#elif defined(HAVE_POSIX_FALLOCATE)
-
-    return posix_fallocate( fd, 0, length );
-
-#elif defined(SYS_DARWIN) 
-
-    fstore_t fst;
-    fst.fst_flags = F_ALLOCATECONTIG;
-    fst.fst_posmode = F_PEOFPOSMODE;
-    fst.fst_offset = 0;
-    fst.fst_length = length;
-    fst.fst_bytesalloc = 0;
-    return fcntl( fd, F_PREALLOCATE, &fst );
+preallocateFile( const char * filename, uint64_t length )
+{
+    int success = 0;
+
+#ifdef WIN32
+
+    HANDLE hFile = CreateFile( filename, GENERIC_WRITE, 0, 0, CREATE_NEW, 0, 0 );
+    if( hFile != INVALID_HANDLE_VALUE )
+    {
+        LARGE_INTEGER li;
+        li.QuadPart = desiredFileSize;
+        success = SetFilePointerEx( hFile, desiredFileSize, NULL, FILE_BEGIN )
+               && SetEndOfFile( hFile );
+        CloseHandle( hFile );
+    }
 
 #else
 
-    #warning no known method to preallocate files on this platform
-    return -1;
-
-#endif
+    int flags = O_RDWR | O_CREAT | O_LARGEFILE;
+    int fd = open( filename, flags, 0666 );
+    if( fd >= 0 )
+    {
+        
+# ifdef HAVE_FALLOCATE
+
+        success = !fallocate( fd, FALLOC_FL_KEEP_SIZE, 0, length );
+
+# elif defined(HAVE_POSIX_FALLOCATE)
+
+        success = !posix_fallocate( fd, 0, length );
+
+# elif defined(SYS_DARWIN) 
+
+        fstore_t fst;
+        fst.fst_flags = F_ALLOCATECONTIG;
+        fst.fst_posmode = F_PEOFPOSMODE;
+        fst.fst_offset = 0;
+        fst.fst_length = length;
+        fst.fst_bytesalloc = 0;
+        success = !fcntl( fd, F_PREALLOCATE, &fst );
+
+# else
+
+        #warning no known method to preallocate files on this platform
+        success = 0;
+
+# endif
+
+        close( fd );
+    }
+
+#endif
+
+    return success;
 }
 
@@ -162 +194 @@
 
     alreadyExisted = !stat( filename, &sb ) && S_ISREG( sb.st_mode );
+
+    if( doWrite && !alreadyExisted && doPreallocate )
+        if( preallocateFile( filename, desiredFileSize ) )
+            tr_inf( _( "Preallocated file \"%s\"" ), filename );
     
     /* open the file */
@@ -181 +217 @@
     }
 
-    if( ( file->fd >= 0 ) && !alreadyExisted && doPreallocate )
-        if( !preallocateFile( file->fd, desiredFileSize ) )
-            tr_inf( _( "Preallocated file \"%s\"" ), filename );
-       
     tr_free( filename );
     return 0;