src/common/execcmn.cpp

   1 ///////////////////////////////////////////////////////////////////////////////
   2 // Name:        src/common/execcmn.cpp
   3 // Purpose:     defines wxStreamTempInputBuffer which is used by Unix and MSW
   4 //              implementations of wxExecute; this file is only used by the
   5 //              library and never by the user code
   6 // Author:      Vadim Zeitlin
   7 // Modified by:
   8 // Created:     20.08.02
   9 // RCS-ID:      $Id$
  10 // Copyright:   (c) 2002 Vadim Zeitlin <vadim@wxwindows.org>
  11 // Licence:     wxWindows licence
  12 ///////////////////////////////////////////////////////////////////////////////
  13
  14 #ifndef _WX_WXEXEC_CPP_
  15 #define _WX_WXEXEC_CPP_
  16
  17 // this file should never be compiled directly, just included by other code
  18 #ifndef _WX_USED_BY_WXEXECUTE_
  19     #error "You should never directly build this file!"
  20 #endif
  21
  22 // ----------------------------------------------------------------------------
  23 // wxStreamTempInputBuffer
  24 // ----------------------------------------------------------------------------
  25
  26 /*
  27    wxStreamTempInputBuffer is a hack which we need to solve the problem of
  28    executing a child process synchronously with IO redirecting: when we do
  29    this, the child writes to a pipe we open to it but when the pipe buffer
  30    (which has finite capacity, e.g. commonly just 4Kb) becomes full we have to
  31    read data from it because the child blocks in its write() until then and if
  32    it blocks we are never going to return from wxExecute() so we dead lock.
  33
  34    So here is the fix: we now read the output as soon as it appears into a temp
  35    buffer (wxStreamTempInputBuffer object) and later just stuff it back into
  36    the stream when the process terminates. See supporting code in wxExecute()
  37    itself as well.
  38
  39    Note that this is horribly inefficient for large amounts of output (count
  40    the number of times we copy the data around) and so a better API is badly
  41    needed! However it's not easy to devise a way to do this keeping backwards
  42    compatibility with the existing wxExecute(wxEXEC_SYNC)...
  43 */
  44
  45 class wxStreamTempInputBuffer
  46 {
  47 public:
  48     wxStreamTempInputBuffer();
  49
  50     // call to associate a stream with this buffer, otherwise nothing happens
  51     // at all
  52     void Init(wxPipeInputStream *stream);
  53
  54     // check for input on our stream and cache it in our buffer if any
  55     //
  56     // return true if anything was done
  57     bool Update();
  58
  59     ~wxStreamTempInputBuffer();
  60
  61 private:
  62     // the stream we're buffering, if NULL we don't do anything at all
  63     wxPipeInputStream *m_stream;
  64
  65     // the buffer of size m_size (NULL if m_size == 0)
  66     void *m_buffer;
  67
  68     // the size of the buffer
  69     size_t m_size;
  70
  71     wxDECLARE_NO_COPY_CLASS(wxStreamTempInputBuffer);
  72 };
  73
  74 inline wxStreamTempInputBuffer::wxStreamTempInputBuffer()
  75 {
  76     m_stream = NULL;
  77     m_buffer = NULL;
  78     m_size = 0;
  79 }
  80
  81 inline void wxStreamTempInputBuffer::Init(wxPipeInputStream *stream)
  82 {
  83     m_stream = stream;
  84 }
  85
  86 inline
  87 bool wxStreamTempInputBuffer::Update()
  88 {
  89     if ( !m_stream || !m_stream->CanRead() )
  90         return false;
  91
  92     // realloc in blocks of 4Kb: this is the default (and minimal) buffer
  93     // size of the Unix pipes so it should be the optimal step
  94     //
  95     // NB: don't use "static int" in this inline function, some compilers
  96     //     (e.g. IBM xlC) don't like it
  97     enum { incSize = 4096 };
  98
  99     void *buf = realloc(m_buffer, m_size + incSize);
 100     if ( !buf )
 101     {
 102         // don't read any more, we don't have enough memory to do it
 103         m_stream = NULL;
 104     }
 105     else // got memory for the buffer
 106     {
 107         m_buffer = buf;
 108         m_stream->Read((char *)m_buffer + m_size, incSize);
 109         m_size += m_stream->LastRead();
 110     }
 111
 112     return true;
 113 }
 114
 115 inline
 116 wxStreamTempInputBuffer::~wxStreamTempInputBuffer()
 117 {
 118     if ( m_buffer )
 119     {
 120         m_stream->Ungetch(m_buffer, m_size);
 121         free(m_buffer);
 122     }
 123 }
 124
 125 #endif // _WX_WXEXEC_CPP_
 126