123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168 |
- #include <features.h>
- #ifdef __USE_GNU
- #include "_stdio.h"
- #ifndef __UCLIBC_HAS_GLIBC_CUSTOM_STREAMS__
- #error no custom streams!
- #endif
- #define COOKIE ((__oms_cookie *) cookie)
- #define MEMSTREAM_BUFSIZ 256
- typedef struct {
- char *buf;
- size_t len;
- size_t pos;
- size_t eof;
- char **bufloc;
- size_t *sizeloc;
- } __oms_cookie;
- static ssize_t oms_write(register void *cookie, const char *buf, size_t bufsize)
- {
- register char *newbuf;
- size_t count;
-
- count = COOKIE->len - COOKIE->pos - 1;
- assert(COOKIE->pos < COOKIE->len);
- if (bufsize > count) {
- newbuf = realloc(COOKIE->buf, COOKIE->len + bufsize - count);
- if (newbuf) {
- *COOKIE->bufloc = COOKIE->buf = newbuf;
- COOKIE->len += (bufsize - count);
- } else {
- bufsize = count;
- if (count == 0) {
- __set_errno(EFBIG);
- return -1;
- }
- }
- }
- memcpy(COOKIE->buf + COOKIE->pos, buf, bufsize);
- COOKIE->pos += bufsize;
- if (COOKIE->pos > COOKIE->eof) {
- *COOKIE->sizeloc = COOKIE->eof = COOKIE->pos;
- COOKIE->buf[COOKIE->eof] = 0;
- }
- return bufsize;
- }
- static int oms_seek(register void *cookie, __offmax_t *pos, int whence)
- {
- __offmax_t p = *pos;
- register char *buf;
- size_t leastlen;
-
- assert(((unsigned int) whence) <= 2);
- if (whence != SEEK_SET) {
- p += (whence == SEEK_CUR) ? COOKIE->pos : COOKIE->eof;
- }
-
-
- if (((uintmax_t) p) >= SIZE_MAX - 1) {
- return -1;
- }
- leastlen = ((size_t) p) + 1;
- if (leastlen >= COOKIE->len) {
- buf = realloc(COOKIE->buf, leastlen);
- if (buf) {
- *COOKIE->bufloc = COOKIE->buf = buf;
- COOKIE->len = leastlen;
- memset(buf + COOKIE->eof, 0, leastlen - COOKIE->eof);
- } else {
-
- return -1;
- }
- }
- *pos = COOKIE->pos = --leastlen;
- if (leastlen > COOKIE->eof) {
- memset(COOKIE->buf + COOKIE->eof, 0, leastlen - COOKIE->eof);
- *COOKIE->sizeloc = COOKIE->eof;
- }
- return 0;
- }
- static int oms_close(void *cookie)
- {
- free(cookie);
- return 0;
- }
- #undef COOKIE
- static const cookie_io_functions_t _oms_io_funcs = {
- NULL, oms_write, oms_seek, oms_close
- };
- FILE *open_memstream(char **bufloc, size_t *sizeloc)
- {
- register __oms_cookie *cookie;
- register FILE *fp;
- if ((cookie = malloc(sizeof(__oms_cookie))) != NULL) {
- if ((cookie->buf = malloc(cookie->len = MEMSTREAM_BUFSIZ)) == NULL) {
- goto EXIT_cookie;
- }
- *cookie->buf = 0;
- *(cookie->bufloc = bufloc) = cookie->buf;
- *(cookie->sizeloc = sizeloc) = cookie->eof = cookie->pos = 0;
- #ifndef __BCC__
- fp = fopencookie(cookie, "w", _oms_io_funcs);
- #else
- fp = fopencookie(cookie, "w", &_oms_io_funcs);
- #endif
-
- if (fp != NULL) {
- __STDIO_STREAM_VALIDATE(fp);
- return fp;
- }
- }
- free(cookie->buf);
- EXIT_cookie:
- free(cookie);
- return NULL;
- }
- libc_hidden_def(open_memstream)
- #endif
|