home *** CD-ROM | disk | FTP | other *** search
- // This is part of the iostream library, providing input/output for C++.
- // Copyright (C) 1991 Per Bothner.
- //
- // This library is free software; you can redistribute it and/or
- // modify it under the terms of the GNU Library General Public
- // License as published by the Free Software Foundation; either
- // version 2 of the License, or (at your option) any later version.
- //
- // This library is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- // Library General Public License for more details.
- //
- // You should have received a copy of the GNU Library General Public
- // License along with this library; if not, write to the Free
- // Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
-
- #ifdef __GNUG__
- #pragma implementation
- #endif
- #include "ioprivat.h"
- #include <strstrea.h>
-
- istrstream::istrstream(char *cp) : istream(NULL)
- {
- _strbuf = new strstreambuf(cp, 0, NULL);
- }
-
- istrstream::istrstream(char *cp, int n) : istream(NULL)
- {
- _strbuf = new strstreambuf(cp, n, NULL);
- }
-
- ostrstream::ostrstream() : ostream(NULL)
- {
- _strbuf = new strstreambuf;
- }
-
- ostrstream::ostrstream(char *cp, int n, int mode) : ostream(NULL)
- {
- char *pstart;
- if (mode == ios::app || mode == ios::ate)
- pstart = cp + strlen(cp);
- else
- pstart = cp;
- _strbuf = new strstreambuf(cp, n, pstart);
- }
-
-
- strstream::strstream() : iostream(NULL)
- {
- _strbuf = new strstreambuf;
- }
-
- strstream::strstream(char *cp, int n, int mode) : iostream(NULL)
- {
- char *pstart;
- if (mode == ios::app || mode == ios::ate)
- pstart = cp + strlen(cp);
- else
- pstart = cp;
- _strbuf = new strstreambuf(cp, n, pstart);
- }
-
-
- char *strstreambuf::str()
- {
- freeze(1);
- return _buffer;
- }
-
- _G_size_t strstreambuf::pcount()
- {
- _G_size_t put_len = pptr() - pbase();
- if (put_len < _len) put_len = _len;
- return put_len;
- }
-
- int strstreambuf::overflow(int c = EOF)
- {
- const int flush_only = c == EOF;
- size_t pos = pptr() - pbase();
- size_t get_pos = gptr() - pbase();
- if (pos > _len) _len = pos;
- if (pos >= _size + flush_only) {
- char *new_buf;
- size_t new_size = 2 * _size;
- if (_frozen) /* not allowed to enlarge */
- return EOF;
- else {
- new_buf = ALLOC_BUF(new_size);
- memcpy(new_buf, _buffer, _size);
- }
- if (new_buf == NULL) {
- // __ferror(fp) = 1;
- return EOF;
- }
- _size = new_size;
- #if 0
- if (lenp == &_len) /* use '\0'-filling */
- memset(new_buf + pos, 0, _size - pos);
- #endif
- _buffer = new_buf;
- setb(new_buf, new_buf + _size, 1);
- }
-
- setp(_buffer, _buffer + _size);
- pbump(pos);
- setg(_buffer, _buffer + get_pos, _buffer + _len);
- if (!flush_only) {
- *pptr() = (unsigned char) c;
- pbump(1);
- }
- return c;
- }
-
- int strstreambuf::underflow()
- {
- size_t ppos = pptr() - pbase();
- if (ppos > _len) _len = ppos;
- setg(_buffer, gptr(), _buffer + _len);
- if (gptr() < egptr())
- return *gptr();
- else
- return EOF;
- }
-
- strstreambuf::strstreambuf()
- {
- _frozen = 0;
- // _flags &= ~ios::dont_close;
- _len = 0;
- _size = 128;
- _buffer = ALLOC_BUF(_size);
- setb(_buffer, _buffer+_size);
- setp(_buffer, _buffer+_size);
- setg(_buffer, _buffer, _buffer);
- xsetflags(_S_CAN_READ | _S_CAN_WRITE);
- }
-
- strstreambuf::strstreambuf(int initial)
- {
- _frozen = 0;
- // _flags &= ~ios::dont_close;
- _len = 0;
- if (initial < 16)
- initial = 16;
- _size = initial;
- _buffer = ALLOC_BUF(_size);
- setb(_buffer, _buffer+_size);
- setp(_buffer, _buffer+_size);
- setg(_buffer, _buffer, _buffer);
- xsetflags(_S_CAN_READ | _S_CAN_WRITE);
- }
-
- strstreambuf::strstreambuf(char *ptr, size_t size, char *pstart)
- {
- // _flags &= ~ios::dont_close;
- _frozen = 1;
- _len = 0;
- if (size == 0)
- _size = strlen(ptr);
- else if (size < 0) {
- // If size is negative 'the characters are assumed to
- // continue indefinitely.'
- // The following semi-portable kludge tries to do that.
- // It assumes that sizeof(long) == sizeof(pointer).
- // Hence, (unsigned long)(-1) should be the largest
- // possible address.
- unsigned long highest = (unsigned long)(-1);
- // Pointers are signed on some brain-damaged systems, in
- // which case we divide by two to get the maximum signed address.
- if ((char*)highest < ptr)
- highest >>= 1;
- _size = (char*)highest - ptr;
- }
- else
- _size = size;
- _buffer = ptr;
- int new_flags = _S_CAN_READ;
- setb(_buffer, _buffer+_size);
- if (pstart) {
- new_flags |= _S_CAN_WRITE;
- setp(_buffer, _buffer+_size);
- pbump(pstart-_buffer);
- setg(_buffer, _buffer, pstart);
- }
- else {
- setp(_buffer, _buffer);
- setg(_buffer, _buffer, _buffer+_size);
- }
- xsetflags(new_flags);
- }
-
- strstreambuf::~strstreambuf()
- {
- if (_frozen == 0)
- FREE_BUF(_buffer);
- }
-
- virtual streampos strstreambuf::seekoff(streamoff off, _seek_dir dir,
- int mode /*=ios::in|ios::out*/)
- {
- size_t cur_size = pcount();
- streampos new_pos = EOF;
- if ((mode & ios::in) && (xflags() & _S_CAN_READ)) {
- switch (dir) {
- case ios::end:
- off += cur_size;
- break;
- case ios::cur:
- off += gptr() - pbase();
- break;
- default: /*case ios::beg: */
- break;
- }
- if (off < 0 || (size_t)off > cur_size)
- return (streampos)(EOF);
- setg(_buffer, _buffer + off, _buffer + cur_size);
- new_pos = off;
- }
- if ((mode & ios::out) && (xflags() & _S_CAN_WRITE)) {
- switch (dir) {
- case ios::end:
- off += cur_size;
- break;
- case ios::cur:
- off += pptr() - pbase();
- break;
- default: /*case ios::beg: */
- break;
- }
- if (off < 0 || (size_t)off > cur_size)
- return (streampos)(EOF);
- pbump(_buffer + off - pptr());
- new_pos = off;
- }
- return new_pos;
- }
-