/*********************************************************/

/*  toksvc: a framework-native token broker service      */

/*  Copyright (C) 2020  Z. Gilboa                        */

/*  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 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;

	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_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);

		}

	}

	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);

		}

	}

	return (dctx->cctx->drvflags & TOKS_DRIVER_MODE_SERVER)

		? NT_STATUS_SERVICE_NOTIFICATION

		: toks_exit(dctx,ret);

}