libs-base/Source/MemoryStream.m

/* Implementation of GNU Objective C memory stream
   Copyright (C) 1994 Free Software Foundation, Inc.
   
   Written by:  R. Andrew McCallum <mccallum@gnu.ai.mit.edu>
   Date: July 1994
   
   This file is part of the GNU Objective C Class Library.

   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.
   */ 

#include <objects/stdobjects.h>
#include <objects/MemoryStream.h>
#include <objects/objc-malloc.h>
#include <objects/Coder.h>
#include <stdarg.h>
#include <assert.h>

/* Deal with memchr: */
#if STDC_HEADERS || HAVE_STRING_H
#include <string.h>
/* An ANSI string.h and pre-ANSI memory.h might conflict.  */
#if !STDC_HEADERS && HAVE_MEMORY_H
#include <memory.h>
#endif /* not STDC_HEADERS and HAVE_MEMORY_H */
#define rindex strrchr
#define bcopy(s, d, n) memcpy ((d), (s), (n))
#define bcmp(s1, s2, n) memcmp ((s1), (s2), (n))
#define bzero(s, n) memset ((s), 0, (n))
#else /* not STDC_HEADERS and not HAVE_STRING_H */
#include <strings.h>
/* memory.h and strings.h conflict on some systems.  */
#endif /* not STDC_HEADERS and not HAVE_STRING_H */

/* This could be done with a set of classes instead. */
enum {MALLOC_MEMORY_STREAM = 0, OBSTACK_MEMORY_STREAM, VM_MEMORY_STREAM};

#define DEFAULT_MEMORY_STREAM_SIZE 64

extern int
objects_vscanf (void *stream, 
		int (*inchar_func)(void*), 
		void (*unchar_func)(void*,int),
		const char *format, va_list argptr);

static BOOL debug_memory_stream = YES;

/* A pretty stupid implementation based on realloc(), but it works for now. */

@implementation MemoryStream

/* xxx This interface will change */
- _initOnMallocBuffer: (char*)b
   size: (unsigned)s		/* size of malloc'ed buffer */
   eofPosition: (unsigned)l	/* length of buffer with data for reading */
   prefix: (unsigned)p		/* never read/write before this position */
   position: (unsigned)i	/* current position for reading/writing */
{
  [super initWithMode:STREAM_READWRITE];
  buffer = b;
  size = s;
  prefix = p;
  position = i;
  eofPosition = l;
  type = MALLOC_MEMORY_STREAM;
  return self;
}

- initWithSize: (unsigned)s
   prefix: (unsigned)p
   position: (unsigned)i
{
  char *b;
  OBJC_MALLOC(b, char, s);
  return [self _initOnMallocBuffer:b size:s eofPosition:i
	       prefix:p position:i];
}

/* xxx Put mode in here */
- initWithSize: (unsigned)s
{
  return [self initWithSize:s prefix:0 position:0];
}

- initWithMode: (int)m
{
  return [self initWithSize: DEFAULT_MEMORY_STREAM_SIZE];
}

- (void) encodeWithCoder: (Coder*)anEncoder
{
  [self notImplemented:_cmd];
}

+ newWithCoder: (Coder*)aDecoder
{
  [self notImplemented:_cmd];
  return self;
}

- (int) writeBytes: (const void*)b length: (int)l
{
  if (prefix+position+l > size)
    {
      size *= 2;
      buffer = (*objc_realloc)(buffer, size);
    }
  memcpy(buffer+prefix+position, b, l);
  position += l;
  if (position > eofPosition)
    eofPosition = position;
  return l;
}

- (int) readBytes: (void*)b length: (int)l
{
  if (position+l > eofPosition)
    l = eofPosition-position;
  memcpy(b, buffer+prefix+position, l);
  position += l;
  return l;
}

- (void) writeLine: (const char *)l
{
  [self writeBytes:l length:strlen(l)];
  [self writeBytes:"\n" length:1];
}

/* This malloc's the buffer pointed to by the return value */
- (char *) readLine
{
  char *nl = memchr(buffer+prefix+position, '\n', eofPosition-position);
  char *ret = NULL;
  if (nl)
    {
      int len = nl-buffer-prefix-position;
      ret = (*objc_malloc)(len+1);
      strncpy(ret, buffer+prefix+position, len);
      ret[len] = '\0';
      position += len+1;
    }
  return ret;
}

/* Making these nested functions (which is what I'd like to do) is
   crashing the va_arg stuff in vscanf().  Why? */
#define MS ((MemoryStream*)s)

int outchar_func(void *s, int c)
{
  if (MS->prefix + MS->position >= MS->size)
    return EOF;
  MS->buffer[MS->prefix + MS->position++] = (char)c;
  return 1;
}

int inchar_func(void *s)
{
  if (MS->prefix + MS->position >= MS->size)
    return EOF;
  return (int) MS->buffer[MS->prefix + MS->position++];
}

void unchar_func(void *s, int c)
{
  if (MS->position > 0)
    MS->position--;
  MS->buffer[MS->prefix + MS->position] = (char)c;
}

#if HAVE_VSPRINTF
- (int) writeFormat: (const char *)format, ...
{
  int ret;
  va_list ap;

  /* xxx Using this ugliness we at least let ourselves safely print
     formatted strings up to 128 bytes long.
     It's digusting, though, and we need to fix it. 
     Using GNU stdio streams would do the trick. 
     */
  if (size - (prefix + position) < 128)
    [self setStreamBufferSize:size*2];

  va_start(ap, format);
  ret = vsprintf(buffer+prefix+position, format, ap);
  va_end(ap);
  position += ret;
  /* xxx Make sure we didn't overrun our buffer.
     As per above kludge, this would happen if we happen to have more than
     128 bytes left in the buffer and we try to write a string longer than
     the num bytes left in the buffer. */
  assert(prefix + position <= size);
  if (position > eofPosition)
    eofPosition = position;
  if (debug_memory_stream)
    {
      *(buffer+prefix+position) = '\0';
      fprintf(stderr, "%s\n", buffer+prefix);
    }
  return ret;
}
#endif

- (int) readFormat: (const char *)format, ...
{
  int ret;
  va_list ap;

  va_start(ap, format);
  ret = objects_vscanf(self, inchar_func, unchar_func, format, ap);
  va_end(ap);
  return ret;
}

- (void) flushStream
{
  /* Nothing needed */
}

- (void) setStreamPosition: (unsigned)i
{
  position = i;
}

- (unsigned) streamPosition
{
  return position;
}

- (void) dealloc
{
  OBJC_FREE(buffer);
  [super dealloc];
}

- (BOOL) streamEof
{
  if (position == eofPosition)
    return YES;
  else
    return NO;
}

- (unsigned) streamBufferSize
{
  return size;
}

- (char *) streamBuffer
{
  return buffer;
}

- (unsigned) prefix
{
  return prefix;
}

- (void) setStreamBufferSize: (unsigned)s
{
  if (s > prefix + eofPosition)
    {
      buffer = (*objc_realloc)(buffer, s);
      size = s;
    }
}

- (unsigned) streamEofPosition
{
  return eofPosition;
}

- (void) setStreamEofPosition: (unsigned)i
{
  if (i < size)
    eofPosition = i;
}  

- (unsigned) streamPrefix
{
  return prefix;
}

- (unsigned) streamBufferLength
{
  return prefix + eofPosition;
}
@end