patch from Dimitri fixing crashes when decoding the invalid GIFs

[wxWidgets.git] / src / common / textfile.cpp

diff --git a/src/common/textfile.cpp b/src/common/textfile.cpp
index 5c74b3ab570b576304866d071a7b696ff4f96ae7..5f4dd6a2687ecfb3b641190e18ecf415847af994 100644 (file)
--- a/src/common/textfile.cpp
+++ b/src/common/textfile.cpp
@@ -1,5 +1,5 @@
 ///////////////////////////////////////////////////////////////////////////////
 ///////////////////////////////////////////////////////////////////////////////

-// Name:        textfile.cpp
+// Name:        src/common/textfile.cpp

 // Purpose:     implementation of wxTextFile class
 // Author:      Vadim Zeitlin
 // Modified by:
 // Purpose:     implementation of wxTextFile class
 // Author:      Vadim Zeitlin
 // Modified by:


@@ -23,231 +23,90 @@
     #pragma hdrstop
 #endif  //__BORLANDC__
 
     #pragma hdrstop
 #endif  //__BORLANDC__
 

-#if !wxUSE_FILE
+#if !wxUSE_FILE || !wxUSE_TEXTBUFFER

     #undef wxUSE_TEXTFILE
     #define wxUSE_TEXTFILE 0
 #endif // wxUSE_FILE
 
     #undef wxUSE_TEXTFILE
     #define wxUSE_TEXTFILE 0
 #endif // wxUSE_FILE
 

+#if wxUSE_TEXTFILE
+

 #ifndef WX_PRECOMP
 #ifndef WX_PRECOMP

-    #include  "wx/string.h"
-    #include  "wx/intl.h"
-    #include  "wx/file.h"
-    #include  "wx/log.h"
+    #include "wx/string.h"
+    #include "wx/intl.h"
+    #include "wx/file.h"
+    #include "wx/log.h"

 #endif
 
 #endif
 

-#include  "wx/textfile.h"
+#include "wx/textfile.h"
+#include "wx/filename.h"

 
 // ============================================================================
 // wxTextFile class implementation
 // ============================================================================
 
 
 // ============================================================================
 // wxTextFile class implementation
 // ============================================================================
 

-// ----------------------------------------------------------------------------
-// static methods (always compiled in)
-// ----------------------------------------------------------------------------
-
-// default type is the native one
-const wxTextFileType wxTextFile::typeDefault =
-#if   defined(__WINDOWS__)
-  wxTextFileType_Dos;
-#elif defined(__UNIX__)
-  wxTextFileType_Unix;
-#elif defined(__WXMAC__)
-  wxTextFileType_Mac;
-#elif defined(__WXPM__)
-  wxTextFileType_Os2;
-#else
-  wxTextFileType_None;
-  #error  "wxTextFile: unsupported platform."
-#endif
-
-const wxChar *wxTextFile::GetEOL(wxTextFileType type)
+wxTextFile::wxTextFile(const wxString& strFileName)
+          : wxTextBuffer(strFileName)

 {
 {

-    switch ( type ) {
-      default:
-        wxFAIL_MSG(wxT("bad file type in wxTextFile::GetEOL."));
-        // fall through nevertheless - we must return something...
-
-      case wxTextFileType_None: return wxT("");
-      case wxTextFileType_Unix: return wxT("\n");
-      case wxTextFileType_Dos:  return wxT("\r\n");
-      case wxTextFileType_Mac:  return wxT("\r");
-    }

 }
 
 
 }
 
 

-wxString wxTextFile::Translate(const wxString& text, wxTextFileType type)
-{
-    // don't do anything if there is nothing to do
-    if ( type == wxTextFileType_None )
-        return text;
-
-    wxString eol = GetEOL(type), result;
-
-    // optimization: we know that the length of the new string will be about
-    // the same as the length of the old one, so prealloc memory to aviod
-    // unnecessary relocations
-    result.Alloc(text.Len());
-
-    wxChar chLast = 0;
-    for ( const wxChar *pc = text.c_str(); *pc; pc++ )
-    {
-        wxChar ch = *pc;
-        switch ( ch ) {
-            case '\n':
-                // Dos/Unix line termination
-                result += eol;
-                chLast = '\n';
-                break;
-
-            case '\r':
-                if ( chLast == '\r' ) {
-                    // Mac empty line
-                    result += eol;
-                }
-                else
-                    chLast = '\r';
-                break;
-
-            default:
-                if ( chLast == '\r' ) {
-                    // Mac line termination
-                    result += eol;
-                    chLast = ch;
-                }
-                else {
-                    // add to the current line
-                    result += ch;
-                }
-        }
-    }
-
-    return result;
-}
-
-#if wxUSE_TEXTFILE
-
-// ----------------------------------------------------------------------------
-// ctors & dtor
-// ----------------------------------------------------------------------------
-
-wxTextFile::wxTextFile(const wxString& strFile) : m_strFile(strFile)
-{
-  m_nCurLine = 0;
-  m_isOpened = FALSE;
-}
-
-wxTextFile::~wxTextFile()
-{
-  // m_file dtor called automatically
-}
-

 // ----------------------------------------------------------------------------
 // file operations
 // ----------------------------------------------------------------------------
 
 // ----------------------------------------------------------------------------
 // file operations
 // ----------------------------------------------------------------------------
 

-bool wxTextFile::Exists() const
+bool wxTextFile::OnExists() const

 {
 {

-    return wxFile::Exists(m_strFile);
+    return wxFile::Exists(m_strBufferName);

 }
 
 }
 

-bool wxTextFile::Open(const wxString& strFile)
-{
-  m_strFile = strFile;
-
-  return Open();
-}

 
 

-bool wxTextFile::Open()
+bool wxTextFile::OnOpen(const wxString &strBufferName, wxTextBufferOpenMode OpenMode)

 {
 {

-  // file name must be either given in ctor or in Open(const wxString&)
-  wxASSERT( !m_strFile.IsEmpty() );
-
-  // open file in read-only mode
-  if ( !m_file.Open(m_strFile) )
-    return FALSE;
+    wxFile::OpenMode FileOpenMode;

 
 

-  // read file into memory
-  m_isOpened = Read();
-
-  m_file.Close();
-
-  return m_isOpened;
-}
+    switch ( OpenMode )
+    {
+        default:
+            wxFAIL_MSG( _T("unknown open mode in wxTextFile::Open") );
+            // fall through

 
 

-// analyse some lines of the file trying to guess it's type.
-// if it fails, it assumes the native type for our platform.
-wxTextFileType wxTextFile::GuessType() const
-{
-  // file should be opened and we must be in it's beginning
-  wxASSERT( m_file.IsOpened() && m_file.Tell() == 0 );
+        case ReadAccess :
+            FileOpenMode = wxFile::read;
+            break;

 
 

-  // scan the file lines
-  size_t nUnix = 0,     // number of '\n's alone
-       nDos  = 0,     // number of '\r\n'
-       nMac  = 0;     // number of '\r's
-
-  // we take MAX_LINES_SCAN in the beginning, middle and the end of file
-  #define MAX_LINES_SCAN    (10)
-  size_t nCount = m_aLines.Count() / 3,
-       nScan =  nCount > 3*MAX_LINES_SCAN ? MAX_LINES_SCAN : nCount / 3;
-
-  #define   AnalyseLine(n)              \
-    switch ( m_aTypes[n] ) {            \
-      case wxTextFileType_Unix: nUnix++; break;   \
-      case wxTextFileType_Dos:  nDos++;  break;   \
-      case wxTextFileType_Mac:  nMac++;  break;   \
-      default: wxFAIL_MSG(_("unknown line terminator")); \
+        case WriteAccess :
+            FileOpenMode = wxFile::write;
+            break;

     }
 
     }
 

-  size_t n;
-  for ( n = 0; n < nScan; n++ )     // the beginning
-    AnalyseLine(n);
-  for ( n = (nCount - nScan)/2; n < (nCount + nScan)/2; n++ )
-    AnalyseLine(n);
-  for ( n = nCount - nScan; n < nCount; n++ )
-    AnalyseLine(n);
-
-  #undef   AnalyseLine
-
-  // interpret the results (FIXME far from being even 50% fool proof)
-  if ( nDos + nUnix + nMac == 0 ) {
-    // no newlines at all
-    wxLogWarning(_("'%s' is probably a binary file."), m_strFile.c_str());
-  }
-  else {
-    #define   GREATER_OF(t1, t2) n##t1 == n##t2 ? typeDefault               \
-                                                : n##t1 > n##t2             \
-                                                    ? wxTextFileType_##t1   \
-                                                    : wxTextFileType_##t2
-
-    // Watcom C++ doesn't seem to be able to handle the macro
-#if !defined(__WATCOMC__)
-    if ( nDos > nUnix )
-      return GREATER_OF(Dos, Mac);
-    else if ( nDos < nUnix )
-      return GREATER_OF(Unix, Mac);
-    else {
-      // nDos == nUnix
-      return nMac > nDos ? wxTextFileType_Mac : typeDefault;
-    }
-#endif // __WATCOMC__
+    return m_file.Open(strBufferName.c_str(), FileOpenMode);
+}

 
 

-    #undef    GREATER_OF
-  }

 
 

-  return typeDefault;
+bool wxTextFile::OnClose()
+{
+    return m_file.Close();

 }
 
 }
 

-bool wxTextFile::Read()
+
+bool wxTextFile::OnRead(wxMBConv& conv)

 {
   // file should be opened and we must be in it's beginning
   wxASSERT( m_file.IsOpened() && m_file.Tell() == 0 );
 
 {
   // file should be opened and we must be in it's beginning
   wxASSERT( m_file.IsOpened() && m_file.Tell() == 0 );
 

+#if wxUSE_UNICODE
+  char conv_mbBuf[2];
+  wchar_t conv_wcBuf[2];
+  conv_mbBuf[1] = 0;
+#else
+  (void)conv;
+#endif
+

   wxString str;
   char ch, chLast = '\0';
   char buf[1024];
   int n, nRead;
   wxString str;
   char ch, chLast = '\0';
   char buf[1024];
   int n, nRead;

-  while ( !m_file.Eof() ) {
+  do {

     nRead = m_file.Read(buf, WXSIZEOF(buf));
     if ( nRead == wxInvalidOffset ) {
       // read error (error message already given in wxFile::Read)
     nRead = m_file.Read(buf, WXSIZEOF(buf));
     if ( nRead == wxInvalidOffset ) {
       // read error (error message already given in wxFile::Read)


@@ -259,8 +118,7 @@ bool wxTextFile::Read()
       switch ( ch ) {
         case '\n':
           // Dos/Unix line termination
       switch ( ch ) {
         case '\n':
           // Dos/Unix line termination

-          m_aLines.Add(str);
-          m_aTypes.Add(chLast == '\r' ? wxTextFileType_Dos
+                AddLine(str, chLast == '\r' ? wxTextFileType_Dos

                                       : wxTextFileType_Unix);
           str.Empty();
           chLast = '\n';
                                       : wxTextFileType_Unix);
           str.Empty();
           chLast = '\n';


@@ -269,8 +127,7 @@ bool wxTextFile::Read()
         case '\r':
           if ( chLast == '\r' ) {
             // Mac empty line
         case '\r':
           if ( chLast == '\r' ) {
             // Mac empty line

-            m_aLines.Add("");
-            m_aTypes.Add(wxTextFileType_Mac);
+                    AddLine(wxEmptyString, wxTextFileType_Mac);

           }
           else
             chLast = '\r';
           }
           else
             chLast = '\r';


@@ -279,56 +136,64 @@ bool wxTextFile::Read()
         default:
           if ( chLast == '\r' ) {
             // Mac line termination
         default:
           if ( chLast == '\r' ) {
             // Mac line termination

-            m_aLines.Add(str);
-            m_aTypes.Add(wxTextFileType_Mac);
+                    AddLine(str, wxTextFileType_Mac);

             chLast = ch;
             chLast = ch;

+#if wxUSE_UNICODE
+            conv_mbBuf[0] = ch;
+            if (conv.MB2WC(conv_wcBuf, conv_mbBuf, 2) == (size_t)-1)
+                conv_wcBuf[0] = ch;
+            str = conv_wcBuf[0];
+#else

             str = ch;
             str = ch;

+#endif

           }
           else {
             // add to the current line
           }
           else {
             // add to the current line

+#if wxUSE_UNICODE
+            conv_mbBuf[0] = ch;
+            if (conv.MB2WC(conv_wcBuf, conv_mbBuf, 2) == (size_t)-1)
+                conv_wcBuf[0] = ch;
+            str += conv_wcBuf[0];
+#else

             str += ch;
             str += ch;

+#endif

           }
       }
     }
           }
       }
     }

-  }
+  } while ( nRead == WXSIZEOF(buf) );

 
   // anything in the last line?
   if ( !str.IsEmpty() ) {
 
   // anything in the last line?
   if ( !str.IsEmpty() ) {

-    m_aTypes.Add(wxTextFileType_None);  // no line terminator
-    m_aLines.Add(str);
+        AddLine(str, wxTextFileType_None);  // no line terminator

   }
 
   return TRUE;
 }
 
   }
 
   return TRUE;
 }
 

-bool wxTextFile::Close()
-{
-    m_aTypes.Clear();
-    m_aLines.Clear();
-    m_nCurLine = 0;
-    m_isOpened = FALSE;
-
-    return TRUE;
-}

 
 

-bool wxTextFile::Write(wxTextFileType typeNew)
+bool wxTextFile::OnWrite(wxTextFileType typeNew, wxMBConv& conv)

 {
 {

-  wxTempFile fileTmp(m_strFile);
+    wxFileName fn = m_strBufferName;
+    if ( !fn.IsAbsolute() )
+        fn.Normalize();

 
 

-  if ( !fileTmp.IsOpened() ) {
-    wxLogError(_("can't write file '%s' to disk."), m_strFile.c_str());
-    return FALSE;
-  }
+    wxTempFile fileTmp(fn.GetFullPath());

 
 

-  size_t nCount = m_aLines.Count();
-  for ( size_t n = 0; n < nCount; n++ ) {
-    fileTmp.Write(m_aLines[n] +
-                  GetEOL(typeNew == wxTextFileType_None ? m_aTypes[n]
-                                                        : typeNew));
-  }
+    if ( !fileTmp.IsOpened() ) {
+        wxLogError(_("can't write buffer '%s' to disk."), m_strBufferName.c_str());
+        return FALSE;
+    }
+
+    size_t nCount = GetLineCount();
+    for ( size_t n = 0; n < nCount; n++ ) {
+        fileTmp.Write(GetLine(n) +
+                      GetEOL(typeNew == wxTextFileType_None ? GetLineType(n)
+                                                            : typeNew),
+                      conv);
+    }

 
 

-  // replace the old file with this one
-  return fileTmp.Commit();
+    // replace the old file with this one
+    return fileTmp.Commit();

 }
 
 #endif // wxUSE_TEXTFILE
 }
 
 #endif // wxUSE_TEXTFILE