sortix--sortix/libc/stdio/open_memstream.c

/*
 * Copyright (c) 2015 Jonas 'Sortie' Termansen.
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 *
 * libc/stdio/open_memstream.c
 * Open a memory stream.
 */

#include <sys/types.h>

#include <errno.h>
#include <limits.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

struct memstream
{
	char** string_out;
	size_t* used_out;
	char* string;
	size_t offset;
	size_t used;
	size_t size;
};

static ssize_t memstream_write(void* ms_ptr, const void* src, size_t count)
{
	struct memstream* ms = (struct memstream*) ms_ptr;
	if ( (size_t) SSIZE_MAX < count )
		count = SSIZE_MAX;
	if ( ms->offset == SIZE_MAX && 1 <= count )
		return errno = EOVERFLOW, -1;
	if ( (size_t) SSIZE_MAX - ms->offset < count )
		count = SIZE_MAX - ms->offset;
	size_t required_size = ms->offset + count;
	if ( ms->size < required_size )
	{
		// TODO: Multiplication overflow.
		size_t new_size = 2 * ms->size;
		if ( new_size < 16 )
			new_size = 16;
		if ( new_size < required_size )
			new_size = required_size;
		if ( SIZE_MAX - new_size < 1 )
			return errno = EOVERFLOW, -1;
		char* new_string = (char*) realloc(ms->string, new_size + 1);
		if ( !new_string )
			return -1;
		ms->string = new_string;
		ms->size = new_size;
	}
	memcpy(ms->string + ms->offset, src, count);
	ms->offset += count;
	if ( ms->used < required_size )
	{
		ms->used = required_size;
		ms->string[ms->used] = '\0';
	}
	*ms->string_out = ms->string;
	*ms->used_out = ms->used;
	return count;
}

static off_t memstream_seek(void* ms_ptr, off_t offset, int whence)
{
	struct memstream* ms = (struct memstream*) ms_ptr;
	off_t base;
	switch ( whence )
	{
	case SEEK_SET: base = 0; break;
	case SEEK_CUR: base = (off_t) ms->offset; break;
	case SEEK_END: base = (off_t) ms->used; break;
	default: return errno = EINVAL, -1;
	}
	if ( offset < -base || base - SSIZE_MAX < offset )
		return errno = EOVERFLOW, -1;
	return (off_t) (ms->offset = (size_t) (base + offset));
}

FILE* open_memstream(char** string_out, size_t* used_out)
{
	char* string = (char*) malloc(1);
	if ( !string )
		return NULL;
	string[0] = '\0';
	struct memstream* ms = (struct memstream*) malloc(sizeof(struct memstream));
	if ( !ms )
		return free(string), (FILE*) NULL;
	FILE* fp = fnewfile();
	if ( !fp )
		return free(ms), free(string), (FILE*) NULL;
	ms->string_out = string_out;
	ms->used_out = used_out;
	ms->string= string;
	ms->offset = 0;
	ms->used = 0;
	ms->size = 0;
	fp->flags |= _FILE_WRITABLE;
	fp->fd = -1;
	fp->user = ms;
	fp->write_func = memstream_write;
	fp->seek_func = memstream_seek;
	*string_out = ms->string;
	*used_out = ms->used;
	return fp;
}