]> git.saurik.com Git - wxWidgets.git/blobdiff - src/common/mstream.cpp
fixing overrelease and out-of-bounds write, fixes #13725
[wxWidgets.git] / src / common / mstream.cpp
index 7f72e82db47c3e7913367ffc62cc817280b0a768..66af403e2f8140fe46874da4ddd3dcf8902f4cd1 100644 (file)
 /////////////////////////////////////////////////////////////////////////////
-// Name:        mstream.cpp
+// Name:        src/common/mstream.cpp
 // Purpose:     "Memory stream" classes
 // Author:      Guilhem Lavaux
-// Modified by:
+// Modified by: VZ (23.11.00): general code review
 // Created:     04/01/98
 // RCS-ID:      $Id$
 // Copyright:   (c) Guilhem Lavaux
-// Licence:     wxWindows license
+// Licence:     wxWindows licence
 /////////////////////////////////////////////////////////////////////////////
 
-#ifdef __GNUG__
-#pragma implementation "mstream.h"
-#endif
+// ============================================================================
+// declarations
+// ============================================================================
+
+// ----------------------------------------------------------------------------
+// headers
+// ----------------------------------------------------------------------------
 
 // For compilers that support precompilation, includes "wx.h".
 #include "wx/wxprec.h"
-#include <stdlib.h>
-#include <wx/stream.h>
-#include <wx/mstream.h>
 
 #ifdef __BORLANDC__
-#pragma hdrstop
+    #pragma hdrstop
 #endif
 
+#if wxUSE_STREAMS
+
+#include "wx/mstream.h"
+
+#ifndef   WX_PRECOMP
+    #include  "wx/stream.h"
+#endif  //WX_PRECOMP
+
+#include <stdlib.h>
+
+// ============================================================================
+// implementation
+// ============================================================================
+
 // ----------------------------------------------------------------------------
-// wxMemoryStreamBase
+// wxMemoryInputStream
 // ----------------------------------------------------------------------------
-wxMemoryStreamBase::wxMemoryStreamBase()
+
+IMPLEMENT_ABSTRACT_CLASS(wxMemoryInputStream, wxInputStream)
+
+wxMemoryInputStream::wxMemoryInputStream(const void *data, size_t len)
 {
-  m_buffer = NULL;
-  m_iolimit = 0;
-  m_persistent = FALSE;
-  m_length = 0;
+    m_i_streambuf = new wxStreamBuffer(wxStreamBuffer::read);
+    m_i_streambuf->SetBufferIO(const_cast<void *>(data), len);
+    m_i_streambuf->SetIntPosition(0); // seek to start pos
+    m_i_streambuf->Fixed(true);
+
+    m_length = len;
 }
 
-wxMemoryStreamBase::~wxMemoryStreamBase()
+wxMemoryInputStream::wxMemoryInputStream(const wxMemoryOutputStream& stream)
 {
-  if (!m_persistent && m_buffer)
-   free(m_buffer);
+    const wxFileOffset lenFile = stream.GetLength();
+    if ( lenFile == wxInvalidOffset )
+    {
+        m_i_streambuf = NULL;
+        m_lasterror = wxSTREAM_EOF;
+        return;
+    }
+
+    const size_t len = wx_truncate_cast(size_t, lenFile);
+    wxASSERT_MSG( len == lenFile + size_t(0), wxT("huge files not supported") );
+
+    m_i_streambuf = new wxStreamBuffer(wxStreamBuffer::read);
+    m_i_streambuf->SetBufferIO(len); // create buffer
+    stream.CopyTo(m_i_streambuf->GetBufferStart(), len);
+    m_i_streambuf->SetIntPosition(0); // seek to start pos
+    m_i_streambuf->Fixed(true);
+    m_length = len;
 }
 
-bool wxMemoryStreamBase::ChangeBufferSize(size_t new_size)
+void
+wxMemoryInputStream::InitFromStream(wxInputStream& stream, wxFileOffset lenFile)
 {
-  if (m_iolimit == 1)
-    return FALSE;
+    if ( lenFile == wxInvalidOffset )
+        lenFile = stream.GetLength();
+
+    if ( lenFile == wxInvalidOffset )
+    {
+        m_i_streambuf = NULL;
+        m_lasterror = wxSTREAM_EOF;
+        return;
+    }
 
-  m_length = new_size;
-  if (!m_buffer)
-    m_buffer = (char *)malloc(m_length);
-  else
-    m_buffer = (char *)realloc(m_buffer, m_length);
+    const size_t len = wx_truncate_cast(size_t, lenFile);
+    wxASSERT_MSG( (wxFileOffset)len == lenFile, wxT("huge files not supported") );
 
-  return (m_buffer != NULL);
+    m_i_streambuf = new wxStreamBuffer(wxStreamBuffer::read);
+    m_i_streambuf->SetBufferIO(len); // create buffer
+    stream.Read(m_i_streambuf->GetBufferStart(), len);
+    m_i_streambuf->SetIntPosition(0); // seek to start pos
+    m_i_streambuf->Fixed(true);
+    m_length = stream.LastRead();
 }
 
-// ----------------------------------------------------------------------------
-// wxMemoryInputStream
-// ----------------------------------------------------------------------------
-
-wxMemoryInputStream::wxMemoryInputStream(const char *data, size_t len)
+bool wxMemoryInputStream::CanRead() const
 {
-  m_persistent = TRUE;
-  m_length = len;
-  m_buffer = (char *)data; // It's bad.
-  m_position_i = 0;
-  m_lastread = 0;
-  m_eof = FALSE;
-  m_iolimit = 1;
-
-  m_i_streambuf->SetBufferIO(0);
+    return m_i_streambuf->GetIntPosition() != m_length;
 }
 
 wxMemoryInputStream::~wxMemoryInputStream()
 {
+    delete m_i_streambuf;
 }
 
 char wxMemoryInputStream::Peek()
 {
-   // wxStreamBuffer is disabled so just peek the current character.
-
-   return m_buffer[m_position_i];
-}
+    char *buf = (char *)m_i_streambuf->GetBufferStart();
+    size_t pos = m_i_streambuf->GetIntPosition();
+    if ( pos == m_length )
+    {
+        m_lasterror = wxSTREAM_READ_ERROR;
 
-size_t wxMemoryInputStream::DoRead(void *buffer, size_t size)
-{
-  if (m_iolimit == 2) {
-    m_eof = TRUE;
-    return 0; 
-  }
-  if (m_position_i+size > m_length)
-    size = m_length-m_position_i;
-
-  memcpy((void *)((unsigned long)buffer+m_position_i), m_buffer, size);
-  m_position_i += size;
+        return 0;
+    }
 
-  return size;
+    return buf[pos];
 }
 
-off_t wxMemoryInputStream::DoSeekInput(off_t pos, wxSeekMode mode)
+size_t wxMemoryInputStream::OnSysRead(void *buffer, size_t nbytes)
 {
-  if (m_iolimit == 2)
-    return 0;
+    size_t pos = m_i_streambuf->GetIntPosition();
+    if ( pos == m_length )
+    {
+        m_lasterror = wxSTREAM_EOF;
 
-  switch (mode) {
-  case wxFromStart:
-    if ((size_t)pos > m_length)
-      return m_position_i;
-    return (m_position_i = pos);
-
-  case wxFromCurrent:
-    if ((size_t)(m_position_i+pos) > m_length)
-      return m_position_i;
+        return 0;
+    }
 
-    return (m_position_i += pos);
+    m_i_streambuf->Read(buffer, nbytes);
+    m_lasterror = wxSTREAM_NO_ERROR;
 
-  case wxFromEnd:
-    if ((size_t)(m_length-pos) > m_length)
-      return m_position_i;
+    return m_i_streambuf->GetIntPosition() - pos;
+}
 
-    return (m_position_i = m_length-pos);
-  }
+wxFileOffset wxMemoryInputStream::OnSysSeek(wxFileOffset pos, wxSeekMode mode)
+{
+    return m_i_streambuf->Seek(pos, mode);
+}
 
-  return m_position_i;
+wxFileOffset wxMemoryInputStream::OnSysTell() const
+{
+    return m_i_streambuf->Tell();
 }
 
 // ----------------------------------------------------------------------------
 // wxMemoryOutputStream
 // ----------------------------------------------------------------------------
 
-wxMemoryOutputStream::wxMemoryOutputStream(char *data, size_t len)
+IMPLEMENT_DYNAMIC_CLASS(wxMemoryOutputStream, wxOutputStream)
+
+wxMemoryOutputStream::wxMemoryOutputStream(void *data, size_t len)
 {
-  m_persistent = FALSE;
-  m_buffer = data;
-  m_length = len;
-  m_position_o = 0;
-  m_lastwrite = 0;
-  m_bad = FALSE;
-  m_iolimit = 2;
-
-  m_o_streambuf->SetBufferIO(0);
+    m_o_streambuf = new wxStreamBuffer(wxStreamBuffer::write);
+    if ( data )
+        m_o_streambuf->SetBufferIO(data, len);
+    m_o_streambuf->Fixed(false);
+    m_o_streambuf->Flushable(false);
 }
 
 wxMemoryOutputStream::~wxMemoryOutputStream()
 {
-  Sync();
+    delete m_o_streambuf;
 }
 
-size_t wxMemoryOutputStream::DoWrite(const void *buffer, size_t size)
+size_t wxMemoryOutputStream::OnSysWrite(const void *buffer, size_t nbytes)
 {
-  if (m_iolimit == 1) {
-    m_bad = TRUE;
-    return 0;
-  }
-  
-  if (m_position_o+size > m_length)
-    if (!ChangeBufferSize(m_position_o+size)) {
-      m_bad = TRUE;
-      return 0;
-    }
+    size_t oldpos = m_o_streambuf->GetIntPosition();
+    m_o_streambuf->Write(buffer, nbytes);
+    size_t newpos = m_o_streambuf->GetIntPosition();
 
-  memcpy(m_buffer+m_position_o, buffer, size);
-  m_position_o += size;
+    // FIXME can someone please explain what this does? (VZ)
+    if ( !newpos )
+        newpos = m_o_streambuf->GetBufferSize();
 
-  return size;
+    return newpos - oldpos;
 }
 
-off_t wxMemoryOutputStream::DoSeekOutput(off_t pos, wxSeekMode mode)
+wxFileOffset wxMemoryOutputStream::OnSysSeek(wxFileOffset pos, wxSeekMode mode)
 {
-  if (m_iolimit == 1)
-    return 0;
-
-  switch (mode) {
-  case wxFromStart:
-    if ((size_t)pos > m_length)
-      return m_position_o;
-    return (m_position_o = pos);
-
-  case wxFromCurrent:
-    if ((size_t)(m_position_o+pos) > m_length)
-      return m_position_o;
-
-    return (m_position_o += pos);
-
-  case wxFromEnd:
-    if ((size_t)(m_length-pos) > m_length)
-      return m_position_o;
-
-    return (m_position_o = m_length-pos);
-  }
-
-  return m_position_o;
+    return m_o_streambuf->Seek(pos, mode);
 }
 
-// ----------------------------------------------------------------------------
-// wxMemoryStream
-// ----------------------------------------------------------------------------
-
-wxMemoryStream::wxMemoryStream(char *data, size_t len)
-  : wxMemoryInputStream(NULL, 0), wxMemoryOutputStream(NULL, 0)
+wxFileOffset wxMemoryOutputStream::OnSysTell() const
 {
-  m_persistent = FALSE;
-  m_buffer = data;
-  m_length = len;
-  m_iolimit = 0;
+    return m_o_streambuf->Tell();
 }
 
-wxMemoryStream::~wxMemoryStream()
+size_t wxMemoryOutputStream::CopyTo(void *buffer, size_t len) const
 {
+    wxCHECK_MSG( buffer, 0, wxT("must have buffer to CopyTo") );
+
+    if ( len > GetSize() )
+        len = GetSize();
+
+    memcpy(buffer, m_o_streambuf->GetBufferStart(), len);
+
+    return len;
 }
+
+#endif // wxUSE_STREAMS