fix filesaving when filename contains extended characters

2 files changed, 121 insertions, 52 deletions

diff --git a/library/filemanager.cpp b/library/filemanager.cpp
index 2b97846..cc657fa 100644
--- a/library/filemanager.cpp
+++ b/library/filemanager.cpp
@@ -25,12 +25,17 @@
 #include <qfileinfo.h>
 #include <qtextstream.h>
 #include <qtextcodec.h>
 
 #include <errno.h>
 #include <stdlib.h>
+#include <unistd.h>
+#include <sys/stat.h>
+#include <dirent.h>
+#include <sys/sendfile.h>
+#include <fcntl.h>
 
 /*!
   \class FileManager
   \brief The FileManager class assists with AppLnk input/output.
 */
 
@@ -53,29 +58,31 @@ 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";
+ QString fn = f.file() + ".new";
     ensurePathExists( fn );
     QFile fl( fn );
-    if ( !fl.open( IO_WriteOnly|IO_Raw ) )
-        return FALSE;
+    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;
+  QFile::remove( fn );
+  return FALSE;
     }
+       qDebug("total written %d out of %d", total_written, data.size());
     // else rename the file...
-    if ( ::rename( fn.latin1(), f.file().latin1() ) < 0 ) {
-        qWarning( "problem renaming file %s to %s, errno: %d", fn.latin1(),
-                  f.file().latin1(), errno );
-        // remove the file...
-        QFile::remove( fn );
+    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...
     }
     return TRUE;
 }
 
 /*!
   Saves \a text as the document specified by \a f.
@@ -83,34 +90,33 @@ bool FileManager::saveFile( const DocLnk &f, const QByteArray &data )
   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";
+ QString fn = f.file() + ".new";
     ensurePathExists( fn );
     QFile fl( fn );
     if ( !fl.open( IO_WriteOnly|IO_Raw ) ) {
-        qDebug( "open failed: %s", fn.latin1() );
-        return FALSE;
+       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;
+       QFile::remove( fn );
+       return FALSE;
     }
-    // okay now rename the file...
-    if ( ::rename( fn.latin1(), f.file().latin1() ) < 0 ) {
-        qWarning( "problem renaming file %s to %s, errno: %d", fn.latin1(),
-                  f.file().latin1(), errno );
-        // remove the tmp file, otherwise, it will just lay around...
-        QFile::remove( fn.latin1() );
+    // 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 );
+
     }
     return TRUE;
 }
 
 
 /*!
@@ -122,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 );
@@ -146,16 +152,16 @@ 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
@@ -164,67 +170,128 @@ bool FileManager::loadFile( const DocLnk &f, QByteArray &ba )
   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;
+  return FALSE;
 
     QString fn = dest.file() + ".new";
     ensurePathExists( fn );
     QFile df( fn );
     if ( !df.open( IO_WriteOnly|IO_Raw ) )
-        return FALSE;
+  return FALSE;
 
     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;
-        }
+  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;
+  }
     }
     
     if ( ok )
-        ok = dest.writeLink();
+  ok = dest.writeLink();
     
     if ( ok ) {
-        // okay now rename the file...
-        if ( ::rename( fn.latin1(), dest.file().latin1() ) < 0 ) {
-            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() );
-        }
+  // 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() );
+  QFile::remove( fn.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;
+         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();
+    // Set file permissions
+  if( stat( (const char *) src, &status ) == 0 ) {
+      chmod( (const char *) dest, status.st_mode );
+    }
+
+  return success;
+}
+
+
+bool FileManager::renameFile( const QString & src, const QString & dest ) {
+   if(copyFile( src, dest )) {
+      if(QFile::remove(src) ) {
+       return true;
+      }
+   }
+    return false;  
+}
+
+
 /*!
   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;
+  delete fl;
         fl = 0;
     }
     return fl;
 }
 
 /*!
@@ -236,15 +303,15 @@ 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();
+  f.writeLink();
     } else {
-        delete fl;
+  delete fl;
         fl = 0;
     }
     return fl;
 }
 
 /*!
@@ -263,12 +330,12 @@ bool FileManager::exists( const DocLnk &f )
 */
 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 ( system(("mkdir -p "+fi.filePath())) )
+      return FALSE;
     }
 
     return TRUE;
 }
diff --git a/library/filemanager.h b/library/filemanager.h
index f8d9425..61a3341 100644
--- a/library/filemanager.h
+++ b/library/filemanager.h
@@ -35,14 +35,16 @@ public:
 
     bool saveFile( const DocLnk&, const QByteArray &data );
     bool saveFile( const DocLnk&, const QString &text );
     bool loadFile( const DocLnk&, QByteArray &data );
     bool loadFile( const DocLnk&, QString &text );
     bool copyFile( const AppLnk &src, const AppLnk &dest );
-
-    // The caller must delete the return values.
+    bool copyFile( const QString & src, const QString & dest );
+    bool renameFile( const QString &, const QString &);
+ 
+// The caller must delete the return values.
     QIODevice* openFile( const DocLnk& );
     QIODevice* saveFile( const DocLnk& );
 
     bool exists( const DocLnk& );
 
 protected: