/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2023 The FreeBSD Foundation
 *
 * This software was developed by Jake Freeland <jfree@FreeBSD.org>
 * under sponsorship from the FreeBSD Foundation.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <assert.h>
#include <err.h>
#include <string.h>

#include "syslogd_cap.h"

struct cfiled_list cfiled_head;

int
cap_p_open(cap_channel_t *chan, size_t filed_idx, const char *prog,
    int *procdesc)
{
	nvlist_t *nvl = nvlist_create(0);
	int error, pipedesc_w;

	nvlist_add_string(nvl, "cmd", "p_open");
	nvlist_add_number(nvl, "filed_idx", filed_idx);
	nvlist_add_string(nvl, "prog", prog);
	nvl = cap_xfer_nvlist(chan, nvl);
	if (nvl == NULL) {
		logerror("Failed to xfer p_open nvlist");
		exit(1);
	}
	error = nvlist_get_number(nvl, "error");
	if (error != 0) {
		errno = error;
		logerror("Failed to open piped command");
	}
	pipedesc_w = dnvlist_take_descriptor(nvl, "pipedesc_w", -1);
	*procdesc = dnvlist_take_descriptor(nvl, "procdesc", -1);

	nvlist_destroy(nvl);
	return (pipedesc_w);
}

int
casper_p_open(nvlist_t *nvlin, nvlist_t *nvlout)
{
	struct cap_filed *cfiled;
	size_t filed_idx;
	int pipedesc_w, procdesc = -1;
	const char *prog;

	filed_idx = nvlist_get_number(nvlin, "filed_idx");
	prog = nvlist_get_string(nvlin, "prog");
	SLIST_FOREACH(cfiled, &cfiled_head, next) {
		if (cfiled->idx != filed_idx)
			continue;
		if (strcmp(cfiled->pipe_cmd, prog) != 0)
			return (-1);

		pipedesc_w = p_open(prog, &procdesc);
		if (pipedesc_w == -1)
			return (-1);
		nvlist_move_descriptor(nvlout, "pipedesc_w", pipedesc_w);
		nvlist_move_descriptor(nvlout, "procdesc", procdesc);
		return (0);
	}

	return (-1);
}

const char *
cap_ttymsg(cap_channel_t *chan, struct iovec *iov, int iovcnt,
    const char *line, int tmout)
{
	nvlist_t *nvl = nvlist_create(0);
	int error;
	static char errbuf[1024];
	char *ret = NULL;

	nvlist_add_string(nvl, "cmd", "ttymsg");
	for (int i = 0; i < iovcnt; ++i)
		nvlist_append_string_array(nvl, "iov_strs", iov[i].iov_base);
	nvlist_add_string(nvl, "line", line);
	nvlist_add_number(nvl, "tmout", tmout);

	nvl = cap_xfer_nvlist(chan, nvl);
	if (nvl == NULL) {
		logerror("Failed to xfer ttymsg nvlist");
		exit(1);
	}
	error = nvlist_get_number(nvl, "error");
	if (error != 0) {
		errno = error;
		logerror("Failed to ttymsg");
	}
	if (nvlist_exists_string(nvl, "errstr")) {
		const char *errstr = nvlist_get_string(nvl, "errstr");
		(void)strlcpy(errbuf, errstr, sizeof(errbuf));
		ret = errbuf;
	}

	nvlist_destroy(nvl);
	return (ret);
}

int
casper_ttymsg(nvlist_t *nvlin, nvlist_t *nvlout)
{
	char **nvlstrs;
	struct iovec *iov;
	size_t iovcnt;
	int tmout;
	const char *line;

	nvlstrs = nvlist_take_string_array(nvlin, "iov_strs", &iovcnt);
	assert(iovcnt <= TTYMSG_IOV_MAX);
	iov = calloc(iovcnt, sizeof(*iov));
	if (iov == NULL)
		err(EXIT_FAILURE, "calloc");
	for (size_t i = 0; i < iovcnt; ++i) {
		iov[i].iov_base = nvlstrs[i];
		iov[i].iov_len = strlen(nvlstrs[i]);
	}
	line = nvlist_get_string(nvlin, "line");
	tmout = nvlist_get_number(nvlin, "tmout");
	line = ttymsg(iov, iovcnt, line, tmout);
	if (line != NULL)
		nvlist_add_string(nvlout, "errstr", line);

	free(iov);
	return (0);
}

void
cap_wallmsg(cap_channel_t *chan, const struct filed *f, struct iovec *iov,
    int iovcnt)
{
	nvlist_t *nvl = nvlist_create(0);
	int error;

	nvlist_add_string(nvl, "cmd", "wallmsg");
	/*
	 * The filed_to_nvlist() function is not needed
	 * here because wallmsg() only uses f_type and
	 * fu_uname members, which are both inline.
	 */
	nvlist_add_binary(nvl, "filed", f, sizeof(*f));
	for (int i = 0; i < iovcnt; ++i)
		nvlist_append_string_array(nvl, "iov_strs", iov[i].iov_base);

	nvl = cap_xfer_nvlist(chan, nvl);
	if (nvl == NULL) {
		logerror("Failed to xfer wallmsg nvlist");
		exit(1);
	}
	error = nvlist_get_number(nvl, "error");
	if (error != 0) {
		errno = error;
		logerror("Failed to wallmsg");
	}
	nvlist_destroy(nvl);
}

int
casper_wallmsg(nvlist_t *nvlin)
{
	const struct filed *f;
	char **nvlstrs;
	struct iovec *iov;
	size_t sz;

	f = nvlist_get_binary(nvlin, "filed", &sz);
	assert(sz == sizeof(*f));
	nvlstrs = nvlist_take_string_array(nvlin, "iov_strs", &sz);
	assert(sz <= TTYMSG_IOV_MAX);
	iov = calloc(sz, sizeof(*iov));
	if (iov == NULL)
		err(EXIT_FAILURE, "calloc");
	for (size_t i = 0; i < sz; ++i) {
		iov[i].iov_base = nvlstrs[i];
		iov[i].iov_len = strlen(nvlstrs[i]);
	}
	wallmsg(f, iov, sz);

	for (size_t i = 0; i < sz; ++i)
		free(iov[i].iov_base);
	free(iov);
	return (0);
}