Index: test/mm/falloc1/test.c
===================================================================
--- test/mm/falloc1/test.c	(revision b2c92f33b5d4bbba5718f4b6fc40ccfbe054b6ca)
+++ test/mm/falloc1/test.c	(revision b2c92f33b5d4bbba5718f4b6fc40ccfbe054b6ca)
@@ -0,0 +1,84 @@
+/*
+ * Copyright (C) 2006 Sergey Bondari
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * - Redistributions of source code must retain the above copyright
+ *   notice, this list of conditions and the following disclaimer.
+ * - Redistributions in binary form must reproduce the above copyright
+ *   notice, this list of conditions and the following disclaimer in the
+ *   documentation and/or other materials provided with the distribution.
+ * - The name of the author may not be used to endorse or promote products
+ *   derived from this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+#include <print.h>
+#include <test.h>
+#include <mm/page.h>
+#include <mm/frame.h>
+#include <arch/mm/page.h>
+#include <arch/types.h>
+#include <debug.h>
+
+#define MAX_FRAMES 1024
+#define MAX_ORDER 8
+#define TEST_RUNS 4
+
+void test(void) {
+	__address frames[MAX_FRAMES];
+	int results[MAX_ORDER+1];
+	
+	int i, order, run;
+	int allocated;
+	int status;
+
+	ASSERT(TEST_RUNS > 1);
+
+	for (run=0;run<=TEST_RUNS;run++) {
+		for (order=0;order<=MAX_ORDER;order++) {
+			printf("Allocating %d frames blocks ... ", 1<<order);
+			allocated = 0;
+			for (i=0;i<MAX_FRAMES>>order;i++) {
+				frames[allocated] = frame_alloc(FRAME_NON_BLOCKING,order, &status);
+				if (status == 0) {
+					allocated++;
+				} else {
+					printf("done. ");
+					break;
+				}
+			}
+		
+			printf("%d blocks alocated.\n", allocated);
+		
+			if (run) {
+				if (results[order] != allocated) {
+					panic("Test failed. Frame leak possible.\n");
+				}
+			} else results[order] = allocated;
+			
+			printf("Deallocating ... ");
+			for (i=0;i<allocated;i++) {
+				frame_free(frames[i]);
+			}
+			printf("done.\n");
+		}
+	}
+
+
+	
+	printf("Test passed\n");
+}
+
Index: test/mm/mapping1/test.c
===================================================================
--- test/mm/mapping1/test.c	(revision 677a6d50fe04baf1fc4de579c24ea8dd8da3099b)
+++ test/mm/mapping1/test.c	(revision b2c92f33b5d4bbba5718f4b6fc40ccfbe054b6ca)
@@ -48,6 +48,6 @@
 	printf("Memory management test mapping #1\n");
 
-	frame0 = frame_alloc(FRAME_KA, ONE_FRAME);
-	frame1 = frame_alloc(FRAME_KA, ONE_FRAME);	
+	frame0 = frame_alloc(FRAME_KA, ONE_FRAME, NULL);
+	frame1 = frame_alloc(FRAME_KA, ONE_FRAME, NULL);	
 
 	printf("Writing %L to physical address %P.\n", VALUE0, KA2PA(frame0));