- remove sendfile. it doesn't work on linux kernel 2.6 and on mac os x

- cleanup code

1 files changed, 130 insertions, 242 deletions

diff --git a/library/filemanager.cpp b/library/filemanager.cpp
index adfe590..47af1c6 100644
--- a/library/filemanager.cpp
+++ b/library/filemanager.cpp
@@ -14,34 +14,24 @@
 ** See http://www.trolltech.com/gpl/ for GPL licensing information.
 **
 ** Contact info@trolltech.com if any conditions of this licensing are
 ** not clear to you.
 **
 **********************************************************************/
+
 #include "filemanager.h"
 #include "applnk.h"
 
+/* QT */
+#include <qdir.h>
 #include <qfileinfo.h>
 #include <qtextstream.h>
 
-#include <errno.h>
+/* STD */
 #include <stdlib.h>
-#include <unistd.h>
 #include <sys/stat.h>
-#include <dirent.h>
-#ifdef Q_OS_MACX
-// MacOS X does not have sendfile.. :(
-// But maybe in the future.. !?
-#  ifdef SENDFILE
-#    include <sys/types.h>
-#    include <sys/socket.h>
-#  endif
-#else
-#  include <sys/sendfile.h>
-#endif /* Q_OS_MACX */
-#include <fcntl.h>
 
 /*!
   \class FileManager
   \brief The FileManager class assists with AppLnk input/output.
 */
 
@@ -54,76 +44,80 @@ FileManager::FileManager()
 
 /*!
   Destroys a FileManager.
 */
 FileManager::~FileManager()
 {
-
 }
 
 /*!
   Saves \a data as the document specified by \a f.
 
   Returns whether the operation succeeded.
 */
 bool FileManager::saveFile( const DocLnk &f, const QByteArray &data )
 {
- QString fn = f.file() + ".new";
-    ensurePathExists( fn );
-    QFile fl( fn );
-    if ( !fl.open( IO_WriteOnly|IO_Raw ) ) {
-       qWarning("open failed");
-       return FALSE;
-    }
-    int total_written = fl.writeBlock( data );
-    fl.close();
-    if ( total_written != int(data.size()) || !f.writeLink() ) {
-  QFile::remove( fn );
-  return FALSE;
+    QString fileName = f.file() + ".new";
+    ensurePathExists( fileName );
+    QFile file( fileName );
+
+    //write data in temporary .new file
+    if ( !file.open( IO_WriteOnly|IO_Raw ) )
+    {
+        qWarning("open failed");
+        return FALSE;
     }
-       qDebug("total written %d out of %d", total_written, data.size());
-    // else rename the file...
-    if ( !renameFile( fn.latin1(), f.file().latin1() ) ) {
-  qWarning( "problem renaming file %s to %s, errno: %d", fn.latin1(),
-      f.file().latin1(), errno );
-  // remove the file...
+    int total_written = file.writeBlock( data );
+    file.close();
+    //check if every was written
+    if ( total_written != int(data.size()) || !f.writeLink() )
+    {
+        QFile::remove( fileName );
+        return FALSE;
     }
+    qDebug("total written %d out of %d", total_written, data.size());
+
+    //rename temporary .new file in original filenam
+    if ( !renameFile( fileName,  f.file() ) )
+        QFile::remove( fileName);
     return TRUE;
 }
 
 /*!
   Saves \a text as the document specified by \a f.
 
   The text is saved in UTF8 format.
 
   Returns whether the operation succeeded.
 */
 bool FileManager::saveFile( const DocLnk &f, const QString &text )
 {
- QString fn = f.file() + ".new";
-    ensurePathExists( fn );
-    QFile fl( fn );
-    if ( !fl.open( IO_WriteOnly|IO_Raw ) ) {
-       qWarning("open failed");
-       return FALSE;
+    QString fileName = f.file() + ".new";
+    ensurePathExists( fileName );
+    QFile file( fileName );
+
+    //write data in temporary .new file
+    if ( !file.open( IO_WriteOnly|IO_Raw ) )
+    {
+        qWarning("open failed");
+        return FALSE;
     }
 
     QCString cstr = text.utf8();
     int total_written;
-    total_written = fl.writeBlock( cstr.data(), cstr.length() );
-    fl.close();
-    if ( total_written != int(cstr.length()) || !f.writeLink() ) {
-       QFile::remove( fn );
-       return FALSE;
+    total_written = file.writeBlock( cstr.data(), cstr.length() );
+    file.close();
+    if ( total_written != int(cstr.length()) || !f.writeLink() )
+    {
+        QFile::remove( fileName );
+        return FALSE;
     }
-    // okay now rename the file..
-    if ( !renameFile( fn.latin1(), f.file().latin1() ) ) {
-       qWarning( "problem renaming file %s to %s, errno: %d", fn.latin1(),
-                 f.file().latin1(), errno );
 
-    }
+    // okay now rename the file..
+    if ( !renameFile( fileName,  f.file() ) )
+        QFile::remove( fileName);
     return TRUE;
 }
 
 
 /*!
   Loads \a text from the document specified by \a f.
@@ -134,13 +128,13 @@ bool FileManager::saveFile( const DocLnk &f, const QString &text )
 */
 bool FileManager::loadFile( const DocLnk &f, QString &text )
 {
     QString fn = f.file();
     QFile fl( fn );
     if ( !fl.open( IO_ReadOnly ) )
-  return FALSE;
+        return FALSE;
     QTextStream ts( &fl );
 #if QT_VERSION <= 230 && defined(QT_NO_CODECS)
     // The below should work, but doesn't in Qt 2.3.0
     ts.setCodec( QTextCodec::codecForMib( 106 ) );
 #else
     ts.setEncoding( QTextStream::UnicodeUTF8 );
@@ -158,239 +152,130 @@ bool FileManager::loadFile( const DocLnk &f, QString &text )
 */
 bool FileManager::loadFile( const DocLnk &f, QByteArray &ba )
 {
     QString fn = f.file();
     QFile fl( fn );
     if ( !fl.open( IO_ReadOnly ) )
-  return FALSE;
+        return FALSE;
     ba.resize( fl.size() );
     if ( fl.size() > 0 )
-  fl.readBlock( ba.data(), fl.size() );
+        fl.readBlock( ba.data(), fl.size() );
     fl.close();
     return TRUE;
 }
 
 /*!
   Copies the document specified by \a src to the document specified
   by \a dest.
 
   Returns whether the operation succeeded.
 */
 bool FileManager::copyFile( const AppLnk &src, const AppLnk &dest )
 {
-    QFile sf( src.file() );
-    if ( !sf.open( IO_ReadOnly ) )
-  return FALSE;
+    QFile srcFile( src.file() );
+    if ( !srcFile.open( IO_ReadOnly ) )
+        return FALSE;
 
-    QString fn = dest.file() + ".new";
-    ensurePathExists( fn );
-    QFile df( fn );
-    if ( !df.open( IO_WriteOnly|IO_Raw ) )
-  return FALSE;
+    QString fileName = dest.file() + ".new";
+
+    ensurePathExists( fileName );
 
-    const int bufsize = 16384;
-    char buffer[bufsize];
     bool ok = TRUE;
-    int bytesRead = 0;
-    while ( ok && !sf.atEnd() ) {
-  bytesRead = sf.readBlock( buffer, bufsize );
-  if ( bytesRead < 0 )
-      ok = FALSE;
-  while ( ok && bytesRead > 0 ) {
-      int bytesWritten = df.writeBlock( buffer, bytesRead );
-      if ( bytesWritten < 0 )
-    ok = FALSE;
-      else
-    bytesRead -= bytesWritten;
-  }
-    }
+    ok = copyFile( src.file(), fileName );
 
     if ( ok )
-  ok = dest.writeLink();
-    
-    if ( ok ) {
-  // okay now rename the file...
-  if ( !renameFile( fn.latin1(), dest.file().latin1() )  ) {
-      qWarning( "problem renaming file %s to %s, errno: %d", fn.latin1(),
-          dest.file().latin1(), errno );
-      // remove the tmp file, otherwise, it will just lay around...
-      QFile::remove( fn.latin1() );
-  }
-    } else {
-  QFile::remove( fn.latin1() );
-    }
+        ok = dest.writeLink();
 
+    if ( ok )
+    {
+        // okay now rename the file...
+        if ( !renameFile( fileName.latin1(), dest.file().latin1() )  )
+            // remove the tmp file, otherwise, it will just lay around...
+            QFile::remove( fileName.latin1() );
+    }
+    else
+    {
+        QFile::remove( fileName.latin1() );
+    }
     return ok;
 }
 
-bool FileManager::copyFile( const QString & src, const QString & dest ) {
-   bool success = true;
-   struct stat status;
-   int read_fd=0;
-   int write_fd=0;
-   struct stat stat_buf;
-   off_t offset = 0;
-   QFile srcFile(src);
-   QFile destFile(dest);
-
-   if(!srcFile.open( IO_ReadOnly|IO_Raw)) {
-         return success = false;
-   }
-   read_fd = srcFile.handle();
-   if(read_fd != -1) {
-      fstat (read_fd, &stat_buf);
-      if( !destFile.open( IO_WriteOnly|IO_Raw ) )
-            return success = false;
-      write_fd = destFile.handle(); 
-      if(write_fd != -1) {
-         int err=0;
-         QString msg;
-#ifdef Q_OS_MACX
-#ifdef SENDFILE
-	 /* FreeBSD does support a different kind of 
-	  * sendfile. (eilers)
-	  * I took this from Very Secure FTPd
-	  * Licence: GPL
-	  * Author: Chris Evans
-	  * sysdeputil.c
-	  */
-          /* XXX - start_pos will truncate on 32-bit machines - can we
-           * say "start from current pos"?
-           */
-          off_t written = 0;
-	  int retval = 0;
-          retval = sendfile(read_fd, write_fd, offset, stat_buf.st_size, NULL,
-                            &written, 0);
-          /* Translate to Linux-like retval */
-          if (written > 0)
-          {
-            err = (int) written;
-          }
-#else /* SENDFILE */
-	  err == -1;
-	  msg = "FAILURE: Using unsupported function \"sendfile()\" Need Workaround !!";
-	  success = false;
-#         warning "Need workaround for sendfile!!(eilers)"
-#endif  /* SENDFILE */
-
-#else
-	  err = sendfile(write_fd, read_fd, &offset, stat_buf.st_size);
-	  if( err == -1) {
-		  switch(errno) {
-		  case EBADF : msg = "The input file was not opened for reading or the output file was not opened for writing. ";
-		  case EINVAL: msg = "Descriptor is not valid or locked. ";
-		  case ENOMEM: msg = "Insufficient memory to read from in_fd.";
-		  case EIO: msg = "Unspecified error while reading from in_fd.";
-		  };
-		  success = false;
-	  }
-#endif /* Q_OS_MACX */
-	  if( !success )
-		  qWarning( msg );
-     } else {
-         qWarning("open write failed %s, %s",src.latin1(), dest.latin1());
-         success = false;
-      }
-   } else {
-      qWarning("open read failed %s, %s",src.latin1(), dest.latin1());
-      success = false;
-   }
-   srcFile.close();
-   destFile.close();
-    // Set file permissions
-  if( stat( (const char *) src, &status ) == 0 ) {
-      chmod( (const char *) dest, status.st_mode );
+bool FileManager::copyFile( const QString & src, const QString & dest )
+{
+    //open read file
+    QFile srcFile( src );
+    if( !srcFile.open( IO_ReadOnly|IO_Raw) )
+    {
+        qWarning( "open read failed %s, %s", src.latin1(), dest.latin1() );
+        return FALSE;
     }
 
-  return success;
-}
-
-
-bool FileManager::renameFile( const QString & src, const QString & dest ) {
-   if(copyFile( src, dest )) {
-      if(QFile::remove(src) ) {
-       return true;
-      }
-   }
-    return false;  
-}
+    //open write file
+    QFile destFile( dest );
+    if( !destFile.open( IO_WriteOnly|IO_Raw ) )
+    {
+        qWarning( "open write failed %s, %s", src.latin1(), dest.latin1() );
+        srcFile.close();
+        return FALSE;
+    }
 
-/*
-bool FileManager::copyFile( const QString & src, const QString & dest ) {
-   bool success = true;
-   struct stat status;
-   int read_fd=0;
-   int write_fd=0;
-   struct stat stat_buf;
-   off_t offset = 0;
-   QFile srcFile(src);
-   QFile destFile(dest);
-
-   if(!srcFile.open( IO_ReadOnly|IO_Raw)) {
-         return success = false;
-   }
-   read_fd = srcFile.handle();
-   if(read_fd != -1) {
-      fstat (read_fd, &stat_buf);
-      if( !destFile.open( IO_WriteOnly|IO_Raw ) )
-            return success = false;
-      write_fd = destFile.handle();
-      if(write_fd != -1) {
-         int err=0;
-         QString msg;
-         err = sendfile(write_fd, read_fd, &offset, stat_buf.st_size);
-         if( err == -1) {
-            switch(err) {
-            case EBADF : msg = "The input file was not opened for reading or the output file was not opened for writing. ";
-            case EINVAL: msg = "Descriptor is not valid or locked. ";
-            case ENOMEM: msg = "Insufficient memory to read from in_fd.";
-            case EIO: msg = "Unspecified error while reading from in_fd.";
-            };
-            success = false;
-         }
-      } else {
-         qWarning("open write failed %s, %s",src.latin1(), dest.latin1());
-         success = false;
-      }
-   } else {
-      qWarning("open read failed %s, %s",src.latin1(), dest.latin1());
-      success = false;
-   }
-   srcFile.close();
-   destFile.close();
+    //copy content
+    const int bufsize = 16384;
+    char buffer[bufsize];
+    bool ok = TRUE;
+    int bytesRead = 0;
+    while ( ok && !srcFile.atEnd() )
+    {
+        bytesRead = srcFile.readBlock( buffer, bufsize );
+        if ( bytesRead < 0 )
+            ok = FALSE;
+        while ( ok && bytesRead > 0 )
+        {
+            int bytesWritten = destFile.writeBlock( buffer, bytesRead );
+            if ( bytesWritten < 0 )
+                ok = FALSE;
+            else
+                bytesRead -= bytesWritten;
+        }
+    }
+    srcFile.close();
+    destFile.close();
     // Set file permissions
-  if( stat( (const char *) src, &status ) == 0 ) {
-      chmod( (const char *) dest, status.st_mode );
+    struct stat status;
+    if( stat( (const char *) src, &status ) == 0 )
+    {
+        chmod( (const char *) dest, status.st_mode );
     }
-
-  return success;
+    return ok;
 }
 
 
-bool FileManager::renameFile( const QString & src, const QString & dest ) {
-   if(copyFile( src, dest )) {
-      if(QFile::remove(src) ) {
-       return true;
-      }
-   }
-    return false;
+bool FileManager::renameFile( const QString & src, const QString & dest )
+{
+    QDir dir( QFileInfo( src ).absFilePath() );
+    if ( !dir.rename( src, dest ) )
+    {
+        qWarning( "problem renaming file %s to %s", src, dest );
+        return false;
+    }
+    return true;
 }
-*/
 
 /*!
   Opens the document specified by \a f as a readable QIODevice.
   The caller must delete the return value.
 
   Returns 0 if the operation fails.
 */
 QIODevice* FileManager::openFile( const DocLnk& f )
 {
     QString fn = f.file();
     QFile* fl = new QFile( fn );
-    if ( !fl->open( IO_ReadOnly ) ) {
-  delete fl;
+    if ( !fl->open( IO_ReadOnly ) )
+    {
+        delete fl;
         fl = 0;
     }
     return fl;
 }
 
 /*!
@@ -401,16 +286,19 @@ QIODevice* FileManager::openFile( const DocLnk& f )
 */
 QIODevice* FileManager::saveFile( const DocLnk& f )
 {
     QString fn = f.file();
     ensurePathExists( fn );
     QFile* fl = new QFile( fn );
-    if ( fl->open( IO_WriteOnly ) ) {
-  f.writeLink();
-    } else {
-  delete fl;
+    if ( fl->open( IO_WriteOnly ) )
+    {
+        f.writeLink();
+    }
+    else
+    {
+        delete fl;
         fl = 0;
     }
     return fl;
 }
 
 /*!
@@ -419,22 +307,22 @@ QIODevice* FileManager::saveFile( const DocLnk& f )
 */
 bool FileManager::exists( const DocLnk &f )
 {
     return QFile::exists(f.file());
 }
 
-
 /*!
   Ensures that the path \a fn exists, by creating required directories.
   Returns TRUE if successful.
 */
 bool FileManager::ensurePathExists( const QString &fn )
 {
     QFileInfo fi(fn);
     fi.setFile( fi.dirPath(TRUE) );
-    if ( !fi.exists() ) {
-  if ( system(("mkdir -p "+fi.filePath())) )
-      return FALSE;
+    if ( !fi.exists() )
+    {
+        if ( system(("mkdir -p "+fi.filePath())) )
+            return FALSE;
     }
 
     return TRUE;
 }