2005-07-01 Martin Hunt <hunt@redhat.com>

	* alloc.c (_stp_alloc): Call _stp_error().
	(_stp_valloc): Ditto.

	* io.c (_stp_warn): New function.
	(_stp_exit): New function.
	(_stp_error): New function.
	(_stp_dbug): New function.

	* runtime.h (dbug): Call _stp_dbug() if	DEBUG is defined.

1 files changed, 12 insertions, 7 deletions

diff --git a/runtime/alloc.c b/runtime/alloc.c
index 480484d3..c9fb8354 100644
--- a/runtime/alloc.c
+++ b/runtime/alloc.c
@@ -1,4 +1,13 @@
-#ifndef _ALLOC_C_ /* -*- linux-c -*- */
+/* Memory allocation functions
+ * Copyright (C) 2005 Red Hat Inc.
+ *
+ * This file is part of systemtap, and is free software.  You can
+ * redistribute it and/or modify it under the terms of the GNU General
+ * Public License (GPL); either version 2, or (at your option) any
+ * later version.
+ */
+
+#ifndef _ALLOC_C_
 #define _ALLOC_C_
 
 /** @file alloc.c
@@ -9,13 +18,9 @@
  * that memory allocation errors will call a handler.  The default will
  * send a signal to the user-space daemon that will trigger the module to
  * be unloaded.
- * @todo Need error handling for memory allocations
  * @{
  */
 
-enum errorcode { ERR_NONE=0, ERR_NO_MEM };
-enum errorcode _stp_error = ERR_NONE;
-
 /** Allocates memory within a probe.
  * This is used for small allocations from within a running
  * probe where the process cannot sleep. 
@@ -28,7 +33,7 @@ void *_stp_alloc(size_t len)
 {
 	void *ptr = kmalloc(len, GFP_ATOMIC);
 	if (unlikely(ptr == NULL))
-		_stp_error = ERR_NO_MEM;
+		_stp_error("_stp_alloc failed.\n");
 	return ptr;
 }
 
@@ -61,7 +66,7 @@ void *_stp_valloc(size_t len)
 	if (likely(ptr))
 		memset(ptr, 0, len);
 	else
-		_stp_error = ERR_NO_MEM;
+		_stp_error("_stp_valloc failed.\n");
 	return ptr;
 }