/* Code covered under 3-Clause BSD License - see LICENSE file for details */

#include <sys/wait.h>

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


int
main(argc, argv)
	int	argc;
	char	*argv[];
{
	register int nforks, i;
	char *cp;
	int pid, child, status, brksize;

	if (argc < 2) {
		printf("usage: %s number-of-forks sbrk-size\n", argv[0]);
		exit(1);
	}

	nforks = atoi(argv[1]);
	if (nforks < 0) {
		printf("%s: bad number of forks\n", argv[1]);
		exit(3);
	}

	brksize = atoi(argv[2]);
	if (brksize < 0) {
		printf("%s: bad size to sbrk\n", argv[2]);
		exit(3);
	}

	cp = (char *)sbrk(brksize);
	if ((int)cp == -1) {
		perror("sbrk");
		exit(4);
	}

	for (i = 0; i < brksize; i += 1024)
		cp[i] = i;

	while (nforks-- > 0) {
		child = fork();
		if (child == -1) {
			perror("fork");
			exit(-1);
		}
		if (child == 0) 
			exit(-1);
		while ((pid = wait(&status)) != -1 && pid != child)
			;
	}

	return 0;
}