//----------------------------------------------------------------------
//  This software is part of the Haiku distribution and is covered
//  by the MIT License.
//
//  This version copyright (c) 2003 Tyler Dauwalder, tyler@dauwalder.net
//  Initial version copyright (c) 2002 Axel Dörfler, axeld@pinc-software.de
//----------------------------------------------------------------------

/*! \file Debug.cpp

	Support code for handy debugging macros.
*/

#include "UdfDebug.h"

#include <stdlib.h>
#include <string.h>
#include <KernelExport.h>
#include <TLS.h>

//----------------------------------------------------------------------
// Long-winded overview of the debug output macros:
//----------------------------------------------------------------------
/*! \def DEBUG_INIT()
	\brief Increases the indentation level, prints out the enclosing function's
	name, and creates a \c DebugHelper object on the stack to automatically
	decrease the indentation level upon function exit.
	
	This macro should be called at the very beginning of any function in
	which you wish to use any of the other debugging macros.
	
	If DEBUG is undefined, does nothing.
*/
//----------------------------------------------------------------------
/*! \def PRINT(x)
	\brief Prints out the enclosing function's name followed by the contents
	of \a x at the current indentation level.
	
	\param x A printf-style format string enclosed in an extra set of parenteses,
	         e.g. PRINT(("%d\n", 0));
	
	If DEBUG is undefined, does nothing.
*/
//----------------------------------------------------------------------
/*! \def LPRINT(x)
	\brief Identical to \c PRINT(x), except that the line number in the source
	file at which the macro is invoked is also printed.
	
	\param x A printf-style format string enclosed in an extra set of parenteses,
	         e.g. PRINT(("%d\n", 0));
	
	If DEBUG is undefined, does nothing.
*/
//----------------------------------------------------------------------
/*! \def SIMPLE_PRINT(x)
	\brief Directly prints the contents of \a x with no extra formatting or
	information included (just like a straight \c printf() call).

	\param x A printf-style format string enclosed in an extra set of parenteses,
	         e.g. PRINT(("%d\n", 0));

	If DEBUG is undefined, does nothing.
*/	         
//----------------------------------------------------------------------
/*! \def PRINT_INDENT()
	\brief Prints out enough indentation characters to indent the current line
	to the current indentation level (assuming the cursor was flush left to
	begin with...).
	
	This function is called by the other \c *PRINT* macros, and isn't really
	intended for general consumption, but you might find it useful.
	
	If DEBUG is undefined, does nothing.
*/
//----------------------------------------------------------------------
/*! \def REPORT_ERROR(error)
	\brief Calls \c LPRINT(x) with a format string listing the error
	code in \c error (assumed to be a \c status_t value) and the
	corresponding text error code returned by a call to \c strerror().
	
	This function is called by the \c RETURN* macros, and isn't really
	intended for general consumption, but you might find it useful.
	
	\param error A \c status_t error code to report.
	
	If DEBUG is undefined, does nothing.
*/
//----------------------------------------------------------------------
/*! \def RETURN_ERROR(error)
	\brief Calls \c REPORT_ERROR(error) if error is a an error code (i.e.
	negative), otherwise remains silent. In either case, the enclosing
	function is then exited with a call to \c "return error;".
		
	\param error A \c status_t error code to report (if negative) and return.
	
	If DEBUG is undefined, silently returns the value in \c error.
*/
//----------------------------------------------------------------------
/*! \def RETURN(error)
	\brief Prints out a description of the error code being returned
	(which, in this case, may be either "erroneous" or "successful")
	and then exits the enclosing function with a call to \c "return error;".
		
	\param error A \c status_t error code to report and return.
	
	If DEBUG is undefined, silently returns the value in \c error.
*/
//----------------------------------------------------------------------
/*! \def FATAL(x)
	\brief Prints out a fatal error message.
	
	This one's still a work in progress...

	\param x A printf-style format string enclosed in an extra set of parenteses,
	         e.g. PRINT(("%d\n", 0));
	
	If DEBUG is undefined, does nothing.
*/
//----------------------------------------------------------------------
/*! \def INFORM(x)
	\brief Directly prints the contents of \a x with no extra formatting or
	information included (just like a straight \c printf() call). Does so
	whether \c DEBUG is defined or not.

	\param x A printf-style format string enclosed in an extra set of parenteses,
	         e.g. PRINT(("%d\n", 0));

	I'll say it again: Prints its output regardless to DEBUG being defined or
	undefined.
*/
//----------------------------------------------------------------------
/*! \def DBG(x)
	\brief If debug is defined, \a x is passed along to the code and
	executed unmodified. If \c DEBUG is undefined, the contents of
	\a x disappear into the ether.
	
	\param x Damn near anything resembling valid C\C++.
*/
//----------------------------------------------------------------------
/*! \def DIE(x)
	\brief Drops the user into the appropriate debugger (user or kernel)
	after printing out the handy message bundled in the parenthesee
	enclosed printf-style format string found in \a x.

	\param x A printf-style format string enclosed in an extra set of parenteses,
	         e.g. PRINT(("%d\n", 0));	
*/


//----------------------------------------------------------------------
// declarations
//----------------------------------------------------------------------

static void indent(uint8 tabCount);
static void unindent(uint8 tabCount);
#if !_KERNEL_MODE
	static int32 get_tls_handle();
#endif

//! Used to keep the tls handle from being allocated more than once.
int32 tls_spinlock = 0;

/*! \brief Used to flag whether the tls handle has been allocated yet.

	Not sure if this really needs to be \c volatile or not...
*/
volatile bool tls_handle_initialized = false;

//! The tls handle of the tls var used to store indentation info.
int32 tls_handle = 0;

//----------------------------------------------------------------------
// public functions
//----------------------------------------------------------------------

/*! \brief Returns the current debug indentation level for the
	current thread.
	
	NOTE: indentation is currently unsupported for R5::kernelland due
	to lack of thread local storage support.
*/
int32
_get_debug_indent_level()
{
#if !_KERNEL_MODE
	return (addr_t)tls_get(get_tls_handle());
#else
	return 1;
#endif
}
	
//----------------------------------------------------------------------
// static functions
//----------------------------------------------------------------------

/*! \brief Increases the current debug indentation level for
	the current thread by 1.
*/
void
indent(uint8 tabCount)
{
#if !_KERNEL_MODE
	tls_set(get_tls_handle(),
		(void*)(addr_t(_get_debug_indent_level()+tabCount)));
#endif
}

/*! \brief Decreases the current debug indentation level for
	the current thread by 1.
*/
void
unindent(uint8 tabCount)
{
#if !_KERNEL_MODE
	tls_set(get_tls_handle(),
		(void*)(addr_t(_get_debug_indent_level()-tabCount)));
#endif
}

#if !_KERNEL_MODE
/*! \brief Returns the thread local storage handle used to store
	indentation information, allocating the handle first if
	necessary.
*/
int32
get_tls_handle()
{
	// Init the tls handle if this is the first call.
	if (!tls_handle_initialized) {
		if (atomic_or(&tls_spinlock, 1) == 0) {
			// First one in gets to init
			tls_handle = tls_allocate();	
			tls_handle_initialized = true;
			atomic_and(&tls_spinlock, 0);
		} else {
			// All others must wait patiently
			while (!tls_handle_initialized) {
				snooze(1);
			}
		}
	}
	return tls_handle;
}
#endif

/*! \brief Helper class for initializing the debugging output
	file.
	
	Note that this hummer isn't threadsafe, but it doesn't really
	matter for our concerns, since the worst it'll result in is
	a dangling file descriptor, and that would be in the case of
	two or more volumes being mounted almost simultaneously...
	not too big of a worry.
*/
class DebugOutputFile {
public:
	DebugOutputFile(const char *filename = NULL) 
		: fFile(-1)
	{
		Init(filename);
	}
	
	~DebugOutputFile() {
		if (fFile >= 0)
			close(fFile);
	}
	
	void Init(const char *filename) {
		if (fFile < 0 && filename)
			fFile = open(filename, O_RDWR | O_CREAT | O_TRUNC);
	}
	
	int File() const { return fFile; }
private:
	int fFile;
};

DebugOutputFile *out = NULL;

/*!	\brief It doesn't appear that the constructor for the global
	\c out variable is called when built as an R5 filesystem add-on,
	so this function needs to be called in udf_mount to let the
	magic happen.
*/
void initialize_debugger(const char *filename)
{
#if DEBUG_TO_FILE
	if (!out) {
		out = new(nothrow) DebugOutputFile(filename);
		dbg_printf("out was NULL!\n");
	} else {
		DebugOutputFile *temp = out;
		out = new(nothrow) DebugOutputFile(filename);
		dbg_printf("out was %p!\n", temp);
	}
#endif
}

// dbg_printf, stolen from Ingo's ReiserFS::Debug.cpp.
void
dbg_printf(const char *format,...)
{
#if DEBUG_TO_FILE
	if (!out)
		return;

	char buffer[1024];
	va_list args;
	va_start(args, format);
	// no vsnprintf() on PPC 
	#if defined(__i386__) && !_KERNEL_MODE
		vsnprintf(buffer, sizeof(buffer) - 1, format, args);
	#else
		vsprintf(buffer, format, args);
	#endif
	va_end(args);
	buffer[sizeof(buffer) - 1] = '\0';
	write(out->File(), buffer, strlen(buffer));
#endif
}

//----------------------------------------------------------------------
// DebugHelper
//----------------------------------------------------------------------

/*! \brief Increases the current indentation level.
*/
DebugHelper::DebugHelper(const char *className, uint8 tabCount)
	: fTabCount(tabCount)
	, fClassName(NULL)
{
	indent(fTabCount);
	if (className) {
		fClassName = (char*)malloc(strlen(className)+1);
		if (fClassName)
			strcpy(fClassName, className);
	}
}

/*! \brief Decreases the current indentation level.
*/
DebugHelper::~DebugHelper()
{
	unindent(fTabCount);
	free(fClassName);
}