/*********************************************************/
/*  toksvc: a framework-native token broker service      */
/*  Copyright (C) 2020  SysDeer Technologies, LLC        */
/*  Released under GPLv2 and GPLv3; see COPYING.TOKSVC.  */
/*********************************************************/

#include <ntapi/ntapi.h>
#include <stdio.h>
#include <unistd.h>

#include <toksvc/toksvc.h>
#include "toksvc_init_impl.h"
#include "toksvc_driver_impl.h"
#include "toksvc_dprintf_impl.h"
#include "toksvc_nolibc_impl.h"
#include "toksvc_log_impl.h"

#ifndef TOKS_DRIVER_FLAGS
#define TOKS_DRIVER_FLAGS	TOKS_DRIVER_VERBOSITY_ERRORS \
				| TOKS_DRIVER_DAEMON_ALWAYS   \
				| TOKS_DRIVER_VERBOSITY_USAGE
#endif

static const char vermsg[] = "%s%s%s (git://midipix.org/toksvc): "
			     "version %s%d.%d.%d%s.\n"
			     "[commit reference: %s%s%s]\n";

static const char * const toks_ver_color[6] = {
		"\x1b[1m\x1b[35m","\x1b[0m",
		"\x1b[1m\x1b[32m","\x1b[0m",
		"\x1b[1m\x1b[34m","\x1b[0m"
};

static const char * const toks_ver_plain[6] = {
		"","",
		"","",
		"",""
};

static ssize_t toks_version(struct toks_driver_ctx * dctx)
{
	const struct toks_source_version * verinfo;
	const char * const * verclr;

	verinfo = toks_source_version();
	verclr  = isatty(STDOUT_FILENO) ? toks_ver_color : toks_ver_plain;

	return toks_dprintf(
			STDOUT_FILENO,vermsg,
			verclr[0],dctx->program,verclr[1],
			verclr[2],verinfo->major,verinfo->minor,
			verinfo->revision,verclr[3],
			verclr[4],verinfo->commit,verclr[5]);
}

static ssize_t toks_output_service_info(const struct toks_service_info * svcinfo)
{
	char guidstr[40];

	toks_uuid_to_string(
		&svcinfo->uuid,
		&guidstr);

	return toks_dprintf(STDOUT_FILENO,
		"toks_uuid:%s\n"
		"toks_syspid:%d\n"
		"toks_systid:%d\n"
		"toks_ctrlpid:%d\n"
		"toks_csyspid:%d\n"
		"toks_allocated:%d\n"
		"toks_available:%d\n"
		"toks_pending:%d\n"
		"toks_tused:%d\n"
		"toks_tfree:%d\n"
		"toks_loglevel:%d\n",
		guidstr,
		svcinfo->syspid,
		svcinfo->systid,
		svcinfo->ctrlpid,
		svcinfo->csyspid,
		svcinfo->allocated,
		svcinfo->available,
		svcinfo->pending,
		svcinfo->tused,
		svcinfo->tfree,
		svcinfo->loglevel);
}

static void toks_output_token_info(const struct toks_token_ctx * tokctx)
{
	int                             idx;
	const struct toks_token_info *  tinfo;
	char                            path[2048];

	toks_dprintf(STDOUT_FILENO,"toks_tokens:{%s",
		tokctx->tused ? "\n" : "");

	for (idx=0,tinfo=tokctx->tokens; idx<tokctx->tused; tinfo++) {
		toks_log_get_arbitrary_process_name(
			&(nt_cid){.process_id=tinfo->tsyspid,.thread_id=0},
			path,sizeof(path));

		toks_dprintf(STDOUT_FILENO,
			"\t{"
			".token_no=%d, "
			".token_id=%08X-%08X-%08X-%08X-%08X-%08X, "
			".token_pid=%d, "
			".token_syspid=%d, "
			".token_refstr=``%s''}\n"
			"\t\t"
			".client_image=%s\n",
			++idx,
			tinfo->keys[0],
			tinfo->keys[1],
			tinfo->keys[2],
			tinfo->keys[3],
			tinfo->keys[4],
			tinfo->keys[5],
			tinfo->tokpid,
			tinfo->tsyspid,
			tinfo->meta,
			path);
	}

	toks_dprintf(STDOUT_FILENO,"}\n");

	toks_dprintf(STDOUT_FILENO,"toks_waiters:{%s",
		tokctx->pending ? "\n" : "");

	for (idx=0; idx<tokctx->pending; tinfo++) {
		toks_log_get_arbitrary_process_name(
			&(nt_cid){.process_id=tinfo->tsyspid,.thread_id=0},
			path,sizeof(path));

		toks_dprintf(STDOUT_FILENO,
			"\t{"
			".waiter_no=%d, "
			".token_id=%08X-%08X-%08X-%08X-%08X-%08X, "
			".token_pid=%d, "
			".token_syspid=%d, "
			".token_refstr=``%s''}\n"
			"\t\t"
			".client_image=%s\n",
			++idx,
			tinfo->keys[0],
			tinfo->keys[1],
			tinfo->keys[2],
			tinfo->keys[3],
			tinfo->keys[4],
			tinfo->keys[5],
			tinfo->tokpid,
			tinfo->tsyspid,
			tinfo->meta,
			path);
	}

	toks_dprintf(STDOUT_FILENO,"}\n");
}

static int toks_exit(struct toks_driver_ctx * dctx, int nerrors)
{
	toks_free_driver_ctx(dctx);
	return nerrors ? 2 : 0;
}

int toks_main(char ** argv, char ** envp)
{
	int				ret;
	int32_t				status;
	struct toks_driver_ctx *	dctx;
	struct toks_token_string	toks;
	struct toks_service_info	svcinfo;
	struct toks_server_snapshot *	snapshot;

	if ((ret = toks_init()))
		return ret;

	if ((ret = toks_get_driver_ctx(argv,envp,TOKS_DRIVER_FLAGS,&dctx)))
		return (ret == TOKS_USAGE)
			? !argv || !argv[0] || !argv[1]
			: 2;

	if (dctx->cctx->drvflags & TOKS_DRIVER_VERSION)
		if ((toks_version(dctx)) < 0)
			return toks_exit(dctx,2);

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_ACQUIRE) {
		ret = (status = toks_client_acquire(dctx))
			? 2 : 0;

		switch (status) {
			case NT_STATUS_SUCCESS:
				break;

			case NT_STATUS_TIMEOUT:
				toks_dprintf(STDERR_FILENO,
					"%s: the operataion timed out [0x%x].\n",
					toks_log_basename(argv[0]),status);
				break;

			default:
				toks_dprintf(STDERR_FILENO,
					"%s: the operataion has failed [0x%x].\n",
					toks_log_basename(argv[0]),status);
		}

		if (ret == 0) {
			toks_client_token_to_str(dctx,&toks);
			toks_dprintf(STDOUT_FILENO,"%s\n",toks.token);
		}
	}

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_RELEASE) {
		ret = toks_client_release(dctx)
			? 2 : 0;

		if (ret == 0) {
			toks_client_token_to_str(dctx,&toks);
			toks_dprintf(STDOUT_FILENO,"token %s has been released\n",toks.token);
		}

		if (ret == 2) {
			toks_dprintf(STDERR_FILENO,"error: invalid token");
		}
	}

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_NTOKENS_GET) {
		toks_dprintf(STDOUT_FILENO,
			"toks_ntokens:%d\n",
			toks_get_driver_ntokens(dctx));
	}

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_NTOKENS_SET) {
		status = toks_service_ioctl(
			dctx,TOKS_IOCTL_SET_TOKEN_COUNT,
			&(uintptr_t){toks_get_driver_ntokens(dctx)},
			0);

		ret = status ? 2 : 0;

		switch (status) {
			case NT_STATUS_SUCCESS:
				toks_dprintf(STDERR_FILENO,
					"%s: the service's ntokens parameter "
					"was successfully set to %d\n",
					toks_log_basename(argv[0]),
					toks_get_driver_ntokens(dctx));
				break;

			default:
				toks_dprintf(STDERR_FILENO,
					"%s: the TOKS_IOCTL_SET_TOKEN_COUNT "
					"ioctl operataion has failed "
					"(check the system's documentation) [0x%x].",
					toks_log_basename(argv[0]),status);
		}
	}

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_LOGLEVEL_GET) {
		toks_dprintf(STDOUT_FILENO,
			"toks_loglevel:%d\n",
			dctx->cctx->loglevel);
	}

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_LOGLEVEL_SET) {
		status = toks_service_ioctl(
			dctx,TOKS_IOCTL_SET_LOG_LEVEL,
			&(uintptr_t){dctx->cctx->loglevel},
			0);

		ret = status ? 2 : 0;

		switch (status) {
			case NT_STATUS_SUCCESS:
				toks_dprintf(STDERR_FILENO,
					"%s: the service's log-level parameter "
					"was successfully set to %d\n",
					toks_log_basename(argv[0]),
					dctx->cctx->loglevel);
				break;

			default:
				toks_dprintf(STDERR_FILENO,
					"%s: the TOKS_IOCTL_SET_LOG_LEVEL "
					"ioctl operataion has failed "
					"(check the system's documentation) [0x%x].",
					toks_log_basename(argv[0]),status);
		}
	}

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_SVCINFO_GET) {
		status = toks_client_query_service(dctx,&svcinfo);
		ret    = status ? 2 : 0;

		switch (status) {
			case NT_STATUS_SUCCESS:
				toks_output_service_info(&svcinfo);
				break;

			default:
				toks_dprintf(STDERR_FILENO,
					"%s: the TOKS_IOCTL_GET_SERVICE_INFO "
					"ioctl operataion has failed "
					"(check the system's documentation) [0x%x].",
					toks_log_basename(argv[0]),status);
		}
	}

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_SVCINFO_LOG) {
		status = toks_service_ioctl(dctx,TOKS_IOCTL_LOG_SERVICE_INFO,0,0);
		ret    = status ? 2 : 0;

		switch (status) {
			case NT_STATUS_SUCCESS:
				toks_dprintf(STDERR_FILENO,
					"%s: a service info log record "
					"was successfully created\n",
					toks_log_basename(argv[0]));
				break;

			default:
				toks_dprintf(STDERR_FILENO,
					"%s: the TOKS_IOCTL_LOG_SERVICE_INFO "
					"ioctl operataion has failed "
					"(check the system's documentation) [0x%x].",
					toks_log_basename(argv[0]),status);
		}
	}

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_TOKINFO_GET) {
		status = toks_get_server_snapshot(dctx,&snapshot);
		ret    = status ? 2 : 0;

		switch (status) {
			case NT_STATUS_SUCCESS:
				toks_output_service_info(&snapshot->svcinfo);
				toks_output_token_info(snapshot->tokctx);
				toks_free_server_snapshot(snapshot);
				break;

			default:
				toks_dprintf(STDERR_FILENO,
					"%s: the TOKS_IOCTL_GET_TOKEN_INFO "
					"ioctl operataion has failed "
					"(check the system's documentation) [0x%x].",
					toks_log_basename(argv[0]),status);
		}
	}

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_TOKINFO_LOG) {
		status = toks_service_ioctl(dctx,TOKS_IOCTL_LOG_TOKEN_INFO,0,0);
		ret    = status ? 2 : 0;

		switch (status) {
			case NT_STATUS_SUCCESS:
				toks_dprintf(STDERR_FILENO,
					"%s: a service info log record, "
					"followed by a info table for currently "
					"allocated tokens, was successfully created\n",
					toks_log_basename(argv[0]));
				break;

			default:
				toks_dprintf(STDERR_FILENO,
					"%s: the TOKS_IOCTL_LOG_TOKEN_INFO "
					"ioctl operataion has failed "
					"(check the system's documentation) [0x%x].",
					toks_log_basename(argv[0]),status);
		}
	}

	if (dctx->cctx->drvflags & TOKS_DRIVER_ACTION_ABORT) {
		ret = (status = toks_service_abort(dctx))
			? 2 : 0;

		switch (status) {
			case NT_STATUS_SUCCESS:
				toks_dprintf(STDERR_FILENO,
					"%s: the server responded with no error.\n",
					toks_log_basename(argv[0]),status);
				break;

			case NT_STATUS_ACCESS_DENIED:
				toks_dprintf(STDERR_FILENO,
					"%s: the abort operataion timed (access denied) [0x%x].\n",
					toks_log_basename(argv[0]),status);
				break;

			default:
				toks_dprintf(STDERR_FILENO,
					"%s: the abort operataion failed ",
					"(check the system's documentation) [0x%x].",
					toks_log_basename(argv[0]),status);
		}
	}

	return (dctx->cctx->drvflags & TOKS_DRIVER_MODE_SERVER)
		? NT_STATUS_SERVICE_NOTIFICATION
		: toks_exit(dctx,ret);
}