/*
 * Copyright (c) 2011 Martin Pieuchot <mpi@openbsd.org>
 *
 * 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.
 */

#include <err.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#define OFFSET 16384

const char start[] = "start";
const char hello[] = "hello";

const char *
_umem_debug_init(void)
{
	return ("default,verbose");
}

const char *
_umem_logging_init(void)
{
	return ("fail,contents");
}

int
main(void)
{
	FILE	*fp;
	char	*buf = (char *)0xff;
	size_t	 size = 0;
	off_t	 off;
	int	 i, failures = 0;

	if ((fp = open_memstream(&buf, &size)) == NULL) {
		warn("open_memstream failed");
		return (1);
	}

	off = ftello(fp);
	if (off != 0) {
		warnx("ftello failed. (1)");
		failures++;
	}

	if (fflush(fp) != 0) {
		warnx("fflush failed. (2)");
		failures++;
	}

	if (size != 0) {
		warnx("string should be empty. (3)");
		failures++;
	}

	if (buf == (char *)0xff) {
		warnx("buf not updated. (4)");
		failures++;
	}

	if (fseek(fp, OFFSET, SEEK_SET) != 0) {
		warnx("failed to fseek. (5)");
		failures++;
	}

	if (fprintf(fp, hello) == EOF) {
		warnx("fprintf failed. (6)");
		failures++;
	}

	if (fflush(fp) == EOF) {
		warnx("fflush failed. (7)");
		failures++;
	}

	if (size != OFFSET + sizeof(hello)-1) {
		warnx("failed, size %zu should be %zu. (8)",
		    size, OFFSET + sizeof(hello)-1);
		failures++;
	}

	if (fseek(fp, 0, SEEK_SET) != 0) {
		warnx("failed to fseek. (9)");
		failures++;
	}

	if (fprintf(fp, start) == EOF) {
		warnx("fprintf failed. (10)");
		failures++;
	}

	if (fflush(fp) == EOF) {
		warnx("fflush failed. (11)");
		failures++;
	}

	if (size != sizeof(start)-1) {
		warnx("failed, size %zu should be %zu. (12)",
		    size, sizeof(start)-1);
		failures++;
	}

	/* Needed for sparse files */
	if (strncmp(buf, start, sizeof(start)-1) != 0) {
		warnx("failed, buffer didn't start with '%s'. (13)", start);
		failures++;
	}
	for (i = sizeof(start)-1; i < OFFSET; i++)
		if (buf[i] != '\0') {
			warnx("failed, buffer non zero (offset %d). (14)", i);
			failures++;
			break;
		}

	if (memcmp(buf + OFFSET, hello, sizeof(hello)-1) != 0) {
		warnx("written string incorrect. (15)");
		failures++;
	}

	/* verify that simply seeking past the end doesn't increase the size */
	if (fseek(fp, 100, SEEK_END) != 0) {
		warnx("failed to fseek. (16)");
		failures++;
	}

	if (fflush(fp) == EOF) {
		warnx("fflush failed. (17)");
		failures++;
	}

	if (size != OFFSET + sizeof(hello)-1) {
		warnx("failed, size %zu should be %zu. (18)",
		    size, OFFSET + sizeof(hello)-1);
		failures++;
	}

	if (fseek(fp, -1, SEEK_END) != 0) {
		warnx("failed to fseek. (19)");
		failures++;
	}

	if (fseek(fp, 8, SEEK_SET) != 0) {
		warnx("failed to fseek. (20)");
		failures++;
	}

	if (ftell(fp) != 8) {
		warnx("failed seek test. (21)");
		failures++;
	}

	/* Try to seek backward */
	if (fseek(fp, -1, SEEK_CUR) != 0) {
		warnx("failed to fseek. (22)");
		failures++;
	}

	if (ftell(fp) != 7) {
		warnx("failed seeking backward. (23)");
		failures++;
	}

	if (fseek(fp, 5, SEEK_CUR) != 0) {
		warnx("failed to fseek. (24)");
		failures++;
	}

	if (fclose(fp) == EOF) {
		warnx("fclose failed. (25)");
		failures++;
	}

	if (size != 12) {
		warnx("failed, size %zu should be %u.  (26)",
		    size, 12);
		failures++;
	}

	free(buf);

	return (failures);
}