diff options
Diffstat (limited to 'tests/pipeconnect.c')
| -rw-r--r-- | tests/pipeconnect.c | 397 |
1 files changed, 397 insertions, 0 deletions
diff --git a/tests/pipeconnect.c b/tests/pipeconnect.c new file mode 100644 index 0000000..ddb4a06 --- /dev/null +++ b/tests/pipeconnect.c @@ -0,0 +1,397 @@ +/* pipeconnect.c - Check the assuan_pipe_connect call. + Copyright (C) 2006, 2009, 2010 Free Software Foundation, Inc. + + This file is part of Assuan. + + Assuan is free software; you can redistribute it and/or modify it + under the terms of the GNU Lesser General Public License as + published by the Free Software Foundation; either version 3 of + the License, or (at your option) any later version. + + Assuan is distributed in the hope that it will be useful, but + WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + Lesser General Public License for more details. + + You should have received a copy of the GNU Lesser General Public + License along with this program; if not, see <http://www.gnu.org/licenses/>. + */ + +/* + This tests creates a program which starts an assuan server and runs + some simple tests on it. The other program is actually the same + program but called with the option --server. +*/ + +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif + +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <assert.h> +#include <errno.h> + +#include "../src/assuan.h" +#include "common.h" + +static assuan_fd_t my_stdin = ASSUAN_INVALID_FD; +static assuan_fd_t my_stdout = ASSUAN_INVALID_FD; +static assuan_fd_t my_stderr = ASSUAN_INVALID_FD; + + +static gpg_error_t +cmd_echo (assuan_context_t ctx, char *line) +{ + log_info ("got ECHO command (%s)\n", line); + + assuan_send_data (ctx, line, strlen (line)); + + return 0; +} + + +static gpg_error_t +cmd_cat (assuan_context_t ctx, char *line) +{ + assuan_fd_t fd, fdout; + int c; + FILE *fp, *fpout; + int nbytes; + + log_info ("got CAT command (%s)\n", line); + + fd = assuan_get_input_fd (ctx); + if (fd == ASSUAN_INVALID_FD) + return gpg_error (GPG_ERR_ASS_NO_INPUT); + fdout = assuan_get_output_fd (ctx); + if (fdout == ASSUAN_INVALID_FD) + return gpg_error (GPG_ERR_ASS_NO_OUTPUT); + fp = fdopen (fd, "r"); + if (!fp) + { + log_error ("fdopen failed on input fd: %s\n", strerror (errno)); + return gpg_error (GPG_ERR_ASS_GENERAL); + } + + fpout = fdopen (fdout, "w"); + if (!fpout) + { + log_error ("fdopen failed on output fd: %s\n", strerror (errno)); + fclose (fp); + return gpg_error (GPG_ERR_ASS_GENERAL); + } + + nbytes = 0; + while ( (c=getc (fp)) != -1) + { + putc (c, fpout); + nbytes++; + } + log_info ("done printing %d bytes to output fd\n", nbytes); + + /* Fixme: This also closes the original fd. */ + fclose (fp); + fclose (fpout); + return 0; +} + + +static gpg_error_t +server_register_commands (assuan_context_t ctx) +{ + static struct + { + const char *name; + gpg_error_t (*handler) (assuan_context_t, char *line); + } table[] = + { + { "ECHO", cmd_echo }, + { "CAT", cmd_cat }, + { "INPUT", NULL }, + { "OUTPUT", NULL }, + { NULL, NULL } + }; + int i; + gpg_error_t rc; + + for (i=0; table[i].name; i++) + { + rc = assuan_register_command (ctx, table[i].name, table[i].handler, NULL); + if (rc) + return rc; + } + return 0; +} + + +static void +run_server (int enable_debug) +{ + int rc; + assuan_context_t ctx; + assuan_fd_t filedes[2]; + + filedes[0] = my_stdin; + filedes[1] = my_stdout; + + rc = assuan_new (&ctx); + if (rc) + log_fatal ("assuan_new failed: %s\n", gpg_strerror (rc)); + + rc = assuan_init_pipe_server (ctx, filedes); + if (rc) + log_fatal ("assuan_init_pipe_server failed: %s\n", + gpg_strerror (rc)); + + rc = server_register_commands (ctx); + if (rc) + log_fatal ("register_commands failed: %s\n", gpg_strerror(rc)); + + if (enable_debug) + assuan_set_log_stream (ctx, stderr); + + for (;;) + { + rc = assuan_accept (ctx); + if (rc) + { + if (rc != -1) + log_error ("assuan_accept failed: %s\n", gpg_strerror (rc)); + break; + } + + log_info ("client connected. Client's pid is %ld\n", + (long)assuan_get_pid (ctx)); + + rc = assuan_process (ctx); + if (rc) + log_error ("assuan_process failed: %s\n", gpg_strerror (rc)); + } + + assuan_release (ctx); +} + + + + + + + +static gpg_error_t +data_cb (void *opaque, const void *buffer, size_t length) +{ + (void)opaque; + + if (buffer) + printf ("Received data `%.*s'\n", (int)length, (char*)buffer); + return 0; +} + + +static void +run_client (const char *servername) +{ + gpg_error_t err; + assuan_context_t ctx; + assuan_fd_t no_close_fds[2]; + const char *arglist[5]; + + no_close_fds[0] = fileno (stderr); + no_close_fds[1] = ASSUAN_INVALID_FD; + + arglist[0] = servername; + arglist[1] = "--server"; + arglist[2] = debug? "--debug" : verbose? "--verbose":NULL; + arglist[3] = NULL; + + err = assuan_new (&ctx); + if (err) + log_fatal ("assuan_new failed: %s\n", gpg_strerror (err)); + + err = assuan_pipe_connect (ctx, servername, arglist, no_close_fds, + NULL, NULL, 0); + if (err) + { + log_error ("assuan_pipe_connect failed: %s\n", + gpg_strerror (err)); + assuan_release (ctx); + return; + } + + log_info ("server started; pid is %ld\n", + (long)assuan_get_pid (ctx)); + + err = assuan_transact (ctx, "ECHO Your lucky number is 3552664958674928. " + "Watch for it everywhere.", + data_cb, NULL, NULL, NULL, NULL, NULL); + if (err) + { + log_error ("sending ECHO failed: %s\n", gpg_strerror (err)); + return; + } + + err = assuan_transact (ctx, "BYE", NULL, NULL, NULL, NULL, NULL, NULL); + if (err) + { + log_error ("sending BYE failed: %s\n", gpg_strerror (err)); + return; + } + + assuan_release (ctx); + return; +} + + +static void +parse_std_file_handles (int *argcp, char ***argvp) +{ +#ifdef HAVE_W32CE_SYSTEM + int argc = *argcp; + char **argv = *argvp; + const char *s; + assuan_fd_t fd; + int i; + int fixup = 0; + + if (!argc) + return; + + for (argc--, argv++; argc; argc--, argv++) + { + s = *argv; + if (*s == '-' && s[1] == '&' && s[2] == 'S' + && (s[3] == '0' || s[3] == '1' || s[3] == '2') + && s[4] == '=' + && (strchr ("-01234567890", s[5]) || !strcmp (s+5, "null"))) + { + if (s[5] == 'n') + fd = ASSUAN_INVALID_FD; + else + fd = _assuan_w32ce_finish_pipe (atoi (s+5), s[3] != '0'); + switch (s[3] - '0') + { + case 0: my_stdin = fd; break; + case 1: my_stdout = fd; break; + case 2: my_stderr = fd; break; + } + + fixup++; + } + else + break; + } + + if (fixup) + { + argc = *argcp; + argc -= fixup; + *argcp = argc; + + argv = *argvp; + for (i=1; i < argc; i++) + argv[i] = argv[i + fixup]; + for (; i < argc + fixup; i++) + argv[i] = NULL; + } +#else + (void)argcp; + (void)argvp; + my_stdin = 0; + my_stdout = 1; + my_stderr = 2; +#endif +} + + +/* + M A I N + */ +int +main (int argc, char **argv) +{ + gpg_error_t err; + const char *myname = "no-pgm"; + int last_argc = -1; + int server = 0; + int silent_client = 0; + int silent_server = 0; + + parse_std_file_handles (&argc, &argv); + if (argc) + { + myname = *argv; + log_set_prefix (*argv); + argc--; argv++; + } + while (argc && last_argc != argc ) + { + last_argc = argc; + if (!strcmp (*argv, "--help")) + { + printf ("usage: %s [options]\n" + "\n" + "Options:\n" + " --verbose Show what is going on\n" + " --server Run in server mode\n", + log_get_prefix ()); + exit (0); + } + if (!strcmp (*argv, "--verbose")) + { + verbose = 1; + argc--; argv++; + } + else if (!strcmp (*argv, "--debug")) + { + verbose = debug = 1; + argc--; argv++; + } + else if (!strcmp (*argv, "--server")) + { + server = 1; + argc--; argv++; + } + else if (!strcmp (*argv, "--silent-server")) + { + silent_server = 1; + argc--; argv++; + } + else if (!strcmp (*argv, "--silent-client")) + { + silent_client = 1; + argc--; argv++; + } + else + log_fatal ("invalid option `%s' (try --help)\n", *argv); + } + + log_set_prefix (xstrconcat (log_get_prefix (), + server? ".server":".client", NULL)); + assuan_set_assuan_log_prefix (log_get_prefix ()); + + err = assuan_sock_init (); + if (err) + log_fatal ("socket init failed: %s\n", gpg_strerror (err)); + + if (server) + { + log_info ("server started\n"); + if (debug && !silent_server) + assuan_set_assuan_log_stream (stderr); + run_server (debug && !silent_server); + log_info ("server finished\n"); + } + else + { + log_info ("client started\n"); + if (debug && !silent_client) + assuan_set_assuan_log_stream (stderr); + run_client (myname); + log_info ("client finished\n"); + } + + return errorcount ? 1 : 0; +} + |