281 lines
		
	
	
		
			8.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			281 lines
		
	
	
		
			8.0 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/* Copyright (C) 2007 Eric Blake
 | 
						|
 * Permission to use, copy, modify, and distribute this software
 | 
						|
 * is freely granted, provided that this notice is preserved.
 | 
						|
 *
 | 
						|
 * Modifications for Android written Jul 2009 by Alan Viverette
 | 
						|
 */
 | 
						|
 | 
						|
#if __ANDROID__
 | 
						|
 | 
						|
/*
 | 
						|
FUNCTION
 | 
						|
<fmemopen>>---open a stream around a fixed-length string
 | 
						|
 | 
						|
INDEX
 | 
						|
	fmemopen
 | 
						|
 | 
						|
ANSI_SYNOPSIS
 | 
						|
	#include <stdio.h>
 | 
						|
	FILE *fmemopen(void *restrict <[buf]>, size_t <[size]>,
 | 
						|
		       const char *restrict <[mode]>);
 | 
						|
 | 
						|
DESCRIPTION
 | 
						|
<<fmemopen>> creates a seekable <<FILE>> stream that wraps a
 | 
						|
fixed-length buffer of <[size]> bytes starting at <[buf]>.  The stream
 | 
						|
is opened with <[mode]> treated as in <<fopen>>, where append mode
 | 
						|
starts writing at the first NUL byte.  If <[buf]> is NULL, then
 | 
						|
<[size]> bytes are automatically provided as if by <<malloc>>, with
 | 
						|
the initial size of 0, and <[mode]> must contain <<+>> so that data
 | 
						|
can be read after it is written.
 | 
						|
 | 
						|
The stream maintains a current position, which moves according to
 | 
						|
bytes read or written, and which can be one past the end of the array.
 | 
						|
The stream also maintains a current file size, which is never greater
 | 
						|
than <[size]>.  If <[mode]> starts with <<r>>, the position starts at
 | 
						|
<<0>>, and file size starts at <[size]> if <[buf]> was provided.  If
 | 
						|
<[mode]> starts with <<w>>, the position and file size start at <<0>>,
 | 
						|
and if <[buf]> was provided, the first byte is set to NUL.  If
 | 
						|
<[mode]> starts with <<a>>, the position and file size start at the
 | 
						|
location of the first NUL byte, or else <[size]> if <[buf]> was
 | 
						|
provided.
 | 
						|
 | 
						|
When reading, NUL bytes have no significance, and reads cannot exceed
 | 
						|
the current file size.  When writing, the file size can increase up to
 | 
						|
<[size]> as needed, and NUL bytes may be embedded in the stream (see
 | 
						|
<<open_memstream>> for an alternative that automatically enlarges the
 | 
						|
buffer).  When the stream is flushed or closed after a write that
 | 
						|
changed the file size, a NUL byte is written at the current position
 | 
						|
if there is still room; if the stream is not also open for reading, a
 | 
						|
NUL byte is additionally written at the last byte of <[buf]> when the
 | 
						|
stream has exceeded <[size]>, so that a write-only <[buf]> is always
 | 
						|
NUL-terminated when the stream is flushed or closed (and the initial
 | 
						|
<[size]> should take this into account).  It is not possible to seek
 | 
						|
outside the bounds of <[size]>.  A NUL byte written during a flush is
 | 
						|
restored to its previous value when seeking elsewhere in the string.
 | 
						|
 | 
						|
RETURNS
 | 
						|
The return value is an open FILE pointer on success.  On error,
 | 
						|
<<NULL>> is returned, and <<errno>> will be set to EINVAL if <[size]>
 | 
						|
is zero or <[mode]> is invalid, ENOMEM if <[buf]> was NULL and memory
 | 
						|
could not be allocated, or EMFILE if too many streams are already
 | 
						|
open.
 | 
						|
 | 
						|
PORTABILITY
 | 
						|
This function is being added to POSIX 200x, but is not in POSIX 2001.
 | 
						|
 | 
						|
Supporting OS subroutines required: <<sbrk>>.
 | 
						|
*/
 | 
						|
 | 
						|
#include <stdlib.h>
 | 
						|
#include <stdio.h>
 | 
						|
#include <errno.h>
 | 
						|
#include <string.h>
 | 
						|
#include <malloc.h>
 | 
						|
//#include "stdioext.h"
 | 
						|
 | 
						|
/* Describe details of an open memstream.  */
 | 
						|
typedef struct fmemcookie {
 | 
						|
  void *storage; /* storage to free on close */
 | 
						|
  char *buf; /* buffer start */
 | 
						|
  size_t pos; /* current position */
 | 
						|
  size_t eof; /* current file size */
 | 
						|
  size_t max; /* maximum file size */
 | 
						|
  char append; /* nonzero if appending */
 | 
						|
  char writeonly; /* 1 if write-only */
 | 
						|
  char saved; /* saved character that lived at pos before write-only NUL */
 | 
						|
} fmemcookie;
 | 
						|
 | 
						|
/* Read up to non-zero N bytes into BUF from stream described by
 | 
						|
   COOKIE; return number of bytes read (0 on EOF).  */
 | 
						|
static int
 | 
						|
fmemread(void *cookie, char *buf, int n)
 | 
						|
{
 | 
						|
  fmemcookie *c = (fmemcookie *) cookie;
 | 
						|
  /* Can't read beyond current size, but EOF condition is not an error.  */
 | 
						|
  if (c->pos > c->eof)
 | 
						|
    return 0;
 | 
						|
  if (n >= c->eof - c->pos)
 | 
						|
    n = c->eof - c->pos;
 | 
						|
  memmove (buf, c->buf + c->pos, n);
 | 
						|
  c->pos += n;
 | 
						|
  return n;
 | 
						|
}
 | 
						|
 | 
						|
/* Write up to non-zero N bytes of BUF into the stream described by COOKIE,
 | 
						|
   returning the number of bytes written or EOF on failure.  */
 | 
						|
static int
 | 
						|
fmemwrite(void *cookie, const char *buf, int n)
 | 
						|
{
 | 
						|
  fmemcookie *c = (fmemcookie *) cookie;
 | 
						|
  int adjust = 0; /* true if at EOF, but still need to write NUL.  */
 | 
						|
 | 
						|
  /* Append always seeks to eof; otherwise, if we have previously done
 | 
						|
     a seek beyond eof, ensure all intermediate bytes are NUL.  */
 | 
						|
  if (c->append)
 | 
						|
    c->pos = c->eof;
 | 
						|
  else if (c->pos > c->eof)
 | 
						|
    memset (c->buf + c->eof, '\0', c->pos - c->eof);
 | 
						|
  /* Do not write beyond EOF; saving room for NUL on write-only stream.  */
 | 
						|
  if (c->pos + n > c->max - c->writeonly)
 | 
						|
    {
 | 
						|
      adjust = c->writeonly;
 | 
						|
      n = c->max - c->pos;
 | 
						|
    }
 | 
						|
  /* Now n is the number of bytes being modified, and adjust is 1 if
 | 
						|
     the last byte is NUL instead of from buf.  Write a NUL if
 | 
						|
     write-only; or if read-write, eof changed, and there is still
 | 
						|
     room.  When we are within the file contents, remember what we
 | 
						|
     overwrite so we can restore it if we seek elsewhere later.  */
 | 
						|
  if (c->pos + n > c->eof)
 | 
						|
    {
 | 
						|
      c->eof = c->pos + n;
 | 
						|
      if (c->eof - adjust < c->max)
 | 
						|
	c->saved = c->buf[c->eof - adjust] = '\0';
 | 
						|
    }
 | 
						|
  else if (c->writeonly)
 | 
						|
    {
 | 
						|
      if (n)
 | 
						|
	{
 | 
						|
	  c->saved = c->buf[c->pos + n - adjust];
 | 
						|
	  c->buf[c->pos + n - adjust] = '\0';
 | 
						|
	}
 | 
						|
      else
 | 
						|
	adjust = 0;
 | 
						|
    }
 | 
						|
  c->pos += n;
 | 
						|
  if (n - adjust)
 | 
						|
    memmove (c->buf + c->pos - n, buf, n - adjust);
 | 
						|
  else
 | 
						|
    {
 | 
						|
      return EOF;
 | 
						|
    }
 | 
						|
  return n;
 | 
						|
}
 | 
						|
 | 
						|
/* Seek to position POS relative to WHENCE within stream described by
 | 
						|
   COOKIE; return resulting position or fail with EOF.  */
 | 
						|
static fpos_t
 | 
						|
fmemseek(void *cookie, fpos_t pos, int whence)
 | 
						|
{
 | 
						|
  fmemcookie *c = (fmemcookie *) cookie;
 | 
						|
  off_t offset = (off_t) pos;
 | 
						|
 | 
						|
  if (whence == SEEK_CUR)
 | 
						|
    offset += c->pos;
 | 
						|
  else if (whence == SEEK_END)
 | 
						|
    offset += c->eof;
 | 
						|
  if (offset < 0)
 | 
						|
    {
 | 
						|
      offset = -1;
 | 
						|
    }
 | 
						|
  else if (offset > c->max)
 | 
						|
    {
 | 
						|
      offset = -1;
 | 
						|
    }
 | 
						|
  else
 | 
						|
    {
 | 
						|
      if (c->writeonly && c->pos < c->eof)
 | 
						|
	{
 | 
						|
	  c->buf[c->pos] = c->saved;
 | 
						|
	  c->saved = '\0';
 | 
						|
	}
 | 
						|
      c->pos = offset;
 | 
						|
      if (c->writeonly && c->pos < c->eof)
 | 
						|
	{
 | 
						|
	  c->saved = c->buf[c->pos];
 | 
						|
	  c->buf[c->pos] = '\0';
 | 
						|
	}
 | 
						|
    }
 | 
						|
  return (fpos_t) offset;
 | 
						|
}
 | 
						|
 | 
						|
/* Reclaim  resources used by stream described by COOKIE.  */
 | 
						|
static int
 | 
						|
fmemclose(void *cookie)
 | 
						|
{
 | 
						|
  fmemcookie *c = (fmemcookie *) cookie;
 | 
						|
  free (c->storage);
 | 
						|
  return 0;
 | 
						|
}
 | 
						|
 | 
						|
/* Open a memstream around buffer BUF of SIZE bytes, using MODE.
 | 
						|
   Return the new stream, or fail with NULL.  */
 | 
						|
FILE *
 | 
						|
fmemopen(void *buf, size_t size, const char *mode)
 | 
						|
{
 | 
						|
  FILE *fp;
 | 
						|
  fmemcookie *c;
 | 
						|
  int flags;
 | 
						|
  int dummy;
 | 
						|
 | 
						|
  if ((flags = __sflags (mode, &dummy)) == 0)
 | 
						|
    return NULL;
 | 
						|
  if (!size || !(buf || flags & __SAPP))
 | 
						|
    {
 | 
						|
      return NULL;
 | 
						|
    }
 | 
						|
  if ((fp = (FILE *) __sfp ()) == NULL)
 | 
						|
    return NULL;
 | 
						|
  if ((c = (fmemcookie *) malloc (sizeof *c + (buf ? 0 : size))) == NULL)
 | 
						|
    {
 | 
						|
      fp->_flags = 0;		/* release */
 | 
						|
 | 
						|
      return NULL;
 | 
						|
    }
 | 
						|
 | 
						|
  c->storage = c;
 | 
						|
  c->max = size;
 | 
						|
  /* 9 modes to worry about.  */
 | 
						|
  /* w/a, buf or no buf: Guarantee a NUL after any file writes.  */
 | 
						|
  c->writeonly = (flags & __SWR) != 0;
 | 
						|
  c->saved = '\0';
 | 
						|
  if (!buf)
 | 
						|
    {
 | 
						|
      /* r+/w+/a+, and no buf: file starts empty.  */
 | 
						|
      c->buf = (char *) (c + 1);
 | 
						|
      *(char *) buf = '\0';
 | 
						|
      c->pos = c->eof = 0;
 | 
						|
      c->append = (flags & __SAPP) != 0;
 | 
						|
    }
 | 
						|
  else
 | 
						|
    {
 | 
						|
      c->buf = (char *) buf;
 | 
						|
      switch (*mode)
 | 
						|
	{
 | 
						|
	case 'a':
 | 
						|
	  /* a/a+ and buf: position and size at first NUL.  */
 | 
						|
	  buf = memchr (c->buf, '\0', size);
 | 
						|
	  c->eof = c->pos = buf ? (char *) buf - c->buf : size;
 | 
						|
	  if (!buf && c->writeonly)
 | 
						|
	    /* a: guarantee a NUL within size even if no writes.  */
 | 
						|
	    c->buf[size - 1] = '\0';
 | 
						|
	  c->append = 1;
 | 
						|
	  break;
 | 
						|
	case 'r':
 | 
						|
	  /* r/r+ and buf: read at beginning, full size available.  */
 | 
						|
	  c->pos = c->append = 0;
 | 
						|
	  c->eof = size;
 | 
						|
	  break;
 | 
						|
	case 'w':
 | 
						|
	  /* w/w+ and buf: write at beginning, truncate to empty.  */
 | 
						|
	  c->pos = c->append = c->eof = 0;
 | 
						|
	  *c->buf = '\0';
 | 
						|
	  break;
 | 
						|
	default:
 | 
						|
	  abort();
 | 
						|
	}
 | 
						|
    }
 | 
						|
 | 
						|
  fp->_file = -1;
 | 
						|
  fp->_flags = flags;
 | 
						|
  fp->_cookie = c;
 | 
						|
  fp->_read = flags & (__SRD | __SRW) ? fmemread : NULL;
 | 
						|
  fp->_write = flags & (__SWR | __SRW) ? fmemwrite : NULL;
 | 
						|
  fp->_seek = fmemseek;
 | 
						|
  fp->_close = fmemclose;
 | 
						|
 | 
						|
  return fp;
 | 
						|
}
 | 
						|
 | 
						|
#endif
 |