]> git.saurik.com Git - wxWidgets.git/blame - src/common/stream.cpp
fixed GetPath() trailing separator bug, rewrote GetFullPath() in terms of GetPath()
[wxWidgets.git] / src / common / stream.cpp
CommitLineData
3d4c6a21 1/////////////////////////////////////////////////////////////////////////////
49e399d8 2// Name: src/common/stream.cpp
3d4c6a21
GL
3// Purpose: wxStream base classes
4// Author: Guilhem Lavaux
49e399d8
VZ
5// Modified by: VZ (23.11.00) to fix realloc()ing new[]ed memory,
6// general code review
3d4c6a21
GL
7// Created: 11/07/98
8// RCS-ID: $Id$
9// Copyright: (c) Guilhem Lavaux
10// Licence: wxWindows license
11/////////////////////////////////////////////////////////////////////////////
12
49e399d8
VZ
13// ============================================================================
14// declarations
15// ============================================================================
16
17// ----------------------------------------------------------------------------
18// headers
19// ----------------------------------------------------------------------------
47fc03d2 20
3d4c6a21 21#ifdef __GNUG__
47fc03d2 22 #pragma implementation "stream.h"
3d4c6a21
GL
23#endif
24
db138a4c
JS
25// For compilers that support precompilation, includes "wx.h".
26#include "wx/wxprec.h"
ce4169a4
RR
27
28#ifdef __BORLANDC__
29 #pragma hdrstop
30#endif
31
32#ifndef WX_PRECOMP
33 #include "wx/defs.h"
34#endif
35
36#if wxUSE_STREAMS
37
6d44bf31 38#include <ctype.h>
3096bd2f
VZ
39#include "wx/stream.h"
40#include "wx/datstrm.h"
733b8ed3 41#include "wx/textfile.h"
adc35078 42#include "wx/log.h"
db138a4c 43
49e399d8
VZ
44// ----------------------------------------------------------------------------
45// constants
46// ----------------------------------------------------------------------------
47
48// the temporary buffer size used when copying from stream to stream
8ef6a930
GL
49#define BUF_TEMP_SIZE 10000
50
49e399d8
VZ
51// ============================================================================
52// implementation
53// ============================================================================
54
6d44bf31
GL
55// ----------------------------------------------------------------------------
56// wxStreamBuffer
57// ----------------------------------------------------------------------------
58
49e399d8
VZ
59void wxStreamBuffer::SetError(wxStreamError err)
60{
61 if ( m_stream->m_lasterror == wxStream_NOERROR )
62 m_stream->m_lasterror = err;
63}
64
65void wxStreamBuffer::InitBuffer()
66{
67 m_buffer_start =
68 m_buffer_end =
69 m_buffer_pos = NULL;
70 m_buffer_size = 0;
71
d4b45f48
VZ
72 // if we are going to allocate the buffer, we should free it later as well
73 m_destroybuf = TRUE;
49e399d8
VZ
74}
75
76void wxStreamBuffer::Init()
77{
78 InitBuffer();
79
80 m_fixed = TRUE;
81}
926c550d 82
75ed1d15 83wxStreamBuffer::wxStreamBuffer(wxStreamBase& stream, BufMode mode)
6d44bf31 84{
a2bd1fbb
VZ
85 Init();
86
49e399d8
VZ
87 m_stream = &stream;
88 m_mode = mode;
89
90 m_flushable = TRUE;
91 m_destroystream = FALSE;
6d44bf31
GL
92}
93
84b46c35 94wxStreamBuffer::wxStreamBuffer(BufMode mode)
84b46c35 95{
a2bd1fbb
VZ
96 Init();
97
ba0b7b32
VS
98 wxASSERT_MSG(mode != read_write, wxT("you have to use the other ctor for read_write mode") );
99 if ( mode == read )
100 m_stream = new wxInputStream;
101 else if ( mode == write)
102 m_stream = new wxOutputStream;
103 else
104 m_stream = NULL;
105
49e399d8
VZ
106 m_mode = mode;
107
108 m_flushable = FALSE;
109 m_destroystream = TRUE;
84b46c35
GL
110}
111
112wxStreamBuffer::wxStreamBuffer(const wxStreamBuffer& buffer)
113{
49e399d8
VZ
114 // doing this has big chances to lead to a crashwhen the source buffer is
115 // destroyed (otherwise assume the caller knows what he does)
116 wxASSERT_MSG( !buffer.m_destroybuf && !buffer.m_destroystream,
117 _T("it's a bad idea to copy this buffer") );
118
119 m_buffer_start = buffer.m_buffer_start;
120 m_buffer_end = buffer.m_buffer_end;
121 m_buffer_pos = buffer.m_buffer_pos;
122 m_buffer_size = buffer.m_buffer_size;
123 m_fixed = buffer.m_fixed;
124 m_flushable = buffer.m_flushable;
125 m_stream = buffer.m_stream;
126 m_mode = buffer.m_mode;
127 m_destroybuf = FALSE;
128 m_destroystream = FALSE;
84b46c35
GL
129}
130
49e399d8 131void wxStreamBuffer::FreeBuffer()
6d44bf31 132{
49e399d8
VZ
133 if ( m_destroybuf )
134 free(m_buffer_start);
6d44bf31
GL
135}
136
49e399d8 137wxStreamBuffer::~wxStreamBuffer()
6d44bf31 138{
49e399d8 139 FreeBuffer();
6d44bf31 140
49e399d8
VZ
141 if ( m_destroystream )
142 delete m_stream;
6d44bf31
GL
143}
144
47fc03d2
VZ
145wxInputStream *wxStreamBuffer::GetInputStream() const
146{
147 return m_mode == write ? NULL : (wxInputStream *)m_stream;
148}
149
150wxOutputStream *wxStreamBuffer::GetOutputStream() const
151{
152 return m_mode == read ? NULL : (wxOutputStream *)m_stream;
153}
154
49e399d8
VZ
155void wxStreamBuffer::SetBufferIO(void *buffer_start,
156 void *buffer_end,
157 bool takeOwnership)
67c8c225
VZ
158{
159 SetBufferIO(buffer_start, (char *)buffer_end - (char *)buffer_start,
160 takeOwnership);
161}
162
163void wxStreamBuffer::SetBufferIO(void *start,
164 size_t len,
165 bool takeOwnership)
6d44bf31 166{
49e399d8
VZ
167 // start by freeing the old buffer
168 FreeBuffer();
885ee235 169
67c8c225
VZ
170 m_buffer_start = (char *)start;
171 m_buffer_end = m_buffer_start + len;
6d44bf31 172
67c8c225 173 m_buffer_size = len;
49e399d8
VZ
174
175 // if we own it, we free it
421db2ad 176 m_destroybuf = takeOwnership;
49e399d8
VZ
177
178 ResetBuffer();
179}
6d44bf31 180
49e399d8
VZ
181void wxStreamBuffer::SetBufferIO(size_t bufsize)
182{
183 // start by freeing the old buffer
184 FreeBuffer();
185
186 if ( bufsize )
187 {
67c8c225 188 SetBufferIO(malloc(bufsize), bufsize, TRUE /* take ownership */);
49e399d8
VZ
189 }
190 else // no buffer size => no buffer
191 {
192 InitBuffer();
193 }
6d44bf31
GL
194}
195
196void wxStreamBuffer::ResetBuffer()
197{
49e399d8
VZ
198 wxCHECK_RET( m_stream, _T("should have a stream in wxStreamBuffer") );
199
200 m_stream->m_lasterror = wxStream_NOERROR;
201 m_stream->m_lastcount = 0;
202 if (m_mode == read && m_flushable)
203 m_buffer_pos = m_buffer_end;
204 else
205 m_buffer_pos = m_buffer_start;
6d44bf31
GL
206}
207
49e399d8 208// fill the buffer with as much data as possible (only for read buffers)
75ed1d15
GL
209bool wxStreamBuffer::FillBuffer()
210{
47fc03d2
VZ
211 wxInputStream *inStream = GetInputStream();
212
213 wxCHECK_MSG( inStream, FALSE, _T("should have a stream in wxStreamBuffer") );
75ed1d15 214
47fc03d2 215 size_t count = inStream->OnSysRead(m_buffer_start, m_buffer_size);
49e399d8
VZ
216 if ( !count )
217 return FALSE;
75ed1d15 218
49e399d8
VZ
219 m_buffer_end = m_buffer_start + count;
220 m_buffer_pos = m_buffer_start;
221
222 return TRUE;
75ed1d15
GL
223}
224
49e399d8 225// write the buffer contents to the stream (only for write buffers)
75ed1d15
GL
226bool wxStreamBuffer::FlushBuffer()
227{
49e399d8 228 wxCHECK_MSG( m_flushable, FALSE, _T("can't flush this buffer") );
75ed1d15 229
49e399d8
VZ
230 // FIXME: what is this check for? (VZ)
231 if ( m_buffer_pos == m_buffer_start )
232 return FALSE;
75ed1d15 233
47fc03d2
VZ
234 wxOutputStream *outStream = GetOutputStream();
235
236 wxCHECK_MSG( outStream, FALSE, _T("should have a stream in wxStreamBuffer") );
49e399d8
VZ
237
238 size_t current = m_buffer_pos - m_buffer_start;
47fc03d2 239 size_t count = outStream->OnSysWrite(m_buffer_start, current);
49e399d8
VZ
240 if ( count != current )
241 return FALSE;
242
243 m_buffer_pos = m_buffer_start;
75ed1d15 244
49e399d8 245 return TRUE;
75ed1d15
GL
246}
247
49e399d8
VZ
248size_t wxStreamBuffer::GetDataLeft()
249{
250 /* Why is this done? RR. */
251 if ( m_buffer_pos == m_buffer_end && m_flushable)
252 FillBuffer();
253
254 return GetBytesLeft();
255}
256
257// copy up to size bytes from our buffer into the provided one
75ed1d15
GL
258void wxStreamBuffer::GetFromBuffer(void *buffer, size_t size)
259{
49e399d8
VZ
260 // don't get more bytes than left in the buffer
261 size_t left = GetBytesLeft();
75ed1d15 262
49e399d8
VZ
263 if ( size > left )
264 size = left;
75ed1d15 265
49e399d8
VZ
266 memcpy(buffer, m_buffer_pos, size);
267 m_buffer_pos += size;
75ed1d15
GL
268}
269
49e399d8 270// copy the contents of the provided buffer into this one
75ed1d15
GL
271void wxStreamBuffer::PutToBuffer(const void *buffer, size_t size)
272{
49e399d8
VZ
273 size_t left = GetBytesLeft();
274 if ( size > left )
275 {
276 if ( m_fixed )
277 {
278 // we can't realloc the buffer, so just copy what we can
279 size = left;
280 }
281 else // !m_fixed
282 {
283 // realloc the buffer to have enough space for the data
284 size_t delta = m_buffer_pos - m_buffer_start;
285
286 char *startOld = m_buffer_start;
287 m_buffer_size += size;
288 m_buffer_start = (char *)realloc(m_buffer_start, m_buffer_size);
289 if ( !m_buffer_start )
290 {
291 // don't leak memory if realloc() failed
292 m_buffer_start = startOld;
293 m_buffer_size -= size;
294
295 // what else can we do?
296 return;
297 }
298
299 // adjust the pointers invalidated by realloc()
300 m_buffer_pos = m_buffer_start + delta;
301 m_buffer_end = m_buffer_start + m_buffer_size;
302 }
56dc1ffd 303 }
49e399d8
VZ
304
305 memcpy(m_buffer_pos, buffer, size);
306 m_buffer_pos += size;
75ed1d15
GL
307}
308
84b46c35
GL
309void wxStreamBuffer::PutChar(char c)
310{
47fc03d2
VZ
311 wxOutputStream *outStream = GetOutputStream();
312
313 wxCHECK_RET( outStream, _T("should have a stream in wxStreamBuffer") );
84b46c35 314
49e399d8
VZ
315 // if we don't have buffer at all, just forward this call to the stream,
316 if ( !HasBuffer() )
317 {
47fc03d2 318 outStream->OnSysWrite(&c, 1);
49e399d8
VZ
319 }
320 else
321 {
322 // otherwise check we have enough space left
323 if ( !GetDataLeft() && !FlushBuffer() )
324 {
325 // we don't
326 SetError(wxStream_WRITE_ERR);
327 }
328 else
329 {
330 PutToBuffer(&c, 1);
331 m_stream->m_lastcount = 1;
332 }
333 }
84b46c35
GL
334}
335
6319afe3
GL
336char wxStreamBuffer::Peek()
337{
49e399d8
VZ
338 wxCHECK_MSG( m_stream && HasBuffer(), 0,
339 _T("should have the stream and the buffer in wxStreamBuffer") );
6319afe3 340
49e399d8
VZ
341 if ( !GetDataLeft() )
342 {
343 SetError(wxStream_READ_ERR);
344 return 0;
345 }
6319afe3 346
49e399d8
VZ
347 char c;
348 GetFromBuffer(&c, 1);
349 m_buffer_pos--;
6319afe3 350
49e399d8 351 return c;
6319afe3
GL
352}
353
84b46c35
GL
354char wxStreamBuffer::GetChar()
355{
47fc03d2
VZ
356 wxInputStream *inStream = GetInputStream();
357
358 wxCHECK_MSG( inStream, 0, _T("should have a stream in wxStreamBuffer") );
84b46c35 359
49e399d8
VZ
360 char c;
361 if ( !HasBuffer() )
362 {
47fc03d2 363 inStream->OnSysRead(&c, 1);
49e399d8
VZ
364 }
365 else
366 {
367 if ( !GetDataLeft() )
368 {
369 SetError(wxStream_READ_ERR);
370 c = 0;
371 }
372 else
373 {
374 GetFromBuffer(&c, 1);
375 m_stream->m_lastcount = 1;
376 }
377 }
84b46c35 378
84b46c35 379 return c;
84b46c35
GL
380}
381
8ef6a930 382size_t wxStreamBuffer::Read(void *buffer, size_t size)
6d44bf31 383{
47fc03d2 384 wxInputStream *inStream = GetInputStream();
6d44bf31 385
47fc03d2 386 wxCHECK_MSG( inStream, 0, _T("should have a stream in wxStreamBuffer") );
7f42cff1 387
49e399d8
VZ
388 // lasterror is reset before all new IO calls
389 m_stream->m_lasterror = wxStream_NOERROR;
390
391 if ( !HasBuffer() )
392 {
47fc03d2 393 m_stream->m_lastcount = inStream->OnSysRead(buffer, size);
49e399d8
VZ
394 }
395 else // we have a buffer, use it
396 {
397 size_t orig_size = size;
398
399 while ( size > 0 )
400 {
401 size_t left = GetDataLeft();
402
403 // if the requested number of bytes if greater than the buffer
404 // size, read data in chunks
405 if ( size > left )
406 {
407 GetFromBuffer(buffer, left);
408 size -= left;
409 buffer = (char *)buffer + left;
410
411 if ( !FillBuffer() )
412 {
413 SetError(wxStream_EOF);
414 break;
415 }
416 }
417 else // otherwise just do it in one gulp
418 {
419 GetFromBuffer(buffer, size);
420 size = 0;
421 }
422 }
423
424 m_stream->m_lastcount = orig_size - size;
6d44bf31 425 }
49e399d8
VZ
426
427 return m_stream->m_lastcount;
6d44bf31
GL
428}
429
49e399d8
VZ
430// this should really be called "Copy()"
431size_t wxStreamBuffer::Read(wxStreamBuffer *dbuf)
8ef6a930 432{
49e399d8 433 wxCHECK_MSG( m_mode != write, 0, _T("can't read from this buffer") );
8ef6a930 434
49e399d8
VZ
435 char buf[BUF_TEMP_SIZE];
436 size_t nRead,
437 total = 0;
7f42cff1 438
49e399d8
VZ
439 do
440 {
441 nRead = Read(dbuf, WXSIZEOF(buf));
442 if ( nRead )
443 {
444 nRead = dbuf->Write(buf, nRead);
445 total += nRead;
446 }
447 }
448 while ( nRead );
449
450 return total;
8ef6a930
GL
451}
452
453size_t wxStreamBuffer::Write(const void *buffer, size_t size)
6d44bf31 454{
47fc03d2
VZ
455 wxOutputStream *outStream = GetOutputStream();
456
457 wxCHECK_MSG( outStream, 0, _T("should have a stream in wxStreamBuffer") );
6d44bf31 458
49e399d8
VZ
459 // lasterror is reset before all new IO calls
460 m_stream->m_lasterror = wxStream_NOERROR;
6d44bf31 461
49e399d8
VZ
462 if ( !HasBuffer() && m_fixed )
463 {
464 // no buffer, just forward the call to the stream
47fc03d2 465 m_stream->m_lastcount = outStream->OnSysWrite(buffer, size);
49e399d8
VZ
466 }
467 else // we [may] have a buffer, use it
468 {
469 size_t orig_size = size;
470
471 while ( size > 0 )
472 {
473 size_t left = GetBytesLeft();
474
475 // if the buffer is too large to fit in the stream buffer, split
476 // it in smaller parts
477 //
478 // NB: If stream buffer isn't fixed (as for wxMemoryOutputStream),
479 // we always go to the second case.
480 //
481 // FIXME: fine, but if it fails we should (re)try writing it by
482 // chunks as this will (hopefully) always work (VZ)
483 if ( size > left && m_fixed )
484 {
485 PutToBuffer(buffer, left);
486 size -= left;
487 buffer = (char *)buffer + left;
488
489 if ( !FlushBuffer() )
490 {
491 SetError(wxStream_WRITE_ERR);
492
493 break;
494 }
495
496 m_buffer_pos = m_buffer_start;
497 }
498 else // we can do it in one gulp
499 {
500 PutToBuffer(buffer, size);
501 size = 0;
502 }
503 }
504
505 m_stream->m_lastcount = orig_size - size;
6d44bf31 506 }
49e399d8
VZ
507
508 return m_stream->m_lastcount;
8ef6a930
GL
509}
510
511size_t wxStreamBuffer::Write(wxStreamBuffer *sbuf)
512{
49e399d8
VZ
513 wxCHECK_MSG( m_mode != read, 0, _T("can't write to this buffer") );
514 wxCHECK_MSG( sbuf->m_mode != write, 0, _T("can't read from that buffer") );
8ef6a930 515
49e399d8
VZ
516 char buf[BUF_TEMP_SIZE];
517 size_t nWrite,
518 total = 0;
7f42cff1 519
49e399d8
VZ
520 do
521 {
522 size_t nRead = sbuf->Read(buf, WXSIZEOF(buf));
523 if ( nRead )
524 {
525 nWrite = Write(buf, nRead);
526 if ( nWrite < nRead )
527 {
528 // put back data we couldn't copy
529 wxInputStream *in_stream = (wxInputStream *)sbuf->GetStream();
530
531 in_stream->Ungetch(buf + nWrite, nRead - nWrite);
532 }
533
534 total += nWrite;
535 }
536 else
537 {
538 nWrite = 0;
539 }
540 }
541 while ( nWrite == WXSIZEOF(buf) );
fae05df5 542
49e399d8 543 return total;
75ed1d15
GL
544}
545
546off_t wxStreamBuffer::Seek(off_t pos, wxSeekMode mode)
547{
49e399d8 548 off_t ret_off, diff;
75ed1d15 549
49e399d8 550 off_t last_access = GetLastAccess();
75ed1d15 551
49e399d8
VZ
552 if ( !m_flushable )
553 {
554 switch (mode)
555 {
556 case wxFromStart:
557 diff = pos;
558 break;
559
560 case wxFromCurrent:
561 diff = pos + GetIntPosition();
562 break;
563
564 case wxFromEnd:
565 diff = pos + last_access;
566 break;
567
568 default:
569 wxFAIL_MSG( _T("invalid seek mode") );
570
571 return wxInvalidOffset;
572 }
573 if (diff < 0 || diff > last_access)
574 return wxInvalidOffset;
575 SetIntPosition(diff);
576 return diff;
d984207c 577 }
49e399d8
VZ
578
579 switch ( mode )
580 {
581 case wxFromStart:
582 // We'll try to compute an internal position later ...
583 ret_off = m_stream->OnSysSeek(pos, wxFromStart);
584 ResetBuffer();
585 return ret_off;
586
587 case wxFromCurrent:
588 diff = pos + GetIntPosition();
589
590 if ( (diff > last_access) || (diff < 0) )
591 {
592 // We must take into account the fact that we have read
593 // something previously.
594 ret_off = m_stream->OnSysSeek(diff-last_access, wxFromCurrent);
595 ResetBuffer();
596 return ret_off;
597 }
598 else
599 {
600 SetIntPosition(diff);
601 return pos;
602 }
603
604 case wxFromEnd:
605 // Hard to compute: always seek to the requested position.
606 ret_off = m_stream->OnSysSeek(pos, wxFromEnd);
607 ResetBuffer();
608 return ret_off;
75ed1d15 609 }
49e399d8
VZ
610
611 return wxInvalidOffset;
75ed1d15
GL
612}
613
614off_t wxStreamBuffer::Tell() const
615{
c7a9fa36 616 off_t pos = m_stream->OnSysTell();
49e399d8 617 if ( pos == wxInvalidOffset )
c7a9fa36
RR
618 return wxInvalidOffset;
619
620 pos += GetIntPosition();
cd6ce4a9 621
49e399d8 622 if ( m_mode == read && m_flushable )
c7a9fa36 623 pos -= GetLastAccess();
cd6ce4a9 624
c7a9fa36 625 return pos;
75ed1d15
GL
626}
627
75ed1d15
GL
628// ----------------------------------------------------------------------------
629// wxStreamBase
630// ----------------------------------------------------------------------------
631
632wxStreamBase::wxStreamBase()
633{
c7a9fa36
RR
634 m_lasterror = wxStream_NOERROR;
635 m_lastcount = 0;
75ed1d15
GL
636}
637
638wxStreamBase::~wxStreamBase()
639{
640}
641
b9138710
VZ
642off_t wxStreamBase::OnSysSeek(off_t WXUNUSED(seek), wxSeekMode WXUNUSED(mode))
643{
644 return wxInvalidOffset;
645}
646
647off_t wxStreamBase::OnSysTell() const
648{
649 return wxInvalidOffset;
650}
651
1678ad78
GL
652// ----------------------------------------------------------------------------
653// wxInputStream
654// ----------------------------------------------------------------------------
655
3d4c6a21 656wxInputStream::wxInputStream()
3d4c6a21 657{
49e399d8
VZ
658 m_wback = NULL;
659 m_wbacksize =
660 m_wbackcur = 0;
6d44bf31
GL
661}
662
fae05df5 663wxInputStream::~wxInputStream()
6d44bf31 664{
fae05df5 665 free(m_wback);
3d4c6a21
GL
666}
667
47fc03d2
VZ
668size_t wxInputStream::OnSysRead(void * WXUNUSED(buffer),
669 size_t WXUNUSED(bufsize))
670{
671 return 0;
672}
673
cd6ce4a9
VZ
674bool wxInputStream::Eof() const
675{
49e399d8 676 wxInputStream *self = wxConstCast(this, wxInputStream);
cd6ce4a9
VZ
677
678 char c;
679 self->Read(&c, 1);
4708a349
VZ
680
681 // some streams can know that they're at EOF before actually trying to
682 // read beyond the end of stream (e.g. files) while others have no way of
683 // knowing it, so to provide the same behaviour in all cases we only
684 // return TRUE from here if the character really couldn't be read
685 if ( !self->LastRead() && GetLastError() == wxSTREAM_EOF )
cd6ce4a9
VZ
686 {
687 return TRUE;
688 }
689
690 self->Ungetch(c);
691
692 return FALSE;
693}
694
fae05df5 695char *wxInputStream::AllocSpaceWBack(size_t needed_size)
3d4c6a21 696{
49e399d8 697 // get number of bytes left from previous wback buffer
c7a9fa36 698 size_t toget = m_wbacksize - m_wbackcur;
fae05df5 699
49e399d8
VZ
700 // allocate a buffer large enough to hold prev + new data
701 char *temp_b = (char *)malloc(needed_size + toget);
fae05df5 702
c7a9fa36
RR
703 if (!temp_b)
704 return NULL;
fae05df5 705
adc35078 706 // copy previous data (and free old buffer) if needed
c7a9fa36
RR
707 if (m_wback)
708 {
709 memmove(temp_b + needed_size, m_wback + m_wbackcur, toget);
710 free(m_wback);
711 }
5ac8158a 712
adc35078 713 // done
c7a9fa36
RR
714 m_wback = temp_b;
715 m_wbackcur = 0;
716 m_wbacksize = needed_size + toget;
783ff666 717
49e399d8 718 return m_wback;
6d44bf31
GL
719}
720
49e399d8 721size_t wxInputStream::GetWBack(void *buf, size_t bsize)
6d44bf31 722{
c7a9fa36
RR
723 if (!m_wback)
724 return 0;
a324a7bc 725
e9f69291
VZ
726 // how many bytes do we have in the buffer?
727 size_t toget = m_wbacksize - m_wbackcur;
728
729 if ( bsize < toget )
730 {
731 // we won't read everything
732 toget = bsize;
733 }
fae05df5 734
e9f69291
VZ
735 // copy the data from the cache
736 memcpy(buf, m_wback + m_wbackcur, toget);
fae05df5 737
49e399d8 738 m_wbackcur += toget;
e9f69291 739 if ( m_wbackcur == m_wbacksize )
c7a9fa36 740 {
e9f69291 741 // TODO: should we really free it here all the time? maybe keep it?
c7a9fa36 742 free(m_wback);
49e399d8 743 m_wback = NULL;
c7a9fa36
RR
744 m_wbacksize = 0;
745 m_wbackcur = 0;
746 }
cd6ce4a9 747
e9f69291 748 // return the number of bytes copied
49e399d8 749 return toget;
6d44bf31
GL
750}
751
8f7173ab 752size_t wxInputStream::Ungetch(const void *buf, size_t bufsize)
fae05df5 753{
c7a9fa36
RR
754 char *ptrback = AllocSpaceWBack(bufsize);
755 if (!ptrback)
756 return 0;
cd6ce4a9 757
c7a9fa36
RR
758 memcpy(ptrback, buf, bufsize);
759 return bufsize;
fae05df5
GL
760}
761
762bool wxInputStream::Ungetch(char c)
1e3eca9d 763{
49e399d8 764 void *ptrback = AllocSpaceWBack(1);
c7a9fa36
RR
765 if (!ptrback)
766 return FALSE;
1e3eca9d 767
49e399d8 768 *(char *)ptrback = c;
c7a9fa36 769 return TRUE;
fae05df5
GL
770}
771
772char wxInputStream::GetC()
773{
c7a9fa36
RR
774 char c;
775 Read(&c, 1);
776 return c;
1e3eca9d
GL
777}
778
49e399d8 779wxInputStream& wxInputStream::Read(void *buf, size_t size)
6d44bf31 780{
c7a9fa36 781 size_t retsize = GetWBack(buf, size);
cd6ce4a9 782 if (retsize == size)
c7a9fa36
RR
783 {
784 m_lastcount = size;
785 m_lasterror = wxStream_NOERROR;
786 return *this;
787 }
788 size -= retsize;
49e399d8 789 buf = (char *)buf + retsize;
fae05df5 790
c7a9fa36
RR
791 m_lastcount = OnSysRead(buf, size) + retsize;
792 return *this;
3d4c6a21
GL
793}
794
75ed1d15
GL
795char wxInputStream::Peek()
796{
c7a9fa36
RR
797 char c;
798 Read(&c, 1);
cd6ce4a9 799 if (m_lasterror == wxStream_NOERROR)
c7a9fa36
RR
800 {
801 Ungetch(c);
802 return c;
803 }
cd6ce4a9 804
c7a9fa36 805 return 0;
75ed1d15
GL
806}
807
3d4c6a21
GL
808wxInputStream& wxInputStream::Read(wxOutputStream& stream_out)
809{
cd6ce4a9 810 char buf[BUF_TEMP_SIZE];
c7a9fa36 811 size_t bytes_read = BUF_TEMP_SIZE;
3d4c6a21 812
cd6ce4a9 813 while (bytes_read == BUF_TEMP_SIZE)
c7a9fa36
RR
814 {
815 bytes_read = Read(buf, bytes_read).LastRead();
816 bytes_read = stream_out.Write(buf, bytes_read).LastWrite();
817 }
818 return *this;
3d4c6a21
GL
819}
820
75ed1d15
GL
821off_t wxInputStream::SeekI(off_t pos, wxSeekMode mode)
822{
adc35078
RR
823 // RR: This code is duplicated in wxBufferedInputStream. This is
824 // not really a good design, but buffered stream are different
825 // from all other in that they handle two stream-related objects,
826 // the stream buffer and parent stream.
827
828 // I don't know whether it should be put as well in wxFileInputStream::OnSysSeek
cd6ce4a9 829 if (m_lasterror==wxSTREAM_EOF)
c7a9fa36
RR
830 m_lasterror=wxSTREAM_NOERROR;
831
adc35078
RR
832 /* RR: A call to SeekI() will automatically invalidate any previous
833 call to Ungetch(), otherwise it would be possible to SeekI() to
c7a9fa36 834 one position, unread some bytes there, SeekI() to another position
19da7237
GRG
835 and the data would be corrupted.
836
837 GRG: Could add code here to try to navigate within the wback
838 buffer if possible, but is it really needed? It would only work
839 when seeking in wxFromCurrent mode, else it would invalidate
adc35078
RR
840 anyway... */
841
cd6ce4a9 842 if (m_wback)
c7a9fa36 843 {
adc35078
RR
844 wxLogDebug( wxT("Seeking in stream which has data written back to it.") );
845
c7a9fa36 846 free(m_wback);
49e399d8 847 m_wback = NULL;
c7a9fa36
RR
848 m_wbacksize = 0;
849 m_wbackcur = 0;
850 }
fe8aa971 851
c7a9fa36 852 return OnSysSeek(pos, mode);
75ed1d15
GL
853}
854
855off_t wxInputStream::TellI() const
856{
19da7237
GRG
857 off_t pos = OnSysTell();
858
859 if (pos != wxInvalidOffset)
860 pos -= (m_wbacksize - m_wbackcur);
861
862 return pos;
75ed1d15
GL
863}
864
1678ad78 865
fae05df5
GL
866// ----------------------------------------------------------------------------
867// wxOutputStream
868// ----------------------------------------------------------------------------
49e399d8 869
fae05df5 870wxOutputStream::wxOutputStream()
1678ad78 871{
1678ad78
GL
872}
873
fae05df5 874wxOutputStream::~wxOutputStream()
123a7fdd 875{
123a7fdd
GL
876}
877
47fc03d2
VZ
878size_t wxOutputStream::OnSysWrite(const void * WXUNUSED(buffer),
879 size_t WXUNUSED(bufsize))
880{
881 return 0;
882}
883
7513f9ff
GRG
884void wxOutputStream::PutC(char c)
885{
49e399d8 886 Write(&c, 1);
7513f9ff
GRG
887}
888
fae05df5 889wxOutputStream& wxOutputStream::Write(const void *buffer, size_t size)
1678ad78 890{
c7a9fa36
RR
891 m_lastcount = OnSysWrite(buffer, size);
892 return *this;
1678ad78
GL
893}
894
fae05df5 895wxOutputStream& wxOutputStream::Write(wxInputStream& stream_in)
38830220 896{
c7a9fa36
RR
897 stream_in.Read(*this);
898 return *this;
38830220
RR
899}
900
fae05df5 901off_t wxOutputStream::TellO() const
38830220 902{
c7a9fa36 903 return OnSysTell();
38830220
RR
904}
905
fae05df5 906off_t wxOutputStream::SeekO(off_t pos, wxSeekMode mode)
38830220 907{
c7a9fa36 908 return OnSysSeek(pos, mode);
38830220
RR
909}
910
fae05df5 911void wxOutputStream::Sync()
1678ad78 912{
1678ad78
GL
913}
914
123a7fdd 915
e2acb9ae
RR
916// ----------------------------------------------------------------------------
917// wxCountingOutputStream
918// ----------------------------------------------------------------------------
919
920wxCountingOutputStream::wxCountingOutputStream ()
e2acb9ae 921{
c7a9fa36 922 m_currentPos = 0;
e2acb9ae
RR
923}
924
925size_t wxCountingOutputStream::GetSize() const
926{
c7a9fa36 927 return m_lastcount;
e2acb9ae
RR
928}
929
49e399d8
VZ
930size_t wxCountingOutputStream::OnSysWrite(const void *WXUNUSED(buffer),
931 size_t size)
e2acb9ae 932{
c7a9fa36 933 m_currentPos += size;
49e399d8
VZ
934 if (m_currentPos > m_lastcount)
935 m_lastcount = m_currentPos;
936
c7a9fa36 937 return m_currentPos;
e2acb9ae
RR
938}
939
940off_t wxCountingOutputStream::OnSysSeek(off_t pos, wxSeekMode mode)
941{
49e399d8
VZ
942 switch ( mode )
943 {
944 case wxFromStart:
945 m_currentPos = pos;
946 break;
947
948 case wxFromEnd:
949 m_currentPos = m_lastcount + pos;
950 break;
951
952 case wxFromCurrent:
953 m_currentPos += pos;
954 break;
955
956 default:
957 wxFAIL_MSG( _T("invalid seek mode") );
958 return wxInvalidOffset;
959 }
cd6ce4a9 960
49e399d8
VZ
961 if (m_currentPos > m_lastcount)
962 m_lastcount = m_currentPos;
cd6ce4a9 963
c7a9fa36 964 return m_currentPos;
e2acb9ae
RR
965}
966
967off_t wxCountingOutputStream::OnSysTell() const
968{
c7a9fa36 969 return m_currentPos;
e2acb9ae 970}
cd6ce4a9 971
1678ad78 972// ----------------------------------------------------------------------------
fae05df5 973// wxFilterInputStream
1678ad78 974// ----------------------------------------------------------------------------
e2acb9ae 975
fae05df5 976wxFilterInputStream::wxFilterInputStream()
3d4c6a21 977{
47fc03d2 978 m_parent_i_stream = NULL;
6d44bf31
GL
979}
980
fae05df5 981wxFilterInputStream::wxFilterInputStream(wxInputStream& stream)
6d44bf31 982{
c7a9fa36 983 m_parent_i_stream = &stream;
3d4c6a21
GL
984}
985
fae05df5 986wxFilterInputStream::~wxFilterInputStream()
3d4c6a21 987{
6d44bf31
GL
988}
989
fae05df5
GL
990// ----------------------------------------------------------------------------
991// wxFilterOutputStream
992// ----------------------------------------------------------------------------
49e399d8 993
fae05df5 994wxFilterOutputStream::wxFilterOutputStream()
6d44bf31 995{
47fc03d2 996 m_parent_o_stream = NULL;
3d4c6a21
GL
997}
998
fae05df5 999wxFilterOutputStream::wxFilterOutputStream(wxOutputStream& stream)
3d4c6a21 1000{
c7a9fa36 1001 m_parent_o_stream = &stream;
6d44bf31
GL
1002}
1003
fae05df5 1004wxFilterOutputStream::~wxFilterOutputStream()
6d44bf31 1005{
6d44bf31
GL
1006}
1007
fae05df5
GL
1008// ----------------------------------------------------------------------------
1009// wxBufferedInputStream
1010// ----------------------------------------------------------------------------
49e399d8 1011
47fc03d2
VZ
1012wxBufferedInputStream::wxBufferedInputStream(wxInputStream& s,
1013 wxStreamBuffer *buffer)
49e399d8 1014 : wxFilterInputStream(s)
6d44bf31 1015{
47fc03d2
VZ
1016 if ( buffer )
1017 {
1018 // use the buffer provided by the user
1019 m_i_streambuf = buffer;
1020 }
1021 else // create a default buffer
1022 {
1023 m_i_streambuf = new wxStreamBuffer(*this, wxStreamBuffer::read);
49e399d8 1024
47fc03d2
VZ
1025 m_i_streambuf->SetBufferIO(1024);
1026 }
6d44bf31
GL
1027}
1028
fae05df5 1029wxBufferedInputStream::~wxBufferedInputStream()
6d44bf31 1030{
1fb45475
VZ
1031 m_parent_i_stream->SeekI(-(off_t)m_i_streambuf->GetBytesLeft(),
1032 wxFromCurrent);
672cedf8 1033
c7a9fa36 1034 delete m_i_streambuf;
6d44bf31
GL
1035}
1036
6319afe3
GL
1037char wxBufferedInputStream::Peek()
1038{
c7a9fa36 1039 return m_i_streambuf->Peek();
6319afe3
GL
1040}
1041
49e399d8 1042wxInputStream& wxBufferedInputStream::Read(void *buf, size_t size)
1e3eca9d 1043{
e9f69291
VZ
1044 // reset the error flag
1045 m_lasterror = wxStream_NOERROR;
1e3eca9d 1046
e9f69291
VZ
1047 // first read from the already cached data
1048 m_lastcount = GetWBack(buf, size);
1049
1050 // do we have to read anything more?
1051 if ( m_lastcount < size )
c7a9fa36 1052 {
e9f69291
VZ
1053 size -= m_lastcount;
1054 buf = (char *)buf + m_lastcount;
1e3eca9d 1055
e9f69291
VZ
1056 // the call to wxStreamBuffer::Read() below will reset our m_lastcount,
1057 // so save it
1058 size_t countOld = m_lastcount;
1059
1060 m_i_streambuf->Read(buf, size);
1061
1062 m_lastcount += countOld;
1063 }
6d44bf31 1064
c7a9fa36 1065 return *this;
6d44bf31
GL
1066}
1067
fae05df5 1068off_t wxBufferedInputStream::SeekI(off_t pos, wxSeekMode mode)
6d44bf31 1069{
adc35078
RR
1070 // RR: Look at wxInputStream for comments.
1071
1072 if (m_lasterror==wxSTREAM_EOF)
1073 m_lasterror=wxSTREAM_NOERROR;
1074
1075 if (m_wback)
1076 {
1077 wxLogDebug( wxT("Seeking in stream which has data written back to it.") );
1078
1079 free(m_wback);
1080 m_wback = NULL;
1081 m_wbacksize = 0;
1082 m_wbackcur = 0;
1083 }
1084
c7a9fa36 1085 return m_i_streambuf->Seek(pos, mode);
6d44bf31
GL
1086}
1087
fae05df5 1088off_t wxBufferedInputStream::TellI() const
6d44bf31 1089{
adc35078
RR
1090 off_t pos = m_i_streambuf->Tell();
1091
1092 if (pos != wxInvalidOffset)
1093 pos -= (m_wbacksize - m_wbackcur);
1094
1095 return pos;
38830220 1096}
6d44bf31 1097
fae05df5 1098size_t wxBufferedInputStream::OnSysRead(void *buffer, size_t bufsize)
38830220 1099{
c7a9fa36 1100 return m_parent_i_stream->Read(buffer, bufsize).LastRead();
38830220
RR
1101}
1102
fae05df5 1103off_t wxBufferedInputStream::OnSysSeek(off_t seek, wxSeekMode mode)
38830220 1104{
c7a9fa36 1105 return m_parent_i_stream->SeekI(seek, mode);
6d44bf31
GL
1106}
1107
fae05df5 1108off_t wxBufferedInputStream::OnSysTell() const
6d44bf31 1109{
c7a9fa36 1110 return m_parent_i_stream->TellI();
38830220 1111}
6d44bf31 1112
47fc03d2
VZ
1113void wxBufferedInputStream::SetInputStreamBuffer(wxStreamBuffer *buffer)
1114{
1115 wxCHECK_RET( buffer, _T("wxBufferedInputStream needs buffer") );
1116
1117 delete m_i_streambuf;
1118 m_i_streambuf = buffer;
1119}
1120
fae05df5
GL
1121// ----------------------------------------------------------------------------
1122// wxBufferedOutputStream
1123// ----------------------------------------------------------------------------
6d44bf31 1124
47fc03d2
VZ
1125wxBufferedOutputStream::wxBufferedOutputStream(wxOutputStream& s,
1126 wxStreamBuffer *buffer)
49e399d8 1127 : wxFilterOutputStream(s)
6d44bf31 1128{
47fc03d2
VZ
1129 if ( buffer )
1130 {
1131 m_o_streambuf = buffer;
1132 }
1133 else // create a default one
1134 {
1135 m_o_streambuf = new wxStreamBuffer(*this, wxStreamBuffer::write);
1136
1137 m_o_streambuf->SetBufferIO(1024);
1138 }
6d44bf31
GL
1139}
1140
fae05df5 1141wxBufferedOutputStream::~wxBufferedOutputStream()
6d44bf31 1142{
c7a9fa36
RR
1143 Sync();
1144 delete m_o_streambuf;
3d4c6a21
GL
1145}
1146
fae05df5 1147wxOutputStream& wxBufferedOutputStream::Write(const void *buffer, size_t size)
123a7fdd 1148{
c7a9fa36
RR
1149 m_lastcount = 0;
1150 m_o_streambuf->Write(buffer, size);
1151 return *this;
123a7fdd
GL
1152}
1153
fae05df5 1154off_t wxBufferedOutputStream::SeekO(off_t pos, wxSeekMode mode)
f4ada568 1155{
c7a9fa36
RR
1156 Sync();
1157 return m_o_streambuf->Seek(pos, mode);
f4ada568
GL
1158}
1159
fae05df5 1160off_t wxBufferedOutputStream::TellO() const
3d4c6a21 1161{
c7a9fa36 1162 return m_o_streambuf->Tell();
3d4c6a21
GL
1163}
1164
fae05df5 1165void wxBufferedOutputStream::Sync()
3d4c6a21 1166{
c7a9fa36
RR
1167 m_o_streambuf->FlushBuffer();
1168 m_parent_o_stream->Sync();
3d4c6a21 1169}
219f895a 1170
fae05df5 1171size_t wxBufferedOutputStream::OnSysWrite(const void *buffer, size_t bufsize)
f4ada568 1172{
c7a9fa36 1173 return m_parent_o_stream->Write(buffer, bufsize).LastWrite();
f4ada568
GL
1174}
1175
fae05df5 1176off_t wxBufferedOutputStream::OnSysSeek(off_t seek, wxSeekMode mode)
219f895a 1177{
c7a9fa36 1178 return m_parent_o_stream->SeekO(seek, mode);
219f895a
RR
1179}
1180
fae05df5 1181off_t wxBufferedOutputStream::OnSysTell() const
219f895a 1182{
c7a9fa36
RR
1183 return m_parent_o_stream->TellO();
1184}
1185
1186size_t wxBufferedOutputStream::GetSize() const
1187{
1188 return m_parent_o_stream->GetSize() + m_o_streambuf->GetIntPosition();
219f895a 1189}
6d44bf31 1190
47fc03d2
VZ
1191void wxBufferedOutputStream::SetOutputStreamBuffer(wxStreamBuffer *buffer)
1192{
1193 wxCHECK_RET( buffer, _T("wxBufferedOutputStream needs buffer") );
1194
1195 delete m_o_streambuf;
1196 m_o_streambuf = buffer;
1197}
1198
6d44bf31
GL
1199// ----------------------------------------------------------------------------
1200// Some IOManip function
1201// ----------------------------------------------------------------------------
1202
1203wxOutputStream& wxEndL(wxOutputStream& stream)
1204{
733b8ed3
VZ
1205 static const wxChar *eol = wxTextFile::GetEOL();
1206
1207 return stream.Write(eol, wxStrlen(eol));
6d44bf31 1208}
ce4169a4
RR
1209
1210#endif
1211 // wxUSE_STREAMS