123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375 |
- /* Copyright (C) 1993-2019 Free Software Foundation, Inc.
- This file is part of the GNU C Library.
- The GNU C Library is free software; you can redistribute it and/or
- modify it under the terms of the GNU Lesser General Public
- License as published by the Free Software Foundation; either
- version 2.1 of the License, or (at your option) any later version.
- The GNU C 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
- Lesser General Public License for more details.
- You should have received a copy of the GNU Lesser General Public
- License along with the GNU C Library; if not, see
- <http://www.gnu.org/licenses/>.
- As a special exception, if you link the code in this file with
- files compiled with a GNU compiler to produce an executable,
- that does not cause the resulting executable to be covered by
- the GNU Lesser General Public License. This exception does not
- however invalidate any other reasons why the executable file
- might be covered by the GNU Lesser General Public License.
- This exception applies to code released by its copyright holders
- in files containing the exception. */
- #include <assert.h>
- #include "strfile.h"
- #include "libioP.h"
- #include <string.h>
- #include <stdio_ext.h>
- void
- _IO_str_init_static_internal (_IO_strfile *sf, char *ptr, size_t size,
- char *pstart)
- {
- FILE *fp = &sf->_sbf._f;
- char *end;
- if (size == 0)
- end = __rawmemchr (ptr, '\0');
- else if ((size_t) ptr + size > (size_t) ptr)
- end = ptr + size;
- else
- end = (char *) -1;
- _IO_setb (fp, ptr, end, 0);
- fp->_IO_write_base = ptr;
- fp->_IO_read_base = ptr;
- fp->_IO_read_ptr = ptr;
- if (pstart)
- {
- fp->_IO_write_ptr = pstart;
- fp->_IO_write_end = end;
- fp->_IO_read_end = pstart;
- }
- else
- {
- fp->_IO_write_ptr = ptr;
- fp->_IO_write_end = ptr;
- fp->_IO_read_end = end;
- }
- /* A null _allocate_buffer function flags the strfile as being static. */
- sf->_s._allocate_buffer_unused = (_IO_alloc_type) 0;
- }
- void
- _IO_str_init_static (_IO_strfile *sf, char *ptr, int size, char *pstart)
- {
- return _IO_str_init_static_internal (sf, ptr, size < 0 ? -1 : size, pstart);
- }
- void
- _IO_str_init_readonly (_IO_strfile *sf, const char *ptr, int size)
- {
- _IO_str_init_static_internal (sf, (char *) ptr, size < 0 ? -1 : size, NULL);
- sf->_sbf._f._flags |= _IO_NO_WRITES;
- }
- int
- _IO_str_overflow (FILE *fp, int c)
- {
- int flush_only = c == EOF;
- size_t pos;
- if (fp->_flags & _IO_NO_WRITES)
- return flush_only ? 0 : EOF;
- if ((fp->_flags & _IO_TIED_PUT_GET) && !(fp->_flags & _IO_CURRENTLY_PUTTING))
- {
- fp->_flags |= _IO_CURRENTLY_PUTTING;
- fp->_IO_write_ptr = fp->_IO_read_ptr;
- fp->_IO_read_ptr = fp->_IO_read_end;
- }
- pos = fp->_IO_write_ptr - fp->_IO_write_base;
- if (pos >= (size_t) (_IO_blen (fp) + flush_only))
- {
- if (fp->_flags & _IO_USER_BUF) /* not allowed to enlarge */
- return EOF;
- else
- {
- char *new_buf;
- char *old_buf = fp->_IO_buf_base;
- size_t old_blen = _IO_blen (fp);
- size_t new_size = 2 * old_blen + 100;
- if (new_size < old_blen)
- return EOF;
- new_buf = malloc (new_size);
- if (new_buf == NULL)
- {
- /* __ferror(fp) = 1; */
- return EOF;
- }
- if (old_buf)
- {
- memcpy (new_buf, old_buf, old_blen);
- free (old_buf);
- /* Make sure _IO_setb won't try to delete _IO_buf_base. */
- fp->_IO_buf_base = NULL;
- }
- memset (new_buf + old_blen, '\0', new_size - old_blen);
- _IO_setb (fp, new_buf, new_buf + new_size, 1);
- fp->_IO_read_base = new_buf + (fp->_IO_read_base - old_buf);
- fp->_IO_read_ptr = new_buf + (fp->_IO_read_ptr - old_buf);
- fp->_IO_read_end = new_buf + (fp->_IO_read_end - old_buf);
- fp->_IO_write_ptr = new_buf + (fp->_IO_write_ptr - old_buf);
- fp->_IO_write_base = new_buf;
- fp->_IO_write_end = fp->_IO_buf_end;
- }
- }
- if (!flush_only)
- *fp->_IO_write_ptr++ = (unsigned char) c;
- if (fp->_IO_write_ptr > fp->_IO_read_end)
- fp->_IO_read_end = fp->_IO_write_ptr;
- return c;
- }
- libc_hidden_def (_IO_str_overflow)
- int
- _IO_str_underflow (FILE *fp)
- {
- if (fp->_IO_write_ptr > fp->_IO_read_end)
- fp->_IO_read_end = fp->_IO_write_ptr;
- if ((fp->_flags & _IO_TIED_PUT_GET) && (fp->_flags & _IO_CURRENTLY_PUTTING))
- {
- fp->_flags &= ~_IO_CURRENTLY_PUTTING;
- fp->_IO_read_ptr = fp->_IO_write_ptr;
- fp->_IO_write_ptr = fp->_IO_write_end;
- }
- if (fp->_IO_read_ptr < fp->_IO_read_end)
- return *((unsigned char *) fp->_IO_read_ptr);
- else
- return EOF;
- }
- libc_hidden_def (_IO_str_underflow)
- /* The size of the valid part of the buffer. */
- ssize_t
- _IO_str_count (FILE *fp)
- {
- return ((fp->_IO_write_ptr > fp->_IO_read_end
- ? fp->_IO_write_ptr : fp->_IO_read_end)
- - fp->_IO_read_base);
- }
- static int
- enlarge_userbuf (FILE *fp, off64_t offset, int reading)
- {
- if ((ssize_t) offset <= _IO_blen (fp))
- return 0;
- ssize_t oldend = fp->_IO_write_end - fp->_IO_write_base;
- /* Try to enlarge the buffer. */
- if (fp->_flags & _IO_USER_BUF)
- /* User-provided buffer. */
- return 1;
- size_t newsize = offset + 100;
- char *oldbuf = fp->_IO_buf_base;
- char *newbuf = malloc (newsize);
- if (newbuf == NULL)
- return 1;
- if (oldbuf != NULL)
- {
- memcpy (newbuf, oldbuf, _IO_blen (fp));
- free (oldbuf);
- /* Make sure _IO_setb won't try to delete
- _IO_buf_base. */
- fp->_IO_buf_base = NULL;
- }
- _IO_setb (fp, newbuf, newbuf + newsize, 1);
- if (reading)
- {
- fp->_IO_write_base = newbuf + (fp->_IO_write_base - oldbuf);
- fp->_IO_write_ptr = newbuf + (fp->_IO_write_ptr - oldbuf);
- fp->_IO_write_end = newbuf + (fp->_IO_write_end - oldbuf);
- fp->_IO_read_ptr = newbuf + (fp->_IO_read_ptr - oldbuf);
- fp->_IO_read_base = newbuf;
- fp->_IO_read_end = fp->_IO_buf_end;
- }
- else
- {
- fp->_IO_read_base = newbuf + (fp->_IO_read_base - oldbuf);
- fp->_IO_read_ptr = newbuf + (fp->_IO_read_ptr - oldbuf);
- fp->_IO_read_end = newbuf + (fp->_IO_read_end - oldbuf);
- fp->_IO_write_ptr = newbuf + (fp->_IO_write_ptr - oldbuf);
- fp->_IO_write_base = newbuf;
- fp->_IO_write_end = fp->_IO_buf_end;
- }
- /* Clear the area between the last write position and th
- new position. */
- assert (offset >= oldend);
- if (reading)
- memset (fp->_IO_read_base + oldend, '\0', offset - oldend);
- else
- memset (fp->_IO_write_base + oldend, '\0', offset - oldend);
- return 0;
- }
- static void
- _IO_str_switch_to_get_mode (FILE *fp)
- {
- if (_IO_in_backup (fp))
- fp->_IO_read_base = fp->_IO_backup_base;
- else
- {
- fp->_IO_read_base = fp->_IO_buf_base;
- if (fp->_IO_write_ptr > fp->_IO_read_end)
- fp->_IO_read_end = fp->_IO_write_ptr;
- }
- fp->_IO_read_ptr = fp->_IO_read_end = fp->_IO_write_ptr;
- fp->_flags &= ~_IO_CURRENTLY_PUTTING;
- }
- off64_t
- _IO_str_seekoff (FILE *fp, off64_t offset, int dir, int mode)
- {
- off64_t new_pos;
- if (mode == 0 && (fp->_flags & _IO_TIED_PUT_GET))
- mode = (fp->_flags & _IO_CURRENTLY_PUTTING ? _IOS_OUTPUT : _IOS_INPUT);
- bool was_writing = (fp->_IO_write_ptr > fp->_IO_write_base
- || _IO_in_put_mode (fp));
- if (was_writing)
- _IO_str_switch_to_get_mode (fp);
- if (mode == 0)
- {
- new_pos = fp->_IO_read_ptr - fp->_IO_read_base;
- }
- else
- {
- ssize_t cur_size = _IO_str_count(fp);
- new_pos = EOF;
- /* Move the get pointer, if requested. */
- if (mode & _IOS_INPUT)
- {
- ssize_t base;
- switch (dir)
- {
- case _IO_seek_set:
- base = 0;
- break;
- case _IO_seek_cur:
- base = fp->_IO_read_ptr - fp->_IO_read_base;
- break;
- default: /* case _IO_seek_end: */
- base = cur_size;
- break;
- }
- ssize_t maxval = SSIZE_MAX - base;
- if (offset < -base || offset > maxval)
- {
- __set_errno (EINVAL);
- return EOF;
- }
- base += offset;
- if (base > cur_size
- && enlarge_userbuf (fp, base, 1) != 0)
- return EOF;
- fp->_IO_read_ptr = fp->_IO_read_base + base;
- fp->_IO_read_end = fp->_IO_read_base + cur_size;
- new_pos = base;
- }
- /* Move the put pointer, if requested. */
- if (mode & _IOS_OUTPUT)
- {
- ssize_t base;
- switch (dir)
- {
- case _IO_seek_set:
- base = 0;
- break;
- case _IO_seek_cur:
- base = fp->_IO_write_ptr - fp->_IO_write_base;
- break;
- default: /* case _IO_seek_end: */
- base = cur_size;
- break;
- }
- ssize_t maxval = SSIZE_MAX - base;
- if (offset < -base || offset > maxval)
- {
- __set_errno (EINVAL);
- return EOF;
- }
- base += offset;
- if (base > cur_size
- && enlarge_userbuf (fp, base, 0) != 0)
- return EOF;
- fp->_IO_write_ptr = fp->_IO_write_base + base;
- new_pos = base;
- }
- }
- return new_pos;
- }
- libc_hidden_def (_IO_str_seekoff)
- int
- _IO_str_pbackfail (FILE *fp, int c)
- {
- if ((fp->_flags & _IO_NO_WRITES) && c != EOF)
- return EOF;
- return _IO_default_pbackfail (fp, c);
- }
- libc_hidden_def (_IO_str_pbackfail)
- void
- _IO_str_finish (FILE *fp, int dummy)
- {
- if (fp->_IO_buf_base && !(fp->_flags & _IO_USER_BUF))
- free (fp->_IO_buf_base);
- fp->_IO_buf_base = NULL;
- _IO_default_finish (fp, 0);
- }
- const struct _IO_jump_t _IO_str_jumps libio_vtable =
- {
- JUMP_INIT_DUMMY,
- JUMP_INIT(finish, _IO_str_finish),
- JUMP_INIT(overflow, _IO_str_overflow),
- JUMP_INIT(underflow, _IO_str_underflow),
- JUMP_INIT(uflow, _IO_default_uflow),
- JUMP_INIT(pbackfail, _IO_str_pbackfail),
- JUMP_INIT(xsputn, _IO_default_xsputn),
- JUMP_INIT(xsgetn, _IO_default_xsgetn),
- JUMP_INIT(seekoff, _IO_str_seekoff),
- JUMP_INIT(seekpos, _IO_default_seekpos),
- JUMP_INIT(setbuf, _IO_default_setbuf),
- JUMP_INIT(sync, _IO_default_sync),
- JUMP_INIT(doallocate, _IO_default_doallocate),
- JUMP_INIT(read, _IO_default_read),
- JUMP_INIT(write, _IO_default_write),
- JUMP_INIT(seek, _IO_default_seek),
- JUMP_INIT(close, _IO_default_close),
- JUMP_INIT(stat, _IO_default_stat),
- JUMP_INIT(showmanyc, _IO_default_showmanyc),
- JUMP_INIT(imbue, _IO_default_imbue)
- };
|