[moira.git] / clients / mrtest / mrtest.c

/* $Id $
 *
 * Bare-bones Moira client
 *
 * Copyright (C) 1987-1998 by the Massachusetts Institute of Technology
 * For copying and distribution information, please see the file
 * <mit-copyright.h>.
 */

#include <mit-copyright.h>
#include <moira.h>

#include <errno.h>
#include <fcntl.h>
#include <setjmp.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

#ifdef USE_READLINE
#include "readline.h"
#include "history.h"
#endif

RCSID("$Header$");

int recursion = 0, interactive;
extern int sending_version_no;
int count, quit = 0, cancel = 0;
char *whoami;
sigjmp_buf jb;

#define MAXARGS 20

void discard_input(void);
char *mr_gets(char *prompt, char *buf, size_t len);
void execute_line(char *cmdbuf);
int parse(char *buf, char *argv[MAXARGS]);
int print_reply(int argc, char **argv, void *hint);
void test_noop(void);
void test_connect(int argc, char **argv);
void test_disconnect(void);
void test_host(void);
void test_new(void);
void test_old(void);
void test_motd(void);
void test_query(int argc, char **argv);
void test_auth(void);
void test_access(int argc, char **argv);
void test_dcm(void);
void test_script(int argc, char **argv);
void test_list_requests(void);

int main(int argc, char **argv)
{
  char cmdbuf[BUFSIZ];
  struct sigaction action;

  whoami = argv[0];
  interactive = (isatty(0) && isatty(1));

  initialize_sms_error_table();
  initialize_krb_error_table();

#ifdef USE_READLINE
  /* we don't want filename completion */
  rl_bind_key('\t', rl_insert);
#endif

  action.sa_handler = discard_input;
  action.sa_flags = 0;
  sigemptyset(&action.sa_mask);
  sigaction(SIGINT, &action, NULL);
  sigsetjmp(jb, 1);

  while (!quit)
    {
      if (!mr_gets("moira:  ", cmdbuf, BUFSIZ))
	break;
      execute_line(cmdbuf);
    }
  mr_disconnect();
  exit(0);
}

void discard_input(void)
{
  putc('\n', stdout);

  /* if we're inside a script, we have to clean up file descriptors,
     so don't jump out yet */
  if (recursion)
    cancel = 1;
  else
    siglongjmp(jb, 1);
}

char *mr_gets(char *prompt, char *buf, size_t len)
{
  char *in;
#ifdef USE_READLINE
  if (interactive)
    {
      in = readline(prompt);

      if (!in)
	return NULL;
      if (*in)
	add_history(in);
      strncpy(buf, in, len - 1);
      buf[len] = 0;
      free(in);

      return buf;
    }
#endif
  printf("%s", prompt);
  fflush(stdout);
  in = fgets(buf, len, stdin);
  if (!in)
    return in;
  if (strchr(buf, '\n'))
    *(strchr(buf, '\n')) = '\0';
  return buf;
}

void execute_line(char *cmdbuf)
{
  int argc;
  char *argv[MAXARGS];

  argc = parse(cmdbuf, argv);
  if (argc == 0)
    return;
  if (!strcmp(argv[0], "noop"))
    test_noop();
  else if (!strcmp(argv[0], "connect") || !strcmp(argv[0], "c"))
    test_connect(argc, argv);
  else if (!strcmp(argv[0], "disconnect") || !strcmp(argv[0], "d"))
    test_disconnect();
  else if (!strcmp(argv[0], "host"))
    test_host();
  else if (!strcmp(argv[0], "new") || !strcmp(argv[0], "2"))
    test_new();
  else if (!strcmp(argv[0], "old") || !strcmp(argv[0], "1"))
    test_old();
  else if (!strcmp(argv[0], "motd"))
    test_motd();
  else if (!strcmp(argv[0], "query") || !strcmp(argv[0], "qy"))
    test_query(argc, argv);
  else if (!strcmp(argv[0], "auth") || !strcmp(argv[0], "a"))
    test_auth();
  else if (!strcmp(argv[0], "access"))
    test_access(argc, argv);
  else if (!strcmp(argv[0], "dcm"))
    test_dcm();
  else if (!strcmp(argv[0], "script") || !strcmp(argv[0], "s"))
    test_script(argc, argv);
  else if (!strcmp(argv[0], "list_requests") ||
	  !strcmp(argv[0], "lr") || !strcmp(argv[0], "?"))
    test_list_requests();
  else if (!strcmp(argv[0], "quit") || !strcmp(argv[0], "Q"))
    quit = 1;
  else
    {
      fprintf(stderr, "moira: Unknown request \"%s\".  "
	      "Type \"?\" for a request list.\n", argv[0]);
    }
}

int parse(char *buf, char *argv[MAXARGS])
{
  char *p;
  int argc, num;

  if (!*buf)
    return 0;

  for (p = buf, argc = 0, argv[0] = buf; *p && *p != '\n'; p++)
    {
      if (*p == '"')
	{
	  char *d = p++;
	  /* skip to close-quote, copying back over open-quote */
	  while (*p != '"')
	    {
	      if (!*p || *p == '\n')
		{
		  fprintf(stderr,
			  "moira: Unbalanced quotes in command line\n");
		  return 0;
		}
	      /* deal with \### or \\ */
	      if (*p == '\\')
		{
		  if (*++p != '"' && (*p < '0' || *p > '9') && (*p != '\\'))
		    {
		      fprintf(stderr, "moira: Bad use of \\\n");
		      return 0;
		    }
		  else if (*p >= '0' && *p <= '9')
		    {
		      num = (*p - '0') * 64 + (*++p - '0') * 8 + (*++p - '0');
		      *p = num;
		    }
		}
	      *d++ = *p++;
	    }
	  if (p == d + 1)
	    {
	      *d = '\0';
	      p++;
	    }
	  else
	    {
	      while (p >= d)
		*p-- = ' ';
	    }
	}
      if (*p == ' ' || *p == '\t')
	{
	  /* skip whitespace */
	  for (*p++ = '\0'; *p == ' ' || *p == '\t'; p++)
	    ;
	  if (*p && *p != '\n')
	    argv[++argc] = p--;
	}
    }
  if (*p == '\n')
    *p = '\0';
  return argc + 1;
}

void test_noop(void)
{
  int status = mr_noop();
  if (status)
    com_err("moira (noop)", status, "");
}

void test_new(void)
{
  sending_version_no = MR_VERSION_2;
}

void test_old(void)
{
  sending_version_no = MR_VERSION_1;
}

void test_connect(int argc, char *argv[])
{
  char *server = "";
  int status;

  if (argc > 1)
    server = argv[1];
  status = mr_connect(server);
  if (status)
    com_err("moira (connect)", status, "");
}

void test_disconnect(void)
{
  int status = mr_disconnect();
  if (status)
    com_err("moira (disconnect)", status, "");
}

void test_host(void)
{
  char host[BUFSIZ];
  int status;

  memset(host, 0, sizeof(host));

  if ((status = mr_host(host, sizeof(host) - 1)))
    com_err("moira (host)", status, "");
  else
    printf("You are connected to host %s\n", host);
}

void test_auth(void)
{
  int status;

  status = mr_auth("mrtest");
  if (status)
    com_err("moira (auth)", status, "");
}

void test_script(int argc, char *argv[])
{
  FILE *inp;
  char input[BUFSIZ], *cp;
  int status, oldstdout, oldstderr;

  if (recursion > 8)
    {
      com_err("moira (script)", 0, "too many levels deep in script files\n");
      return;
    }

  if (argc < 2)
    {
      com_err("moira (script)", 0, "Usage: script input_file [ output_file ]");
      return;
    }

  inp = fopen(argv[1], "r");
  if (!inp)
    {
      sprintf(input, "Cannot open input file %s", argv[1]);
      com_err("moira (script)", 0, input);
      return;
    }

  if (argc == 3)
    {
      printf("Redirecting output to %s\n", argv[2]);
      fflush(stdout);
      oldstdout = dup(1);
      close(1);
      status = open(argv[2], O_CREAT|O_WRONLY|O_APPEND, 0664);
      if (status != 1)
	{
	  close(status);
	  dup2(oldstdout, 1);
	  argc = 2;
	  sprintf(input, "Unable to redirect output to %s\n", argv[2]);
	  com_err("moira (script)", errno, input);
	}
      else
	{
	  fflush(stderr);
	  oldstderr = dup(2);
	  close(2);
	  dup2(1, 2);
	}
    }

  recursion++;

  while (!cancel)
    {
      if (!fgets(input, BUFSIZ, inp))
	break;
      if ((cp = strchr(input, '\n')))
	*cp = '\0';
      if (input[0] == 0)
	{
	  printf("\n");
	  continue;
	}
      if (input[0] == '%')
	{
	  for (cp = &input[1]; *cp && isspace(*cp); cp++)
	    ;
	  printf("Comment: %s\n", cp);
	  continue;
	}
      printf("Executing: %s\n", input);
      execute_line(input);
    }

  recursion--;
  if (!recursion)
    cancel = 0;

  fclose(inp);
  if (argc == 3)
    {
      fflush(stdout);
      close(1);
      dup2(oldstdout, 1);
      close(oldstdout);
      fflush(stderr);
      close(2);
      dup2(oldstderr, 2);
      close(oldstderr);
    }
}

int print_reply(int argc, char **argv, void *hint)
{
  int i;
  for (i = 0; i < argc; i++)
    {
      if (i != 0)
	printf(", ");
      printf("%s", argv[i]);
    }
  printf("\n");
  count++;
  return MR_CONT;
}

void test_query(int argc, char **argv)
{
  int status;
  sigset_t sigs;

  if (argc < 2)
    {
      com_err("moira (query)", 0, "Usage: query handle [ args ... ]");
      return;
    }

  count = 0;
  /* Don't allow ^C during the query: it will confuse libmoira's
     internal state. (Yay static variables) */
  sigemptyset(&sigs);
  sigaddset(&sigs, SIGINT);
  sigprocmask(SIG_BLOCK, &sigs, NULL);
  status = mr_query(argv[1], argc - 2, argv + 2, print_reply, NULL);
  sigprocmask(SIG_UNBLOCK, &sigs, NULL);
  printf("%d tuple%s\n", count, ((count == 1) ? "" : "s"));
  if (status)
    com_err("moira (query)", status, "");
}

void test_access(int argc, char **argv)
{
  int status;
  if (argc < 2)
    {
      com_err("moira (access)", 0, "Usage: access handle [ args ... ]");
      return;
    }
  status = mr_access(argv[1], argc - 2, argv + 2);
  if (status)
    com_err("moira (access)", status, "");
}

void test_dcm()
{
  int status;

  if ((status = mr_do_update()))
    com_err("moira (dcm)", status, " while triggering dcm");
}

void test_motd()
{
  int status;
  char *motd;

  if ((status = mr_motd(&motd)))
    com_err("moira (motd)", status, " while getting motd");
  if (motd)
    printf("%s\n", motd);
  else
    printf("No message of the day.\n");
}

void test_list_requests(void)
{
  printf("Available moira requests:\n");
  printf("\n");
  printf("noop\t\t\tAsk Moira to do nothing\n");
  printf("connect, c\t\tConnect to Moira server\n");
  printf("disconnect, d\t\tDisconnect from server\n");
  printf("host\t\t\tIdentify the server host\n");
  printf("new, 2\t\t\tUse new protocol\n");
  printf("old, 1\t\t\tUse old protocol\n");
  printf("motd, m\t\t\tGet the Message of the Day\n");
  printf("query, qy\t\tMake a query.\n");
  printf("auth, a\t\t\tAuthenticate to Moira.\n");
  printf("access\t\t\tCheck access to a Moira query.\n");
  printf("dcm\t\t\tTrigger the DCM\n");
  printf("script, s\t\tRead commands from a script.\n");
  printf("list_requests, lr, ?\tList available commands.\n");
  printf("quit, Q\t\t\tLeave the subsystem.\n");
}
Commit	Line	Data
7ac48069	1	/* $Id $
6e6374cb	2	*
7ac48069	3	* Bare-bones Moira client
6e6374cb	4	*
7ac48069	5	* Copyright (C) 1987-1998 by the Massachusetts Institute of Technology
	6	* For copying and distribution information, please see the file
	7	* <mit-copyright.h>.
6e6374cb	8	*/
6e6374cb	9
c801de4c	10	#include <mit-copyright.h>
8defc06b	11	#include <moira.h>
7ac48069	12
	13	#include <errno.h>
	14	#include <fcntl.h>
f281aa29	15	#include <setjmp.h>
f281aa29	16	#include <signal.h>
7ac48069	17	#include <stdio.h>
	18	#include <stdlib.h>
	19	#include <string.h>
	20	#include <unistd.h>
f281aa29	21
	22	#ifdef USE_READLINE
	23	#include "readline.h"
7ac48069	24	#include "history.h"
f281aa29	25	#endif
6e6374cb	26
7ac48069	27	RCSID("$Header$");
7ac48069	28
fcb0b7fa	29	int recursion = 0, interactive;
59af7b90	30	extern int sending_version_no;
5eaef520	31	int count, quit = 0, cancel = 0;
1a5c774c	32	char *whoami;
74c84b52	33	sigjmp_buf jb;
1a5c774c	34
1a5c774c	35	#define MAXARGS 20
6e6374cb	36
7ac48069	37	void discard_input(void);
	38	char mr_gets(char prompt, char *buf, size_t len);
	39	void execute_line(char *cmdbuf);
	40	int parse(char buf, char argv[MAXARGS]);
	41	int print_reply(int argc, char *argv, void hint);
	42	void test_noop(void);
	43	void test_connect(int argc, char **argv);
	44	void test_disconnect(void);
	45	void test_host(void);
	46	void test_new(void);
	47	void test_old(void);
	48	void test_motd(void);
	49	void test_query(int argc, char **argv);
	50	void test_auth(void);
	51	void test_access(int argc, char **argv);
	52	void test_dcm(void);
	53	void test_script(int argc, char **argv);
	54	void test_list_requests(void);
f281aa29	55
5eaef520	56	int main(int argc, char **argv)
	57	{
	58	char cmdbuf[BUFSIZ];
	59	struct sigaction action;
6e6374cb	60
5eaef520	61	whoami = argv[0];
5eaef520	62	interactive = (isatty(0) && isatty(1));
3442308f	63
5eaef520	64	initialize_sms_error_table();
	65	initialize_krb_error_table();
	66
	67	#ifdef USE_READLINE
	68	/* we don't want filename completion */
	69	rl_bind_key('\t', rl_insert);
74c84b52	70	#endif
f281aa29	71
5eaef520	72	action.sa_handler = discard_input;
	73	action.sa_flags = 0;
	74	sigemptyset(&action.sa_mask);
	75	sigaction(SIGINT, &action, NULL);
	76	sigsetjmp(jb, 1);
	77
	78	while (!quit)
	79	{
	80	if (!mr_gets("moira: ", cmdbuf, BUFSIZ))
	81	break;
	82	execute_line(cmdbuf);
	83	}
	84	mr_disconnect();
	85	exit(0);
1a5c774c	86	}
1a5c774c	87
5eaef520	88	void discard_input(void)
f281aa29	89	{
f281aa29	90	putc('\n', stdout);
fcb0b7fa	91
	92	/* if we're inside a script, we have to clean up file descriptors,
	93	so don't jump out yet */
5eaef520	94	if (recursion)
	95	cancel = 1;
	96	else
fcb0b7fa	97	siglongjmp(jb, 1);
f281aa29	98	}
f281aa29	99
5eaef520	100	char mr_gets(char prompt, char *buf, size_t len)
f281aa29	101	{
	102	char *in;
	103	#ifdef USE_READLINE
5eaef520	104	if (interactive)
	105	{
	106	in = readline(prompt);
	107
	108	if (!in)
	109	return NULL;
	110	if (*in)
	111	add_history(in);
	112	strncpy(buf, in, len - 1);
	113	buf[len] = 0;
	114	free(in);
	115
	116	return buf;
f281aa29	117	}
f281aa29	118	#endif
	119	printf("%s", prompt);
	120	fflush(stdout);
5eaef520	121	in = fgets(buf, len, stdin);
	122	if (!in)
	123	return in;
	124	if (strchr(buf, '\n'))
	125	*(strchr(buf, '\n')) = '\0';
f281aa29	126	return buf;
	127	}
	128
7ac48069	129	void execute_line(char *cmdbuf)
1a5c774c	130	{
	131	int argc;
	132	char *argv[MAXARGS];
	133
5eaef520	134	argc = parse(cmdbuf, argv);
	135	if (argc == 0)
	136	return;
	137	if (!strcmp(argv[0], "noop"))
1a5c774c	138	test_noop();
5eaef520	139	else if (!strcmp(argv[0], "connect") \|\| !strcmp(argv[0], "c"))
1a5c774c	140	test_connect(argc, argv);
5eaef520	141	else if (!strcmp(argv[0], "disconnect") \|\| !strcmp(argv[0], "d"))
1a5c774c	142	test_disconnect();
5eaef520	143	else if (!strcmp(argv[0], "host"))
1a5c774c	144	test_host();
5eaef520	145	else if (!strcmp(argv[0], "new") \|\| !strcmp(argv[0], "2"))
1a5c774c	146	test_new();
5eaef520	147	else if (!strcmp(argv[0], "old") \|\| !strcmp(argv[0], "1"))
1a5c774c	148	test_old();
5eaef520	149	else if (!strcmp(argv[0], "motd"))
1a5c774c	150	test_motd();
5eaef520	151	else if (!strcmp(argv[0], "query") \|\| !strcmp(argv[0], "qy"))
1a5c774c	152	test_query(argc, argv);
5eaef520	153	else if (!strcmp(argv[0], "auth") \|\| !strcmp(argv[0], "a"))
7ac48069	154	test_auth();
5eaef520	155	else if (!strcmp(argv[0], "access"))
1a5c774c	156	test_access(argc, argv);
5eaef520	157	else if (!strcmp(argv[0], "dcm"))
1a5c774c	158	test_dcm();
5eaef520	159	else if (!strcmp(argv[0], "script") \|\| !strcmp(argv[0], "s"))
1a5c774c	160	test_script(argc, argv);
5eaef520	161	else if (!strcmp(argv[0], "list_requests") \|\|
5eaef520	162	!strcmp(argv[0], "lr") \|\| !strcmp(argv[0], "?"))
1a5c774c	163	test_list_requests();
5eaef520	164	else if (!strcmp(argv[0], "quit") \|\| !strcmp(argv[0], "Q"))
	165	quit = 1;
	166	else
	167	{
	168	fprintf(stderr, "moira: Unknown request \"%s\". "
	169	"Type \"?\" for a request list.\n", argv[0]);
	170	}
1a5c774c	171	}
1a5c774c	172
5eaef520	173	int parse(char buf, char argv[MAXARGS])
1a5c774c	174	{
	175	char *p;
	176	int argc, num;
5eaef520	177
	178	if (!*buf)
	179	return 0;
	180
	181	for (p = buf, argc = 0, argv[0] = buf; p && p != '\n'; p++)
	182	{
	183	if (*p == '"')
	184	{
	185	char *d = p++;
	186	/* skip to close-quote, copying back over open-quote */
	187	while (*p != '"')
	188	{
	189	if (!p \|\| p == '\n')
	190	{
	191	fprintf(stderr,
	192	"moira: Unbalanced quotes in command line\n");
	193	return 0;
	194	}
	195	/* deal with \### or \\ */
	196	if (*p == '\\')
	197	{
	198	if (++p != '"' && (p < '0' \|\| p > '9') && (p != '\\'))
	199	{
	200	fprintf(stderr, "moira: Bad use of \\\n");
	201	return 0;
	202	}
	203	else if (p >= '0' && p <= '9')
	204	{
	205	num = (p - '0') 64 + (++p - '0') 8 + (*++p - '0');
	206	*p = num;
	207	}
	208	}
	209	d++ = p++;
	210	}
	211	if (p == d + 1)
	212	{
	213	*d = '\0';
	214	p++;
	215	}
	216	else
	217	{
	218	while (p >= d)
	219	*p-- = ' ';
	220	}
2c031a2a	221	}
5eaef520	222	if (p == ' ' \|\| p == '\t')
	223	{
	224	/* skip whitespace */
	225	for (p++ = '\0'; p == ' ' \|\| *p == '\t'; p++)
	226	;
	227	if (p && p != '\n')
	228	argv[++argc] = p--;
6e6374cb	229	}
1a5c774c	230	}
5eaef520	231	if (*p == '\n')
	232	*p = '\0';
	233	return argc + 1;
6e6374cb	234	}
6e6374cb	235
7ac48069	236	void test_noop(void)
6e6374cb	237	{
5eaef520	238	int status = mr_noop();
	239	if (status)
	240	com_err("moira (noop)", status, "");
6e6374cb	241	}
6e6374cb	242
7ac48069	243	void test_new(void)
59af7b90	244	{
5eaef520	245	sending_version_no = MR_VERSION_2;
59af7b90	246	}
59af7b90	247
7ac48069	248	void test_old(void)
59af7b90	249	{
5eaef520	250	sending_version_no = MR_VERSION_1;
59af7b90	251	}
59af7b90	252
7ac48069	253	void test_connect(int argc, char *argv[])
6e6374cb	254	{
5eaef520	255	char *server = "";
	256	int status;
	257
	258	if (argc > 1)
	259	server = argv[1];
	260	status = mr_connect(server);
	261	if (status)
	262	com_err("moira (connect)", status, "");
6e6374cb	263	}
6e6374cb	264
7ac48069	265	void test_disconnect(void)
6e6374cb	266	{
5eaef520	267	int status = mr_disconnect();
	268	if (status)
	269	com_err("moira (disconnect)", status, "");
6e6374cb	270	}
6e6374cb	271
7ac48069	272	void test_host(void)
66848930	273	{
5eaef520	274	char host[BUFSIZ];
5eaef520	275	int status;
66848930	276
5eaef520	277	memset(host, 0, sizeof(host));
66848930	278
5eaef520	279	if ((status = mr_host(host, sizeof(host) - 1)))
	280	com_err("moira (host)", status, "");
	281	else
	282	printf("You are connected to host %s\n", host);
66848930	283	}
66848930	284
7ac48069	285	void test_auth(void)
6e6374cb	286	{
5eaef520	287	int status;
59af7b90	288
5eaef520	289	status = mr_auth("mrtest");
	290	if (status)
	291	com_err("moira (auth)", status, "");
6e6374cb	292	}
6e6374cb	293
7ac48069	294	void test_script(int argc, char *argv[])
58b825a4	295	{
5eaef520	296	FILE *inp;
	297	char input[BUFSIZ], *cp;
	298	int status, oldstdout, oldstderr;
	299
	300	if (recursion > 8)
	301	{
	302	com_err("moira (script)", 0, "too many levels deep in script files\n");
	303	return;
58b825a4	304	}
58b825a4	305
5eaef520	306	if (argc < 2)
	307	{
	308	com_err("moira (script)", 0, "Usage: script input_file [ output_file ]");
	309	return;
58b825a4	310	}
58b825a4	311
5eaef520	312	inp = fopen(argv[1], "r");
	313	if (!inp)
	314	{
	315	sprintf(input, "Cannot open input file %s", argv[1]);
	316	com_err("moira (script)", 0, input);
	317	return;
58b825a4	318	}
58b825a4	319
5eaef520	320	if (argc == 3)
	321	{
	322	printf("Redirecting output to %s\n", argv[2]);
	323	fflush(stdout);
	324	oldstdout = dup(1);
	325	close(1);
	326	status = open(argv[2], O_CREAT\|O_WRONLY\|O_APPEND, 0664);
	327	if (status != 1)
	328	{
	329	close(status);
	330	dup2(oldstdout, 1);
	331	argc = 2;
	332	sprintf(input, "Unable to redirect output to %s\n", argv[2]);
	333	com_err("moira (script)", errno, input);
	334	}
	335	else
	336	{
	337	fflush(stderr);
	338	oldstderr = dup(2);
	339	close(2);
	340	dup2(1, 2);
58b825a4	341	}
	342	}
	343
5eaef520	344	recursion++;
	345
	346	while (!cancel)
	347	{
	348	if (!fgets(input, BUFSIZ, inp))
	349	break;
	350	if ((cp = strchr(input, '\n')))
	351	*cp = '\0';
	352	if (input[0] == 0)
	353	{
	354	printf("\n");
	355	continue;
6a592314	356	}
5eaef520	357	if (input[0] == '%')
	358	{
	359	for (cp = &input[1]; cp && isspace(cp); cp++)
	360	;
	361	printf("Comment: %s\n", cp);
	362	continue;
58b825a4	363	}
5eaef520	364	printf("Executing: %s\n", input);
5eaef520	365	execute_line(input);
58b825a4	366	}
58b825a4	367
5eaef520	368	recursion--;
	369	if (!recursion)
	370	cancel = 0;
	371
	372	fclose(inp);
	373	if (argc == 3)
	374	{
	375	fflush(stdout);
	376	close(1);
	377	dup2(oldstdout, 1);
	378	close(oldstdout);
	379	fflush(stderr);
	380	close(2);
	381	dup2(oldstderr, 2);
	382	close(oldstderr);
58b825a4	383	}
	384	}
	385
7ac48069	386	int print_reply(int argc, char *argv, void hint)
6e6374cb	387	{
5eaef520	388	int i;
	389	for (i = 0; i < argc; i++)
	390	{
	391	if (i != 0)
	392	printf(", ");
	393	printf("%s", argv[i]);
	394	}
	395	printf("\n");
	396	count++;
	397	return MR_CONT;
6e6374cb	398	}
6e6374cb	399
7ac48069	400	void test_query(int argc, char **argv)
6e6374cb	401	{
5eaef520	402	int status;
5eaef520	403	sigset_t sigs;
fcb0b7fa	404
5eaef520	405	if (argc < 2)
	406	{
	407	com_err("moira (query)", 0, "Usage: query handle [ args ... ]");
	408	return;
	409	}
59af7b90	410
5eaef520	411	count = 0;
	412	/* Don't allow ^C during the query: it will confuse libmoira's
	413	internal state. (Yay static variables) */
	414	sigemptyset(&sigs);
	415	sigaddset(&sigs, SIGINT);
	416	sigprocmask(SIG_BLOCK, &sigs, NULL);
	417	status = mr_query(argv[1], argc - 2, argv + 2, print_reply, NULL);
	418	sigprocmask(SIG_UNBLOCK, &sigs, NULL);
	419	printf("%d tuple%s\n", count, ((count == 1) ? "" : "s"));
	420	if (status)
	421	com_err("moira (query)", status, "");
6e6374cb	422	}
6e6374cb	423
7ac48069	424	void test_access(int argc, char **argv)
6e6374cb	425	{
5eaef520	426	int status;
	427	if (argc < 2)
	428	{
	429	com_err("moira (access)", 0, "Usage: access handle [ args ... ]");
	430	return;
	431	}
	432	status = mr_access(argv[1], argc - 2, argv + 2);
	433	if (status)
	434	com_err("moira (access)", status, "");
6e6374cb	435	}
fc3ac28e	436
7ac48069	437	void test_dcm()
fc3ac28e	438	{
5eaef520	439	int status;
fc3ac28e	440
5eaef520	441	if ((status = mr_do_update()))
5eaef520	442	com_err("moira (dcm)", status, " while triggering dcm");
fc3ac28e	443	}
093b57b1	444
7ac48069	445	void test_motd()
093b57b1	446	{
5eaef520	447	int status;
	448	char *motd;
	449
	450	if ((status = mr_motd(&motd)))
	451	com_err("moira (motd)", status, " while getting motd");
	452	if (motd)
	453	printf("%s\n", motd);
	454	else
	455	printf("No message of the day.\n");
093b57b1	456	}
1a5c774c	457
7ac48069	458	void test_list_requests(void)
1a5c774c	459	{
5eaef520	460	printf("Available moira requests:\n");
	461	printf("\n");
	462	printf("noop\t\t\tAsk Moira to do nothing\n");
	463	printf("connect, c\t\tConnect to Moira server\n");
	464	printf("disconnect, d\t\tDisconnect from server\n");
	465	printf("host\t\t\tIdentify the server host\n");
	466	printf("new, 2\t\t\tUse new protocol\n");
	467	printf("old, 1\t\t\tUse old protocol\n");
	468	printf("motd, m\t\t\tGet the Message of the Day\n");
	469	printf("query, qy\t\tMake a query.\n");
	470	printf("auth, a\t\t\tAuthenticate to Moira.\n");
	471	printf("access\t\t\tCheck access to a Moira query.\n");
	472	printf("dcm\t\t\tTrigger the DCM\n");
	473	printf("script, s\t\tRead commands from a script.\n");
	474	printf("list_requests, lr, ?\tList available commands.\n");
	475	printf("quit, Q\t\t\tLeave the subsystem.\n");
1a5c774c	476	}