More chnages to use estream. Add a way to replace the standard
[gnupg.git] / tools / gpg-connect-agent.c
index ba97c1a..2bd5f85 100644 (file)
@@ -49,6 +49,7 @@ enum cmd_and_opt_values
     oQuiet      = 'q',
     oVerbose   = 'v',
     oRawSocket  = 'S',
+    oTcpSocket  = 'T',
     oExec       = 'E',
     oRun        = 'r',
     oSubst      = 's',
@@ -72,6 +73,8 @@ static ARGPARSE_OPTS opts[] = {
   ARGPARSE_s_n (oDecode,"decode",    N_("decode received data lines")),
   ARGPARSE_s_s (oRawSocket, "raw-socket", 
                 N_("|NAME|connect to Assuan socket NAME")),
+  ARGPARSE_s_s (oTcpSocket, "tcp-socket", 
+                N_("|ADDR|connect to Assuan server at ADDR")),
   ARGPARSE_s_n (oExec, "exec", 
                 N_("run the Assuan server given on the command line")),
   ARGPARSE_s_n (oNoExtConnect, "no-ext-connect",
@@ -96,6 +99,7 @@ struct
   int hex;              /* Print data lines in hex format. */
   int decode;           /* Decode received data lines.  */
   const char *raw_socket; /* Name of socket to connect in raw mode. */
+  const char *tcp_socket; /* Name of server to connect in tcp mode. */
   int exec;             /* Run the pgm given on the command line. */
   unsigned int connect_flags;    /* Flags used for connecting. */
   int enable_varsubst;  /* Set if variable substitution is enabled.  */
@@ -159,7 +163,8 @@ static struct
 
 /*-- local prototypes --*/
 static char *substitute_line_copy (const char *buffer);
-static int read_and_print_response (assuan_context_t ctx, int *r_goterr);
+static int read_and_print_response (assuan_context_t ctx, int withhash,
+                                    int *r_goterr);
 static assuan_context_t start_agent (void);
 
 
@@ -177,8 +182,8 @@ my_strusage( int level )
       break;
     case 13: p = VERSION; break;
     case 17: p = PRINTABLE_OS_NAME; break;
-    case 19: p = _("Please report bugs to <" PACKAGE_BUGREPORT ">.\n");
-      break;
+    case 19: p = _("Please report bugs to <@EMAIL@>.\n"); break;
+
     case 1:
     case 40: p = _("Usage: gpg-connect-agent [options] (-h for help)");
       break;
@@ -215,7 +220,7 @@ gnu_getcwd (void)
 }
 
 
-/* Unescale STRING and returned the malloced result.  The surrounding
+/* Unescape STRING and returned the malloced result.  The surrounding
    quotes must already be removed from STRING.  */
 static char *
 unescape_string (const char *string)
@@ -1086,7 +1091,7 @@ do_showopen (void)
 
 
 
-static int
+static gpg_error_t
 getinfo_pid_cb (void *opaque, const void *buffer, size_t length)
 {
   membuf_t *mb = opaque;
@@ -1153,13 +1158,13 @@ main (int argc, char **argv)
 
   /* Make sure that our subsystems are ready.  */
   i18n_init();
-  init_common_subsystems ();
+  init_common_subsystems (&argc, &argv);
 
-  assuan_set_assuan_err_source (0);
+  assuan_set_gpg_err_source (0);
 
 
   opt.homedir = default_homedir ();
-  opt.connect_flags = 1; /* Use extended connect mode.  */
+  opt.connect_flags = 1;
 
   /* Parse the command line. */
   pargs.argc  = &argc;
@@ -1176,6 +1181,7 @@ main (int argc, char **argv)
         case oHex:       opt.hex = 1; break;
         case oDecode:    opt.decode = 1; break;
         case oRawSocket: opt.raw_socket = pargs.r.ret_str; break;
+        case oTcpSocket: opt.tcp_socket = pargs.r.ret_str; break;
         case oExec:      opt.exec = 1; break;
         case oNoExtConnect: opt.connect_flags &= ~(1); break;
         case oRun:       opt_run = pargs.r.ret_str; break;
@@ -1206,8 +1212,23 @@ main (int argc, char **argv)
     cmdline_commands = argv;
 
   if (opt.exec && opt.raw_socket)
-    log_info (_("option \"%s\" ignored due to \"%s\"\n"),
-              "--raw-socket", "--exec");
+    {
+      opt.raw_socket = NULL;
+      log_info (_("option \"%s\" ignored due to \"%s\"\n"),
+                "--raw-socket", "--exec");
+    }
+  if (opt.exec && opt.tcp_socket)
+    {
+      opt.tcp_socket = NULL;
+      log_info (_("option \"%s\" ignored due to \"%s\"\n"),
+                "--tcp-socket", "--exec");
+    }
+  if (opt.tcp_socket && opt.raw_socket)
+    {
+      opt.tcp_socket = NULL;
+      log_info (_("option \"%s\" ignored due to \"%s\"\n"),
+                "--tcp-socket", "--raw-socket");
+    }
 
   if (opt_run && !(script_fp = fopen (opt_run, "r")))
     {
@@ -1221,12 +1242,20 @@ main (int argc, char **argv)
     {
       int no_close[3];
 
-      no_close[0] = fileno (stderr);
-      no_close[1] = log_get_fd ();
+      no_close[0] = assuan_fd_from_posix_fd (fileno (stderr));
+      no_close[1] = assuan_fd_from_posix_fd (log_get_fd ());
       no_close[2] = -1;
-      rc = assuan_pipe_connect_ext (&ctx, *argv, (const char **)argv,
-                                    no_close, NULL, NULL,
-                                    opt.connect_flags);
+
+      rc = assuan_new (&ctx);
+      if (rc)
+       {
+          log_error ("assuan_new failed: %s\n", gpg_strerror (rc));
+         exit (1);
+       }
+
+      rc = assuan_pipe_connect
+       (ctx, *argv, (const char **)argv, no_close, NULL, NULL,
+        (opt.connect_flags & 1) ? ASSUAN_PIPE_CONNECT_FDPASSING : 0);
       if (rc)
         {
           log_error ("assuan_pipe_connect_ext failed: %s\n",
@@ -1240,8 +1269,16 @@ main (int argc, char **argv)
     }
   else if (opt.raw_socket)
     {
-      rc = assuan_socket_connect_ext (&ctx, opt.raw_socket, 0,
-                                      opt.connect_flags);
+      rc = assuan_new (&ctx);
+      if (rc)
+       {
+          log_error ("assuan_new failed: %s\n", gpg_strerror (rc));
+         exit (1);
+       }
+
+      rc = assuan_socket_connect
+       (ctx, opt.raw_socket, 0,
+        (opt.connect_flags & 1) ? ASSUAN_SOCKET_CONNECT_FDPASSING : 0);
       if (rc)
         {
           log_error ("can't connect to socket `%s': %s\n",
@@ -1252,6 +1289,32 @@ main (int argc, char **argv)
       if (opt.verbose)
         log_info ("connection to socket `%s' established\n", opt.raw_socket);
     }
+  else if (opt.tcp_socket)
+    {
+      char *url;
+
+      url = xstrconcat ("assuan://", opt.tcp_socket, NULL);
+
+      rc = assuan_new (&ctx);
+      if (rc)
+       {
+          log_error ("assuan_new failed: %s\n", gpg_strerror (rc));
+         exit (1);
+       }
+
+      rc = assuan_socket_connect (ctx, opt.tcp_socket, 0, 0);
+      if (rc)
+        {
+          log_error ("can't connect to server `%s': %s\n",
+                     opt.tcp_socket, gpg_strerror (rc));
+          exit (1);
+        }
+
+      if (opt.verbose)
+        log_info ("connection to socket `%s' established\n", url);
+
+      xfree (url);
+    }
   else
     ctx = start_agent ();
 
@@ -1259,7 +1322,7 @@ main (int argc, char **argv)
      assuan did not run the initial handshaking).  */
   if (assuan_pending_line (ctx))
     {
-      rc = read_and_print_response (ctx, &cmderr);
+      rc = read_and_print_response (ctx, 0, &cmderr);
       if (rc)
         log_info (_("receiving line failed: %s\n"), gpg_strerror (rc) );
     }
@@ -1747,7 +1810,9 @@ main (int argc, char **argv)
       if (*line == '#' || !*line)
         continue; /* Don't expect a response for a comment line. */
 
-      rc = read_and_print_response (ctx, &cmderr);
+      rc = read_and_print_response (ctx, (!ascii_strncasecmp (line, "HELP", 4)
+                                          && (spacep (line+4) || !line[4])),
+                                    &cmderr);
       if (rc)
         log_info (_("receiving line failed: %s\n"), gpg_strerror (rc) );
       if ((rc || cmderr) && script_fp)
@@ -1876,14 +1941,15 @@ handle_inquire (assuan_context_t ctx, char *line)
 
 
 /* Read all response lines from server and print them.  Returns 0 on
-   success or an assuan error code.  Set R_GOTERR to true if the
-   command did not returned OK.  */
+   success or an assuan error code.  If WITHHASH istrue, comment lines
+   are printed.  Sets R_GOTERR to true if the command did not returned
+   OK.  */
 static int
-read_and_print_response (assuan_context_t ctx, int *r_goterr)
+read_and_print_response (assuan_context_t ctx, int withhash, int *r_goterr)
 {
   char *line;
   size_t linelen;
-  assuan_error_t rc;
+  gpg_error_t rc;
   int i, j;
   int need_lf = 0;
 
@@ -1896,7 +1962,7 @@ read_and_print_response (assuan_context_t ctx, int *r_goterr)
           if (rc)
             return rc;
 
-          if (opt.verbose > 1 && *line == '#')
+          if ((withhash || opt.verbose > 1) && *line == '#')
             {
               fwrite (line, linelen, 1, stdout);
               putchar ('\n');
@@ -2079,15 +2145,23 @@ start_agent (void)
   int rc = 0;
   char *infostr, *p;
   assuan_context_t ctx;
+  session_env_t session_env;
 
   infostr = getenv ("GPG_AGENT_INFO");
   if (!infostr || !*infostr)
     {
       char *sockname;
 
+      rc = assuan_new (&ctx);
+      if (rc)
+       {
+          log_error ("assuan_new failed: %s\n", gpg_strerror (rc));
+         exit (1);
+       }
+
       /* Check whether we can connect at the standard socket.  */
       sockname = make_filename (opt.homedir, "S.gpg-agent", NULL);
-      rc = assuan_socket_connect (&ctx, sockname, 0);
+      rc = assuan_socket_connect (ctx, sockname, 0, 0);
 
 #ifdef HAVE_W32_SYSTEM
       /* If we failed to connect under Windows, we fire up the agent.  */
@@ -2114,7 +2188,14 @@ start_agent (void)
               /* Give the agent some time to prepare itself. */
               gnupg_sleep (3);
               /* Now try again to connect the agent.  */
-              rc = assuan_socket_connect (&ctx, sockname, 0);
+             rc = assuan_new (&ctx);
+             if (rc)
+               {
+                 log_error ("assuan_new failed: %s\n", gpg_strerror (rc));
+                 exit (1);
+               }
+
+              rc = assuan_socket_connect (ctx, sockname, 0, 0);
             }
           if (rc)
             rc = save_rc;
@@ -2147,7 +2228,14 @@ start_agent (void)
           exit (1);
         }
 
-      rc = assuan_socket_connect (&ctx, infostr, pid);
+      rc = assuan_new (&ctx);
+      if (rc)
+       {
+          log_error ("assuan_new failed: %s\n", gpg_strerror (rc));
+         exit (1);
+       }
+
+      rc = assuan_socket_connect (ctx, infostr, pid, 0);
       xfree (infostr);
     }
 
@@ -2168,8 +2256,14 @@ start_agent (void)
       exit (1);
     }
 
+  session_env = session_env_new ();
+  if (!session_env)
+    log_fatal ("error allocating session environment block: %s\n",
+               strerror (errno));
+
   rc = send_pinentry_environment (ctx, GPG_ERR_SOURCE_DEFAULT,
-                                  NULL, NULL, NULL, NULL, NULL, NULL, NULL);
+                                  NULL, NULL, session_env);
+  session_env_release (session_env);
   if (rc)
     {
       log_error (_("error sending standard options: %s\n"), gpg_strerror (rc));