/* Copyright 2007 TeX Users Group

   This program is free software; you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation; either version 2 of the License, or
   (at your option) any later version.

   This program 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
   General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with this program; if not, write to the Free Software
   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
   02110-1301 USA.  */

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

void fail (char *s) {
	puts(s);
	exit(1);
}

int main (int argc, char *argv[]) {
	char *p;
	unsigned meg;
	unsigned long maxallocs;
	unsigned long allocs;

	if (argc < 2) meg = 1;
	else meg = atoi (argv[1]);

	printf ("Allocating %u megabyte%s... ", meg, (meg > 1) ? "s" : "");
	maxallocs = ((long) meg) << 10;

	for (allocs = 0; allocs < maxallocs; allocs++) {
		if (p = malloc(1024)) {
			*p = 'x';
			p[1023] = 'y';
		}
		else {
			printf ("Only %lu bytes available\n", allocs << 10);
			fail("Insufficient Memory!");
		}
	}
	puts("OK");
	return 0;
}