/*
 *	@(#)holey.c	1.3 2001/08/25 Connectathon Testsuite
 *	1.3 Lachman ONC Test Suite source
 *
 * test read/write of holey files
 */

#if defined (DOS) || defined (WIN32)
#define DOSorWIN32
#include "../tests.h"
#endif

#ifndef DOSorWIN32
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#if defined(SVR3) || defined(SVR4)
#include <fcntl.h>
#else
#include <sys/file.h>
#endif
#endif /* DOSorWIN32 */

#ifndef L_INCR
#define L_INCR	1
#define L_SET	0
#endif

#ifndef MIN
#define MIN(a, b)	((a) < (b) ? (a) : (b))
#endif

#define BUFSZ   8192
#define FILESZ	70000
#define DATASZ	4321
#ifndef DOSorWIN32
#define HOLESZ	9012
#define FILENM  "holeyfile"
#else
#define HOLESZ	8814			/* XXX why different size? */
#define FILENM  "holefile"
#endif

static int Debug = 0;
static char *Prog;

int
main(argc, argv)
int argc;
char *argv[];
{
	int fd, i, tot, ct, sz, bytes, ret;
	char *filenm = FILENM;
	char buf[BUFSZ];
	int filesz = FILESZ;
	int datasz = DATASZ;
	int holesz = HOLESZ;
	int serrno;
	int oflags;			/* open flags */

	Prog = argv[0];
	if (argc > 1 && strcmp(argv[1], "-d") == 0) {
		Debug = 1;
		argc--;
		argv++;
	}
	if (argc > 5 || (argc > 1 && strcmp(argv[1], "-h") == 0)) {
		fprintf(stderr,
			"usage: %s [filename filesize datasize holesize]\n",
			Prog);
		exit(1);
	}
	if (argc > 1 && strcmp(argv[1], "-") != 0)
		filenm = argv[1];
	if (argc > 2 && strcmp(argv[2], "-") != 0)
		filesz = atoi(argv[2]);
	if (argc > 3 && strcmp(argv[3], "-") != 0)
		datasz = atoi(argv[3]);
	if (argc > 4 && strcmp(argv[4], "-") != 0)
		holesz = atoi(argv[4]);

	umask(0);
	if (datasz > BUFSZ) {
		fprintf(stderr, "%s: datasize (%d) greater than maximum (%d)\n",
			Prog, datasz, BUFSZ);
		exit(1);
	}
	oflags = O_CREAT|O_TRUNC|O_RDWR;
#ifdef DOSorWIN32
	oflags |= O_BINARY;
#endif
	if ((fd = open(filenm, oflags, 0666)) < 0) {
		sprintf(buf, "%s: create of %s", Prog, filenm);
		perror(buf);
		exit(1);
	}
	if (close(fd) < 0) {
		sprintf(buf, "%s: close of %s after creat", Prog, filenm);
		perror(buf);
		exit(1);
	}
	if ((fd = open(filenm, O_RDWR)) < 0) {
		sprintf(buf, "%s: open of %s", Prog, filenm);
		perror(buf);
		exit(1);
	}
	for (i = 0; i < BUFSZ / sizeof (int); i++)
		((int *)buf)[i] = i;

	for (sz = filesz; sz > 0; ) {
		if (datasz || sz == 1) {
			bytes = MIN(sz, datasz);
			if (bytes == 0)
				bytes = 1;
			if ((ret = write(fd, buf, bytes)) != bytes) {
				serrno = errno;
				fprintf(stderr, "write ret %d (expect %d)\n",
					ret, bytes);
				if (serrno) {
					errno = serrno;
					perror("write");
				}
				exit(1);
			}
			sz -= bytes;
		}
		if (sz && holesz) {
			bytes = MIN(sz - 1, holesz);
			if (lseek(fd, bytes, L_INCR) == -1L) {
				perror("lseek (write)");
				exit(1);
			}
			sz -= bytes;
		}
	}
	if (lseek(fd, 0, L_SET) == -1L) {
		perror("lseek (rewind)");
		exit(1);
	}

#ifdef BSD
	printf("BSD workaround: reopening file read-only.\n");
	if (close(fd) < 0) {
		sprintf(buf, "%s: close of %s after write", Prog, filenm);
		perror(buf);
		exit(1);
	}
	if ((fd = open(filenm, O_RDONLY)) < 0) {
		sprintf(buf, "%s: open of %s after write", Prog, filenm);
		perror(buf);
		exit(1);
	}
#endif

	for (sz = filesz; sz > 0; ) {
		if (datasz || sz == 1) {
			bytes = MIN(sz, datasz);
			if (bytes == 0)
				bytes = 1;
			sz -= bytes;
			for (; bytes > 0; bytes -= ret) {
				if (Debug) {
					fprintf(stderr,
				"--data read: offset %d, sz = %d, bytes = %d\n",
						filesz - sz - bytes, sz, bytes);
				}
				if ((ret = read(fd, buf, bytes)) <= 0) {
					serrno = errno;
					fprintf(stderr,
	"read (data) offset %d, sz = %d, bytes = %d (ret = %d), datasz = %d\n",
						filesz - sz - bytes, sz, bytes,
						ret, datasz);
					if (ret < 0) {
						errno = serrno;
						perror("read");
					}
					exit(1);
				}
				ct = bytes - (bytes % sizeof (int));
				if (Debug) {
					fprintf(stderr, "  ret = %d, ct = %d\n",
						ret, ct);
				}
				for (i = 0; i < ct / sizeof (int); i++) {
					if (((int *)buf)[i] != i) {
						fprintf(stderr,
							"bad data in %s\n",
							filenm);
						if (Debug) {
							fprintf(stderr, 
			"address= %d, valueis= %d, shouldbe= %d\n",
							    i, 
							    ((int *)buf)[i], i);
						}
						exit(1);
					}
				}
			}
		}
		if (sz && holesz) {
			tot   = MIN(holesz, sz - 1);
			sz -= tot;
			for (ct = 0; tot > 0; tot -= ret, ct += ret) {
				bytes = MIN(tot, BUFSZ);
				if (Debug) {
					fprintf(stderr,
		"++hole read: offset %d, sz = %d, tot = %d, bytes = %d\n",
						filesz - sz - tot, sz, tot,
						bytes);
				}
				if ((ret = read(fd, buf, bytes)) <= 0) {
					serrno = errno;
					fprintf(stderr,
	"read (hole) offset %d, sz = %d, bytes = %d (ret %d), holesz = %d\n",
						filesz - sz - tot, sz, bytes,
						ret, holesz);
					if (ret < 0) {
						errno = serrno;
						perror("read");
					}
					exit(1);
				}
				if (Debug)
					fprintf(stderr, "  ret = %d\n", ret);
				for (i = 0; i < ret; i++) {
					if (buf[i] != '\0') {
						fprintf(stderr,
			"non-zero data read back from hole (offset %d)\n",
							filesz - sz + ct + i);
						exit(1);
					}
				}
			}
		}
	}

	printf("Holey file test ok\n");
	exit(0);
}