/*****************************************************************************/
/*  dalist: a zero-dependency book-keeping library                           */
/*  Copyright (C) 2013--2021  SysDeer Technologies, LLC                      */
/*  Released under GPLv2 and GPLv3; see COPYING.DALIST.                      */
/*****************************************************************************/

#include <dalist/dalist.h>
#include "dalist_impl.h"

dalist_api
int dalist_debug_setup(
	struct dalist_ex *	dlist,
	struct dalist_debug *	dlist_debug,
	enum dalist_dbgenv	dbgenv,
	dalist_sprintf *	pfn_sprintf,
	dalist_write *		pfn_write,
	dalist_write_file *	pfn_write_file)
{
	if (dlist)
		/* in case we fail */
		dlist->debug = 0;
	else
		return DALIST_ELIST;

	if (!dlist_debug)
		return DALIST_EDEBUGSTRUCT;
	else if (!pfn_sprintf)
		return DALIST_EDEBUGSPRINTF;

	switch (dbgenv) {
		case DALIST_DEBUG_ENV_POSIX:
			if (!pfn_write)
				return DALIST_EDEBUGWRITE;
			break;

		case DALIST_DEBUG_ENV_NT:
			if (!pfn_write_file)
				return DALIST_EDEBUGWRITEFILE;
			break;

		default:
			return DALIST_EDEBUGENV;
	}

	dlist_debug->dbgenv        = dbgenv;
	dlist_debug->sprintf       = pfn_sprintf;
	dlist_debug->write         = pfn_write;
	dlist_debug->zw_write_file = pfn_write_file;

	dlist->debug = dlist_debug;

	return DALIST_OK;
}


static const char dalist_fmt_header_32[] =
	"dlist:\t\t0x%08x\n"
	"dlist->head:\t0x%08x\n"
	"dlist->tail:\t0x%08x\n\n";

static const char dalist_fmt_header_64[] =
	"dlist:\t\t0x%016x\n"
	"dlist->head:\t0x%016x\n"
	"dlist->tail:\t0x%016x\n\n";

static const char dalist_fmt_node_32[] =
	"node: 0x%08x\t"
	"prev: 0x%08x\t"
	"next: 0x%08x\t"
	"any:  0x%08x\t\n";

static const char dalist_fmt_node_64[] =
	"node: 0x%016x\t"
	"prev: 0x%016x\t"
	"next: 0x%016x\t"
	"any:  0x%016x\t\n";

static const char * dalist_fmt_headers[2] = {
	dalist_fmt_header_32,
	dalist_fmt_header_64
};

static const char * dalist_fmt_nodes[2] = {
	dalist_fmt_node_32,
	dalist_fmt_node_64
};


static int dalist_dbg_write_posix(
	struct dalist_ex *	dlist,
	intptr_t		fildes_or_hfile,
	const void *		buf,
	size_t			nbyte)
{
	int		fildes;
	dalist_write *	pfn_write;
	ssize_t		bytes_written;

	fildes = (int)fildes_or_hfile;
	pfn_write = (dalist_write *)dlist->debug->write;

	bytes_written = pfn_write(fildes,buf,nbyte);

	if (bytes_written < 0)
		return DALIST_EDEBUGENV;

	else if (bytes_written == (ssize_t)nbyte)
		return DALIST_OK;

	else
		return DALIST_EDEBUGENV;
}


static int dalist_dbg_write_nt(
	struct dalist_ex *	dlist,
	intptr_t		fildes_or_hfile,
	const void *		buf,
	size_t			nbyte)
{
	void *			hfile;
	struct dalist_iosb	iosb;
	dalist_write_file *	pfn_write_file;

	hfile          = (void *)fildes_or_hfile;
	pfn_write_file = (dalist_write_file *)dlist->debug->zw_write_file;

	return pfn_write_file(
		hfile,
		(void *)0,
		(void *)0,
		(void *)0,
		&iosb,
		(void *)buf,
		(uint32_t)nbyte,
		(int64_t *)0,
		(uint32_t *)0);
}


dalist_api
int dalist_debug_print(
	struct dalist_ex *	dlist,
	intptr_t		fildes_or_hfile)
{
	const char *		fmt_header;
	const char *		fmt_node;
	char			dbg_buf[128];
	intptr_t		nbyte;

	int			status;
	struct dalist_node *	node;
	dalist_sprintf *	dbg_sprintf;
	dalist_dbg_write *	dbg_write;

	if (!dlist)
		return DALIST_ELIST;
	else if (!dlist->debug)
		return DALIST_EDEBUGENV;

	/* initial setup */
	dbg_sprintf = (dalist_sprintf *)dlist->debug->sprintf;
	fmt_header  = dalist_fmt_headers[(sizeof(size_t)-4)/4];
	fmt_node    = dalist_fmt_nodes[(sizeof(size_t)-4)/4];

	if (dlist->debug->dbgenv == DALIST_DEBUG_ENV_POSIX)
		dbg_write = dalist_dbg_write_posix;
	else if (dlist->debug->dbgenv == DALIST_DEBUG_ENV_NT)
		dbg_write = dalist_dbg_write_nt;
	else
		dbg_write = 0;

	if ((!dbg_sprintf) || (!dbg_write))
		return DALIST_EDEBUGENV;

	/* print list header */
	nbyte = dbg_sprintf((char *)dbg_buf,fmt_header,dlist,dlist->head,dlist->tail);

	if (nbyte < 0)
		return DALIST_EDEBUGENV;

	status = dbg_write(dlist,fildes_or_hfile,dbg_buf,nbyte);

	if (status != NT_STATUS_SUCCESS)
		return DALIST_EDEBUGENV;

	/* print node information */
	status = DALIST_OK;
	node   = (struct dalist_node *)dlist->head;

	while (node && (status == DALIST_OK)) {
		/* create output line */
		nbyte = dbg_sprintf(
			(char *)dbg_buf,
			fmt_node,
			node,
			node->prev,
			node->next,
			node->any);

		if (nbyte < 0)
			return DALIST_EDEBUGENV;

		/* write output line */
		status = dbg_write(
			dlist,
			fildes_or_hfile,
			dbg_buf,
			nbyte);

		node = node->next;
	}

	return DALIST_OK;
}