xref: /src/tests/lib/libcurses/slave/commands.c

/*	$NetBSD: commands.c,v 1.15 2021/06/13 12:46:01 rillig Exp $	*/

/*-
 * Copyright 2009 Brett Lymn <blymn (at) NetBSD.org>
 * Copyright 2021 Roland Illig <rillig (at) NetBSD.org>
 *
 * All rights reserved.
 *
 * This code has been donated to The NetBSD Foundation by the Author.
 *
 * 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. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``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 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 <curses.h>
#include <string.h>
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <err.h>
#include <sys/types.h>

#include "returns.h"
#include "slave.h"
#include "command_table.h"

extern int initdone;

static void report_message(data_enum_t, const char *);

/*
 * Match the passed command string and execute the associated test
 * function.
 */
void
command_execute(char *func, int nargs, char **args)
{
	size_t i, j;

	i = 0;
	while (i < ncmds) {
		if (strcmp(func, commands[i].name) == 0) {
			/* Check only restricted set of functions is called before
			 * initscr/newterm */
			if (!initdone) {
				j = 0;
				while (j < nrcmds) {
					if (strcmp(func, restricted_commands[j]) == 0) {
						if (strcmp(func, "initscr") == 0  ||
							strcmp(func, "newterm") == 0)
							initdone = 1;
						/* matched function */
						commands[i].func(nargs, args);
						return;
					}
					j++;
				}
				report_status("YOU NEED TO CALL INITSCR/NEWTERM FIRST");
				return;
			}
			/* matched function */
			commands[i].func(nargs, args);
			return;
		}
		i++;
	}

	report_status("UNKNOWN_FUNCTION");
}

static void
write_to_director(const void *mem, size_t size)
{
	ssize_t nwritten = write(to_director, mem, size);
	if (nwritten == -1)
		err(1, "writing to director failed");
	if ((size_t)nwritten != size)
		errx(1, "short write to director, expected %zu, got %zd",
		    size, nwritten);
}

static void
write_to_director_int(int i)
{
	write_to_director(&i, sizeof(i));
}

static void
write_to_director_type(data_enum_t return_type)
{
	write_to_director_int(return_type);
}

/*
 * Report an pointer value back to the director
 */
void
report_ptr(void *ptr)
{
	char *string;

	if (ptr == NULL)
		asprintf(&string, "NULL");
	else
		asprintf(&string, "%p", ptr);
	report_status(string);
	free(string);
}

/*
 * Report an integer value back to the director
 */
void
report_int(int value)
{
	char *string;

	asprintf(&string, "%d", value);
	report_status(string);
	free(string);
}

/*
 * Report either an ERR or OK back to the director
 */
void
report_return(int status)
{
	if (status == ERR)
		write_to_director_type(data_err);
	else if (status == OK)
		write_to_director_type(data_ok);
	else if (status == KEY_CODE_YES)
		report_int(status);
	else
		report_status("INVALID_RETURN");
}

/*
 * Report the number of returns back to the director via the command pipe
 */
void
report_count(int count)
{
	write_to_director_type(data_count);
	write_to_director_int(count);
}

/*
 * Report the status back to the director via the command pipe
 */
void
report_status(const char *status)
{
	report_message(data_string, status);
}

/*
 * Report an error message back to the director via the command pipe.
 */
void
report_error(const char *status)
{
	report_message(data_slave_error, status);
}

/*
 * Report the message with the given type back to the director via the
 * command pipe.
 */
static void
report_message(data_enum_t type, const char *status)
{
	size_t len = strlen(status);
	write_to_director_type(type);
	write_to_director_int(len);
	write_to_director(status, len);
}

/*
 * Report a string of chtype back to the director via the command pipe.
 */
void
report_byte(chtype c)
{
	chtype string[2];

	string[0] = c;
	string[1] = A_NORMAL | '\0';
	report_nstr(string);
}

/*
 * Report a string of chtype back to the director via the command pipe.
 */
void
report_nstr(chtype *string)
{
	size_t size;
	chtype *p;

	for (p = string; (*p & __CHARTEXT) != 0; p++)
		continue;

	size = (size_t)(p + 1 - string) * sizeof(*p);

	write_to_director_type(data_byte);
	write_to_director_int(size);
	write_to_director(string, size);
}

/*
 * Report a cchar_t back to the director via the command pipe.
 */
void
report_cchar(cchar_t c)
{

	write_to_director_type(data_cchar);
	write_to_director_int(sizeof(c));
	write_to_director(&c, sizeof(c));
}

/*
 * Report a wchar_t back to the director via the command pipe.
 */
void
report_wchar(wchar_t ch)
{
	wchar_t wstr[2];

	wstr[0] = ch;
	wstr[1] = L'\0';
	report_wstr(wstr);
}


/*
 * Report a string of wchar_t back to the director via the command pipe.
 */
void
report_wstr(wchar_t *wstr)
{
	size_t size;
	wchar_t *p;

	for (p = wstr; *p != L'\0'; p++)
		continue;
	size = (size_t)(p + 1 - wstr) * sizeof(*p);


	write_to_director_type(data_wchar);
	write_to_director_int(size);
	write_to_director(wstr, size);
}

/*
 * Check the number of args we received are what we expect.  Return an
 * error if they do not match.
 */
int
check_arg_count(int nargs, int expected)
{
	if (nargs != expected) {
		report_count(1);
		report_error("INCORRECT_ARGUMENT_NUMBER");
		return 1;
	}

	return 0;
}